How to use setHeadersMap method of com.paypal.selion.grid.servlets.transfer.TransferContext class

Best SeLion code snippet using com.paypal.selion.grid.servlets.transfer.TransferContext.setHeadersMap

Source:UploadRequestProcessor.java

...162            managedArtifactList.add(managedArtifact);163        }164        private void populateHeadersMap() {165            checkRequiredParameters();166            transferContext.setHeadersMap(getRequestHeadersMap());167        }168        private void checkRequiredParameters() {169            if (StringUtils.isBlank(httpServletRequest.getHeader(ManagedArtifact.ARTIFACT_FILE_NAME))) {170                throw new ArtifactUploadException("Required header [" + ManagedArtifact.ARTIFACT_FILE_NAME171                        + "] is missing or has no value");172            }173            for (String param : managedArtifactRequestParameters.getParameters().keySet()) {174                boolean isRequired = managedArtifactRequestParameters.isRequired(param);175                if (isRequired && StringUtils.isBlank(httpServletRequest.getHeader(param))) {176                    throw new ArtifactUploadException("Required header [" + param + "] is missing or has no value");177                }178            }179        }180        private byte[] parseFileContents() throws IOException {181            int fileSize = httpServletRequest.getContentLength();182            if (fileSize <= 0) {183                throw new ArtifactUploadException("File is empty");184            }185            return IOUtils.toByteArray(httpServletRequest.getInputStream());186        }187    }188    /**189     * <code>MultipartUploadRequestProcessor</code> is an implementation of {@link AbstractUploadRequestProcessor} for190     * {@link DefaultManagedArtifact}. The implementation relies on 'commons-fileupload' library for parsing191     * 'multipart/form-data' type requests. Multiple artifact uploads are saved into repository and returned as a192     * {@link List} after processing. The clients pass 'folderName' is an optional parameter. The clients may choose to193     * pass them as either HTTP headers or request parameters: if using CURL then -F option (name=value) pair or -H194     * (HTTP headers). Additional HTTP headers or request parameters may apply and are defined by the195     * {@link ManagedArtifact} implementation. The implementation limits to only one file upload. Sample curl command196     * for uploading a multipart file197     * 198     * <pre>199     * {@code200     * curl -v -H 'folderName:<folderName>' -F file=@/path/tofile http://[hostname]:[port]/[upload-context-path]201     * }202     * </pre>203     */204    final class MultipartUploadRequestProcessor extends AbstractUploadRequestProcessor {205        private static final SeLionGridLogger LOGGER = SeLionGridLogger206                .getLogger(MultipartUploadRequestProcessor.class);207        private ServletFileUpload servletFileUpload;208        private List<FileItem> fileItems;209        public MultipartUploadRequestProcessor(TransferContext transferContext) {210            super(transferContext);211            initializeApacheCommonsSystem();212        }213        public void populateManagedArtifactList() {214            LOGGER.entering();215            try {216                saveUploadedData();217            } catch (FileUploadException e) {218                throw new ArtifactUploadException(e.getMessage());219            }220            LOGGER.exiting();221        }222        private void saveUploadedData() throws FileUploadException {223            LOGGER.entering();224            int count = parseRequestAsFileItems();225            if (count > 1) {226                throw new ArtifactUploadException("Only one file supported for upload using multipart");227            }228            // Get parameters from headers and override it with request parameters.229            populateHeadersMap();230            for (FileItem fileItem : fileItems) {231                if (!fileItem.isFormField()) {232                    UploadedArtifact uploadedArtifact = createUploadedArtifactUsing(transferContext.getHeadersMap(),233                            fileItem.get());234                    ManagedArtifact managedArtifact = repository.saveContents(uploadedArtifact);235                    managedArtifactList.add(managedArtifact);236                }237            }238            LOGGER.exiting();239        }240        private int parseRequestAsFileItems() throws FileUploadException {241            int fileCount = 0;242            if (fileItems == null) {243                fileItems = servletFileUpload.parseRequest(httpServletRequest);244            }245            for (FileItem fileItem : fileItems) {246                if (!fileItem.isFormField()) {247                    ++fileCount;248                }249            }250            return fileCount;251        }252        private void populateHeadersMap() {253            Map<String, String> headersMap = getRequestHeadersMap();254            Map<String, Boolean> artifactParams = managedArtifactRequestParameters.getParameters();255            for (FileItem fileItem : fileItems) {256                if (fileItem.isFormField()) {257                    String parameter = fileItem.getFieldName().trim();258                    if (artifactParams.containsKey(parameter)) {259                        headersMap.put(parameter, fileItem.getString().trim());260                    }261                } else {262                    // TODO fix assumption that the only other parameter is the fileName263                    headersMap.put(ManagedArtifact.ARTIFACT_FILE_NAME, isNotBlank(fileItem.getName().trim()));264                }265            }266            checkForRequiredParameters(headersMap);267            transferContext.setHeadersMap(headersMap);268        }269        // TODO See if this can be merged with ApplicationUploadRequestProcessor#checkRequiredParameters270        private void checkForRequiredParameters(Map<String, String> headersMap) {271            if (!headersMap.containsKey(ManagedArtifact.ARTIFACT_FILE_NAME)) {272                throw new ArtifactUploadException("Required input ["273                        + ManagedArtifact.ARTIFACT_FILE_NAME + "] is missing or has no value");274            }275            for (String param : managedArtifactRequestParameters.getParameters().keySet()) {276                boolean isRequired = managedArtifactRequestParameters.isRequired(param);277                if (isRequired && StringUtils.isBlank(headersMap.get(param))) {278                    throw new ArtifactUploadException("Required input [" + param279                            + "] is missing or has no value");280                }281            }...

setHeadersMap

Using AI Code Generation

1package com.paypal.selion.grid.servlets.transfer;2import java.io.IOException;3import java.util.ArrayList;4import java.util.Arrays;5import java.util.List;6import java.util.Map;7import java.util.Set;8import javax.servlet.ServletException;9import javax.servlet.http.HttpServletRequest;10import javax.servlet.http.HttpServletResponse;11import org.apache.commons.lang.StringUtils;12import org.openqa.selenium.remote.server.rest.ResultConfig;13import org.openqa.selenium.remote.server.rest.ResultType;14import com.google.gson.JsonElement;15import com.google.gson.JsonObject;16import com.google.gson.JsonParser;17import com.paypal.selion.grid.servlets.transfer.TransferContext.TransferMode;18public class TransferContext {19    private static final String TRANSFER_MODE = "transferMode";20    private static final String TRANSFER_MODES = "transferModes";21    private static final String TRANSFER_MODES_LIST = "transferModesList";22    private static final String TRANSFER_MODES_MAP = "transferModesMap";23    private static final String TRANSFER_MODES_SET = "transferModesSet";24    private static final String TRANSFER_MODES_ARRAY = "transferModesArray";25    private static final String TRANSFER_MODES_JSON_OBJECT = "transferModesJsonObject";26    private static final String TRANSFER_MODES_JSON_ARRAY = "transferModesJsonArray";27    private static final String TRANSFER_MODES_JSON_ELEMENT = "transferModesJsonElement";28    private static final String TRANSFER_MODES_JSON_PARSER = "transferModesJsonParser";29    private static final String TRANSFER_MODES_RESULT_TYPE = "transferModesResultType";30    private static final String TRANSFER_MODES_RESULT_CONFIG = "transferModesResultConfig";31    private static final String TRANSFER_MODES_STRING = "transferModesString";32    private static final String TRANSFER_MODES_STRING_ARRAY = "transferModesStringArray";33    private static final String TRANSFER_MODES_STRING_LIST = "transferModesStringList";34    private static final String TRANSFER_MODES_STRING_SET = "transferModesStringSet";35    private static final String TRANSFER_MODES_STRING_MAP = "transferModesStringMap";36    private static final String TRANSFER_MODES_STRING_JSON_OBJECT = "transferModesStringJsonObject";37    private static final String TRANSFER_MODES_STRING_JSON_ARRAY = "transferModesStringJsonArray";38    private static final String TRANSFER_MODES_STRING_JSON_ELEMENT = "transferModesStringJsonElement";39    private static final String TRANSFER_MODES_STRING_JSON_PARSER = "transferModesStringJsonParser";

setHeadersMap

Using AI Code Generation

1import com.paypal.selion.grid.servlets.transfer.TransferContext;2import com.paypal.selion.grid.servlets.transfer.TransferException;3import java.util.HashMap;4import java.util.Map;5public class TransferContextExample {6    public static void main(String[] args) throws TransferException {7        TransferContext context = new TransferContext();8        Map<String, String> headers = new HashMap<String, String>();9        headers.put("MyHeader", "MyValue");10        context.setHeadersMap(headers);

setHeadersMap

Using AI Code Generation

1TransferContext.setHeadersMap(headersMap);2        context.performTransfer();3    }4}5import com.paypal.selion.grid.servlets.transfer.TransferContext;6import com.paypal.selion.grid.servlets.transfer.TransferException;7public class TransferContextExample {8    public static void main(String[] args) throws TransferException {9        TransferContext context = new TransferContext();10        context.setHeaders("MyHeader: MyValue");11        context.performTransfer();12    }13}14[INFO] --- maven-surefire-plugin:2.19.1:test (default-test) @ selion-grid ---

setHeadersMap

Using AI Code Generation

1package com.paypal.selion.grid.servlets.transfer;2import java.util.HashMap;3import java.util.Map;4import org.testng.Assert;5import org.testng.annotations.Test;6public class TransferContextTest {7    public void testSetHeadersMap() {8        TransferContext context = new TransferContext();9        Map<String, String> headers = new HashMap<String, String>();10        headers.put("Content-Type", "text/plain");11        headers.put("Content-Length", "1234");12        context.setHeadersMap(headers);13        Assert.assertEquals(context.getHeadersMap().get("Content-Type"), "text/plain");14        Assert.assertEquals(context.getHeadersMap().get("Content-Length"), "1234");15    }16}

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.