JHttpScenarioGlobalContext.java

Go to the documentation of this file.

package com.griddynamics.jagger.invoker.scenario;

import com.griddynamics.jagger.invoker.v2.JHttpEndpoint;
import org.springframework.http.HttpHeaders;

import java.util.List;
import java.util.Map;

import static com.google.common.collect.Lists.newArrayList;

public class JHttpScenarioGlobalContext {
    private String userName;
    private String password;
    private JHttpEndpoint globalEndpoint;
    private HttpHeaders globalHeaders;

    public JHttpScenarioGlobalContext copy() {
        JHttpScenarioGlobalContext copy = new JHttpScenarioGlobalContext();
        return copy.withBasicAuth(this.userName, this.password)
                .withGlobalEndpoint(JHttpEndpoint.copyOf(this.globalEndpoint))
                .withGlobalHeaders(CopyUtil.copyOf(this.globalHeaders));
    }

    public JHttpScenarioGlobalContext withGlobalEndpoint(JHttpEndpoint globalEndpoint) {
        this.globalEndpoint = globalEndpoint;
        return this;
    }

    public JHttpScenarioGlobalContext withBasicAuth(String userName, String password) {
        this.userName = userName;
        this.password = password;
        return this;
    }

    public JHttpScenarioGlobalContext withGlobalHeaders(HttpHeaders globalHeaders) {
        this.globalHeaders = globalHeaders;
        return this;
    }

    public JHttpScenarioGlobalContext withGlobalHeaders(Map<String, List<String>> globalHeaders) {
        initHeadersIfNull();
        this.globalHeaders.putAll(globalHeaders);
        return this;
    }

    public JHttpScenarioGlobalContext withGlobalHeader(String key, String value) {
        initHeadersIfNull();
        globalHeaders.put(key, newArrayList(value));
        return this;
    }

    public JHttpScenarioGlobalContext withGlobalCookies(Map<String, String> cookies) {
        initHeadersIfNull();
        cookies.entrySet().forEach(entry -> this.globalHeaders.add("Cookie", entry.getKey() + "=" + entry.getValue()));
        return this;
    }

    public JHttpScenarioGlobalContext withGlobalCookie(String name, String value) {
        initHeadersIfNull();
        this.globalHeaders.add("Cookie", name + "=" + value);
        return this;
    }

    private void initHeadersIfNull() {
        if (this.globalHeaders == null)
            this.globalHeaders = new HttpHeaders();
    }

    public JHttpEndpoint getGlobalEndpoint() {
        return globalEndpoint;
    }

    public String getUserName() {
        return userName;
    }

    public String getPassword() {
        return password;
    }

    public HttpHeaders getGlobalHeaders() {
        return globalHeaders;
    }

    public List<String> getGlobalCookies() {
        return globalHeaders.get("Cookie");
    }
}