liquibase.sdk.verifytest.VerifiedTest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of liquibase-core Show documentation
There is a newer version: 4.30.0
package liquibase.sdk.verifytest;

import java.util.*;

public class VerifiedTest {

    private String testClass;
    private String testName;

    private Map permutationsByKey = new HashMap();
    private List unkeyedPermutations = new ArrayList();

    public VerifiedTest(String testClass, String testName) {
        this.testName = testName;
        this.testClass = testClass;
    }

    public String getTestClass() {
        return testClass;
    }

    public boolean hasGroups() {
        Collection permutations = getPermutations();
        return permutations != null && permutations.size() > 0 && permutations.iterator().next().getGroup() != null;
    }

    public Set getGroups() {
        Set returnSet = new HashSet();
        for (TestPermutation permutation : getPermutations()) {
            if (permutation.getGroup() != null) {
                returnSet.add(permutation.getGroup());
            }
        }

        return returnSet;
    }

    public String getTestName() {
        return testName;
    }

    public Collection getPermutations() {
        ArrayList returnList = new ArrayList();
        returnList.addAll(permutationsByKey.values());
        returnList.addAll(unkeyedPermutations);

        return Collections.unmodifiableCollection(returnList);
    }

    public TestPermutation getPermutation(String key) {
        if (!permutationsByKey.containsKey(key)) {
            rebuildPermutationsByKey();
        }
        return permutationsByKey.get(key);
    }

    public TestPermutation addPermutation(TestPermutation permutation) {
        String key = permutation.getKey();
        if (key.equals("")) {
            unkeyedPermutations.add(permutation);
        } else {
            permutationsByKey.put(key, permutation);
        }

        return permutation;
    }

    public void replacePermutation(TestPermutation permutation) {
        removePermutation(permutation.getKey());
        addPermutation(permutation);
    }

    public void removePermutation(String key) {
        TestPermutation permutation = getPermutation(key);
        if (permutation != null) {
            permutationsByKey.remove(permutation.getKey());
        }
    }

    private void rebuildPermutationsByKey() {
        List newUnkeyed = new ArrayList();
        for (TestPermutation permutation : unkeyedPermutations) {
            if (permutation.getKey().equals("")) {
                newUnkeyed.add(permutation);
            } else {
                permutationsByKey.put(permutation.getKey(), permutation);
            }
        }
        this.unkeyedPermutations = newUnkeyed;
    }
}