net.jqwik.engine.providers.AbstractCollectionArbitraryProvider 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.providers;

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

import net.jqwik.api.*;
import net.jqwik.api.providers.*;

abstract class AbstractCollectionArbitraryProvider implements ArbitraryProvider {

	@Override
	public boolean canProvideFor(TypeUsage targetType) {
		return targetType.isAssignableFrom(getProvidedType());
	}

	protected abstract Class getProvidedType();

	@Override
	public Set> provideFor(TypeUsage targetType, SubtypeProvider subtypeProvider) {
		TypeUsage innerType = targetType.getTypeArguments().isEmpty() ? //
			TypeUsage.forType(Object.class) //
			: targetType.getTypeArguments().get(0);
		Set> elementArbitraries = subtypeProvider.apply(innerType);
		return elementArbitraries.stream() //
								 .map(this::create) //
								 .collect(Collectors.toSet());
	}

	protected abstract Arbitrary create(Arbitrary innerArbitrary);

}