jenkinsci/hpe-application-automation-tools-plugin

/*
 * Certain versions of software accessible here may contain branding from Hewlett-Packard Company (now HP Inc.) and Hewlett Packard Enterprise Company.
 * This software was acquired by Micro Focus on September 1, 2017, and is now offered by OpenText.
 * Any reference to the HP and Hewlett Packard Enterprise/HPE marks is historical in nature, and the HP and Hewlett Packard Enterprise/HPE marks are the property of their respective owners.
 * __________________________________________________________________
 * MIT License
 *
 * Copyright 2012-2024 Open Text
 *
 * The only warranties for products and services of Open Text and
 * its affiliates and licensors ("Open Text") are as may be set forth
 * in the express warranty statements accompanying such products and services.
 * Nothing herein should be construed as constituting an additional warranty.
 * Open Text shall not be liable for technical or editorial errors or
 * omissions contained herein. The information contained herein is subject
 * to change without notice.
 *
 * Except as specifically indicated otherwise, this document contains
 * confidential information and a valid license is required for possession,
 * use or copying. If this work is provided to the U.S. Government,
 * consistent with FAR 12.211 and 12.212, Commercial Computer Software,
 * Computer Software Documentation, and Technical Data for Commercial Items are
 * licensed to the U.S. Government under vendor's standard commercial license.
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ___________________________________________________________________
 */

package com.microfocus.application.automation.tools.results.service.rest;

import com.microfocus.application.automation.tools.sse.sdk.Client;
import com.microfocus.application.automation.tools.sse.sdk.ResourceAccessLevel;
import com.microfocus.application.automation.tools.sse.sdk.Response;
import com.microfocus.application.automation.tools.sse.sdk.request.PostRequest;
import com.microfocus.adm.performancecenter.plugins.common.rest.RESTConstants;

import java.util.HashMap;
import java.util.Map;

public class CreateAttachment extends PostRequest {

    private Client client;
    private String testsetId;
    private String fileName;
    private byte[] filecontent;
    private String entityCollectionName;

    private static final String CONTENT_TYPE = "application/octet-stream";

    public CreateAttachment(String entityCollectionName, Client client, String id, String fileName, byte[] filecontent) {
        super(client, "");
        this.entityCollectionName = entityCollectionName;
        this.client = client;
        this.testsetId = id;
        this.fileName = fileName;
        this.filecontent = filecontent;
    }

    @Override
    protected Map<String, String> getHeaders() {
        Map<String, String> ret = new HashMap<String, String>();
        ret.put(RESTConstants.CONTENT_TYPE, CONTENT_TYPE + ";");
        ret.put("Slug", fileName);
        ret.put("X-XSRF-TOKEN", client.getXsrfTokenValue());
        return ret;
    }

    @Override
    protected String getSuffix() {
        StringBuilder builder = new StringBuilder();
        builder.append(entityCollectionName).append("/").append(testsetId).append("/attachments");
        return builder.toString();
    }

    @Override
    public Response perform() {
        return client.httpPost(
                getUrl(),
                getDataBytes(),
                getHeaders(),
                ResourceAccessLevel.PROTECTED);
    }

    private byte[] getDataBytes() {
        return filecontent;
    }
}