com.redfin.fuzzy.cases.ExcludingCase Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fuzzy-core Show documentation

The newest version!

package com.redfin.fuzzy.cases;

import com.redfin.fuzzy.Case;
import com.redfin.fuzzy.FuzzyPreconditions;
import com.redfin.fuzzy.Subcase;
import com.redfin.fuzzy.Subcases;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;

public class ExcludingCase implements Case {

	private static final int MAX_ATTEMPTS = 100;

	private final Case baseCase;
	private final Set excludedValues = new HashSet();

	public ExcludingCase(Case baseCase, T excludedValue) {
		this(baseCase, Collections.singleton(excludedValue));
	}

	public ExcludingCase(Case baseCase, T... excludedValues) {
		this(baseCase, excludedValues == null ? null : Arrays.asList(excludedValues));
	}

	public ExcludingCase(Case baseCase, Iterable excludedValues) {
		FuzzyPreconditions.checkNotNull(baseCase);

		this.baseCase = baseCase;

		if(excludedValues != null) {
			for (T t : excludedValues)
				this.excludedValues.add(t);
		}
	}

	@Override
	public Set> getSubcases() {
		return Subcases.map(
			baseCase.getSubcases(),
			subcase -> (r -> {
				for(int i = 0; i < MAX_ATTEMPTS; i++) {
					T t = subcase.generate(r);
					if(!excludedValues.contains(t)) {
						return t;
					}
				}
				throw new IllegalStateException(String.format(
					"Failed to exclude unwanted values from a base case of type %s.",
					baseCase.getClass()
				));
			})
		);
	}

	@Override
	public ExcludingCase excluding(Iterable values) {
		if(values != null) {
			for(T t : values)
				excludedValues.add(t);
		}
		return this;
	}

}