net.jqwik.engine.properties.arbitraries.combinations.DefaultCombinator7 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 DefaultCombinator7
	extends DefaultCombinator6
	implements Combinators.Combinator7 {

	protected final Arbitrary a7;

	public DefaultCombinator7(
		Arbitrary a1, Arbitrary a2, Arbitrary a3, Arbitrary a4, Arbitrary a5, Arbitrary a6, Arbitrary a7
	) {
		super(a1, a2, a3, a4, a5, a6);
		this.a7 = a7;
	}

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

	@Override
	public Combinators.Combinator7 filter(Combinators.F7 filter) {
		return new Filtered<>(a1, a2, a3, a4, a5, a6, a7, filter);
	}

	@SuppressWarnings("unchecked")
	protected  Function, R> combineFunction(Combinators.F7 combinator) {
		return params -> combinator.apply(
			(T1) params.get(0), (T2) params.get(1),
			(T3) params.get(2), (T4) params.get(3),
			(T5) params.get(4), (T6) params.get(5),
			(T7) params.get(6)
		);
	}

	private static class Filtered extends DefaultCombinator7 {
		private final Combinators.F7 filter;

		private Filtered(
			Arbitrary a1,
			Arbitrary a2,
			Arbitrary a3,
			Arbitrary a4,
			Arbitrary a5,
			Arbitrary a6,
			Arbitrary a7,
			Combinators.F7 filter
		) {
			super(a1, a2, a3, a4, a5, a6, a7);
			this.filter = filter;
		}

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

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

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

	}
}