net.jqwik.time.internal.properties.arbitraries.valueRanges.AllowedMonths Maven / Gradle / Ivy

Go to download

package net.jqwik.time.internal.properties.arbitraries.valueRanges;

import java.time.*;
import java.util.*;
import java.util.stream.*;

import net.jqwik.api.support.*;

public class AllowedMonths extends AllowedUnits {

	@Override
	protected void setDefaultAllowed() {
		allowed = new LinkedHashSet<>(Arrays.asList(Month.values()));
	}

	public void set(MonthBetween monthBetween) {
		allowed = Arrays.stream(Month.values())
						.filter(m -> m.compareTo(monthBetween.getMin()) >= 0 && m.compareTo(monthBetween.getMax()) <= 0)
						.collect(CollectorsSupport.toLinkedHashSet());
	}

}