net.jqwik.engine.properties.arbitraries.combinations.DefaultCombinator4 Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jqwik-engine Show documentation

Jqwik Engine module

There is a newer version: 1.9.1

package net.jqwik.engine.properties.arbitraries.combinations;

import java.util.*;
import java.util.function.*;

import net.jqwik.api.*;

import org.jspecify.annotations.*;

public class DefaultCombinator4
	extends DefaultCombinator3
	implements Combinators.Combinator4 {

	protected final Arbitrary a4;

	public DefaultCombinator4(Arbitrary a1, Arbitrary a2, Arbitrary a3, Arbitrary a4) {
		super(a1, a2, a3);
		this.a4 = a4;
	}

	@Override
	public  Arbitrary as(Combinators.F4 combinator) {
		return new CombineArbitrary<>(combineFunction(combinator), a1, a2, a3, a4);
	}

	@Override
	public Combinators.Combinator4 filter(Combinators.F4 filter) {
		return new Filtered<>(a1, a2, a3, a4, filter);
	}

	@SuppressWarnings("unchecked")
	protected  Function, R> combineFunction(Combinators.F4 combinator) {
		return params -> combinator.apply((T1) params.get(0), (T2) params.get(1), (T3) params.get(2), (T4) params.get(3));
	}

	private static class Filtered extends DefaultCombinator4 {
		private final Combinators.F4 filter;

		private Filtered(
			Arbitrary a1,
			Arbitrary a2,
			Arbitrary a3,
			Arbitrary a4,
			Combinators.F4 filter
		) {
			super(a1, a2, a3, a4);
			this.filter = filter;
		}

		@Override
		public  Arbitrary as(Combinators.F4 combinator) {
			return new CombineArbitrary<>(Function.identity(), a1, a2, a3, a4)
					   .filter(combineFunction(filter)::apply)
					   .map(combineFunction(combinator));
		}

		@Override
		public Combinators.Combinator4 filter(Combinators.F4 filter) {
			return super.filter(combineFilters(this.filter, filter));
		}

		private Combinators.F4 combineFilters(
			Combinators.F4 first,
			Combinators.F4 second
		) {
			return (p1, p2, p3, p4) -> first.apply(p1, p2, p3, p4) && second.apply(p1, p2, p3, p4);
		}

	}
}