JHttpResponseStatusValidatorProvider.java

Go to the documentation of this file.

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

import com.griddynamics.jagger.coordinator.NodeContext;

import java.util.Arrays;
import java.util.Collection;
import java.util.List;
import java.util.Set;
import java.util.regex.Pattern;
import java.util.regex.PatternSyntaxException;
import java.util.stream.IntStream;

import static com.google.common.collect.Sets.newHashSet;
import static java.util.stream.Collectors.toSet;

public class JHttpResponseStatusValidatorProvider implements ResponseValidatorProvider {

    private Set<Integer> validStatusCodes = newHashSet();
    private Pattern validStatusCodesRegexp;

    private JHttpResponseStatusValidatorProvider(Collection<Integer> validStatusCodes) {
        this.validStatusCodes.addAll(validStatusCodes);
    }

    private JHttpResponseStatusValidatorProvider(Pattern validStatusCodesRegexp) {
        this.validStatusCodesRegexp = validStatusCodesRegexp;
    }

    @Override
    public ResponseValidator<?, ?, ?> provide(String sessionId, String taskId, NodeContext kernelContext) {
        JHttpResponseStatusValidator jHttpResponseStatusValidator = new JHttpResponseStatusValidator(taskId, sessionId, kernelContext);
        jHttpResponseStatusValidator.setValidStatusCodes(validStatusCodes);
        jHttpResponseStatusValidator.setValidStatusCodesRegexp(validStatusCodesRegexp);
        return jHttpResponseStatusValidator;
    }

    public static JHttpResponseStatusValidatorProvider of(Integer... codes) {
        return new JHttpResponseStatusValidatorProvider(Arrays.asList(codes));
    }

    public static JHttpResponseStatusValidatorProvider of(List<Integer> codes) {
        return new JHttpResponseStatusValidatorProvider(codes);
    }

    public static JHttpResponseStatusValidatorProvider of(IntStream range) {
        return new JHttpResponseStatusValidatorProvider(range.boxed().collect(toSet()));
    }

    public static JHttpResponseStatusValidatorProvider of(String stringRegExp) {
        return new JHttpResponseStatusValidatorProvider(Pattern.compile(stringRegExp));
    }
}