JLoadScenario.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.loadscenario.ExampleLoadScenarioListener;
import com.griddynamics.jagger.engine.e1.collector.loadscenario.LoadScenarioListener;
import com.griddynamics.jagger.user.test.configurations.auxiliary.Id;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

public class JLoadScenario {

    private final String id;
    private final List<JParallelTestsGroup> testGroups;
    private final List<Double> percentileValues;
    private final List<Provider<LoadScenarioListener>> listeners;

    private JLoadScenario(Builder builder) {
        this.id = builder.id.value();
        this.testGroups = builder.testGroups;
        this.percentileValues = builder.percentileValues;
        this.listeners = builder.listeners;
    }

    public static Builder builder(Id id, List<JParallelTestsGroup> testGroups) {
        return new Builder(id, testGroups);
    }

    public static Builder builder(Id id, JParallelTestsGroup testGroup, JParallelTestsGroup... testGroups) {
        List<JParallelTestsGroup> jParallelTestsGroupList = new ArrayList<>();
        jParallelTestsGroupList.add(testGroup);
        Collections.addAll(jParallelTestsGroupList, testGroups);
        
        return new Builder(id, jParallelTestsGroupList);
    }

    public static class Builder {
        private final Id id;
        private final List<JParallelTestsGroup> testGroups;
        private List<Double> percentileValues;
        private List<Provider<LoadScenarioListener>> listeners = Lists.newArrayList();
        
        public Builder(Id id, List<JParallelTestsGroup> testGroups) {
            this.id = id;
            this.testGroups = testGroups;
        }

        public Builder withLatencyPercentiles(List<Double> percentileValues) {
            Objects.requireNonNull(percentileValues);
            percentileValues.stream().filter(percentile -> percentile <= 0.0 || percentile >= 100.0)
                    .findAny()
                    .ifPresent(badPercentile -> {
                        throw new IllegalArgumentException("Percentile value must be > 0 and < 100. Provided value is " + badPercentile);
                    });

            this.percentileValues = percentileValues;
            return this;
        }
    
        public Builder addListener(Provider<LoadScenarioListener> listener) {
            this.listeners.add(listener);
            return this;
        }
    
        public Builder addListeners(List<Provider<LoadScenarioListener>> listeners) {
            this.listeners.addAll(listeners);
            return this;
        }

        public JLoadScenario build() {
    
            if (testGroups.stream().map(JParallelTestsGroup::getId).collect(Collectors.toSet()).size() < testGroups.size()) {
                throw new IllegalStateException("JParallelTestsGroup ids inside one JLoadScenario are not all unique");
            }
            
            return new JLoadScenario(this);
        }

    }

    public List<JParallelTestsGroup> getTestGroups() {
        return testGroups;
    }

    public String getId() {
        return id;
    }

    public List<Double> getPercentileValues() {
        return percentileValues;
    }
    
    public List<Provider<LoadScenarioListener>> getListeners() {
        return listeners;
    }
}