net.jqwik.vavr.providers.control.VavrTryArbitraryProvider Maven / Gradle / Ivy

Go to download
package net.jqwik.vavr.providers.control;

import io.vavr.collection.List;
import io.vavr.control.Try;
import net.jqwik.api.Arbitraries;
import net.jqwik.api.Arbitrary;
import net.jqwik.api.providers.TypeUsage;
import net.jqwik.vavr.arbitraries.control.VavrTryArbitrary;
import net.jqwik.vavr.providers.base.AbstractArbitraryProvider;

import java.util.Collections;
import java.util.Set;
import java.util.stream.Collectors;

public class VavrTryArbitraryProvider extends AbstractArbitraryProvider {

    private static final List DEFAULT_EXCEPTIONS = List.of(
            new Throwable(),
            new Exception(),
            new RuntimeException()
    );

    @Override
    protected Class getProvidedType() {
        return Try.class;
    }

    @Override
    public Set> provideFor(final TypeUsage targetType, final SubtypeProvider subtypeProvider) {
        final TypeUsage innerType = targetType.getTypeArgument(0);
        final Set> innerArbitraries = subtypeProvider.apply(innerType);
        final Set> exceptionArbitraries = Collections.singleton(
                Arbitraries.of(DEFAULT_EXCEPTIONS.toJavaArray(Throwable[]::new)));
        return innerArbitraries.stream()
                .flatMap(firstArbitrary -> exceptionArbitraries.stream()
                        .map(exceptionArbitrary -> create(firstArbitrary, exceptionArbitrary)))
                .collect(Collectors.toSet());
    }

    private Arbitrary create(final Arbitrary innerArbitrary, final Arbitrary exceptionArbitrary) {
        return new VavrTryArbitrary<>(innerArbitrary, exceptionArbitrary);
    }

}