JHttpResponseStatusValidator.java

Go to the documentation of this file.

package com.griddynamics.jagger.engine.e1.collector;

import com.griddynamics.jagger.coordinator.NodeContext;
import com.griddynamics.jagger.invoker.v2.JHttpEndpoint;
import com.griddynamics.jagger.invoker.v2.JHttpQuery;
import com.griddynamics.jagger.invoker.v2.JHttpResponse;

import java.util.Set;
import java.util.regex.Pattern;

import static com.google.common.collect.Sets.newHashSet;

public class JHttpResponseStatusValidator extends ResponseValidator<JHttpQuery, JHttpEndpoint, JHttpResponse> {

    private Set<Integer> validStatusCodes = newHashSet();

    private Pattern validStatusCodesRegexp;

    public JHttpResponseStatusValidator(String taskId, String sessionId, NodeContext kernelContext) {
        super(taskId, sessionId, kernelContext);
    }

    @Override
    public String getName() {
        return "Default Http Response Status Validator";
    }

    @Override
    public boolean validate(JHttpQuery query, JHttpEndpoint endpoint, JHttpResponse result, long duration) {
        if (validStatusCodesRegexp != null) {
            return validStatusCodesRegexp.matcher(result.getStatus().toString()).matches();
        }
        return validStatusCodes.contains(result.getStatus().value());
    }

    public Set<Integer> getValidStatusCodes() {
        return validStatusCodes;
    }

    public void setValidStatusCodes(Set<Integer> validStatusCodes) {
        this.validStatusCodes = validStatusCodes;
    }

    public Pattern getValidStatusCodesRegexp() {
        return validStatusCodesRegexp;
    }

    public void setValidStatusCodesRegexp(Pattern validStatusCodesRegexp) {
        this.validStatusCodesRegexp = validStatusCodesRegexp;
    }
}