JTestDefinition.java

Go to the documentation of this file.

package com.griddynamics.jagger.user.test.configurations;

import com.google.common.collect.Lists;
import com.griddynamics.jagger.engine.e1.Provider;
import com.griddynamics.jagger.engine.e1.collector.ResponseValidatorProvider;
import com.griddynamics.jagger.engine.e1.collector.invocation.InvocationListener;
import com.griddynamics.jagger.invoker.Invoker;
import com.griddynamics.jagger.invoker.QueryPoolLoadBalancer;
import com.griddynamics.jagger.invoker.RandomLoadBalancer;
import com.griddynamics.jagger.invoker.RoundRobinLoadBalancer;
import com.griddynamics.jagger.invoker.RoundRobinPairSupplierFactory;
import com.griddynamics.jagger.invoker.v2.DefaultHttpInvoker;
import com.griddynamics.jagger.invoker.v2.DefaultInvokerProvider;
import com.griddynamics.jagger.user.test.configurations.auxiliary.Id;

import java.util.List;

public class JTestDefinition {

    private final String id;
    private final Iterable endpoints;

    private final String comment;
    private final Iterable queries;
    private final Provider<Invoker> invoker;
    private final List<ResponseValidatorProvider> validators;
    private final List<Provider<InvocationListener>> listeners;
    private final QueryPoolLoadBalancer loadBalancer;

    private JTestDefinition(Builder builder) {
        this.id = builder.id.value();
        this.endpoints = builder.endpointsProvider;

        this.comment = (builder.comment == null) ? "" : builder.comment;
        this.queries = builder.queries;
        this.invoker = builder.invoker;
        this.validators = builder.validators;
        this.listeners = builder.listeners;
        this.loadBalancer = builder.loadBalancer;
    }

    public static Builder builder(Id id, Iterable endpointsProvider) {
        return new Builder(id, endpointsProvider);
    }

    public static class Builder {
        private final Id id;
        private final Iterable endpointsProvider;

        private String comment = "";
        private Iterable queries;
        private Provider<Invoker> invoker = DefaultInvokerProvider.of(DefaultHttpInvoker.class);
        private List<ResponseValidatorProvider> validators = Lists.newArrayList();
        private List<Provider<InvocationListener>> listeners = Lists.newArrayList();
        private QueryPoolLoadBalancer loadBalancer;

        private Builder(Id id, Iterable endpointsProvider) {
            this.id = id;
            this.endpointsProvider = endpointsProvider;
            this.loadBalancer = new RandomLoadBalancer() {{
                setPairSupplierFactory(new RoundRobinPairSupplierFactory());
                setRandomSeed(31);
            }};
        }

        public Builder withComment(String comment) {
            this.comment = comment;
            return this;
        }

        public Builder withQueryProvider(Iterable queryProvider) {
            this.queries = queryProvider;
            return this;
        }

        public Builder withLoadBalancer(QueryPoolLoadBalancer loadBalancer) {
            this.loadBalancer = loadBalancer;
            return this;
        }

        public Builder withInvoker(Provider<Invoker> invoker) {
            this.invoker = invoker;
            return this;
        }

        public Builder addValidators(List<ResponseValidatorProvider> validators) {
            this.validators.addAll(validators);
            return this;
        }
    
        public Builder addValidator(ResponseValidatorProvider validator) {
            this.validators.add(validator);
            return this;
        }
    
        public Builder addListeners(List<Provider<InvocationListener>> listeners) {
            this.listeners.addAll(listeners);
            return this;
        }
    
        public Builder addListener(Provider<InvocationListener> listener) {
            this.listeners.add(listener);
            return this;
        }
    
        public JTestDefinition build() {
            return new JTestDefinition(this);
        }
    }

    public String getId() {
        return id;
    }

    public String getDescription() {
        return comment;
    }

    public Iterable getEndpoints() {
        return endpoints;
    }

    public Iterable getQueries() {
        return queries;
    }

    public Provider<Invoker> getInvoker() {
        return invoker;
    }

    public String getComment() {
        return comment;
    }

    public List<ResponseValidatorProvider> getValidators() {
        return validators;
    }
    
    public List<Provider<InvocationListener>> getListeners() {
        return listeners;
    }

    public QueryPoolLoadBalancer getLoadBalancer() {
        return loadBalancer;
    }
}