net.jqwik.engine.properties.shrinking.ShrunkSampleRecreator 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.shrinking;

import java.util.*;
import java.util.concurrent.atomic.*;
import java.util.function.*;
import java.util.stream.*;

import net.jqwik.api.*;
import net.jqwik.api.lifecycle.*;
import net.jqwik.engine.properties.*;

public class ShrunkSampleRecreator {

	private final List> shrinkables;

	public ShrunkSampleRecreator(List> shrinkables) {
		this.shrinkables = shrinkables;
	}

	public Optional>> recreateFrom(List shrinkingSequence) {
		List recreatingSequence = new ArrayList<>(shrinkingSequence);
		Falsifier> recreatingFalsifier = falsifier(recreatingSequence);

		FalsifiedSample originalSample = createFalsifiedSample();

		AtomicInteger shrinkingSteps = new AtomicInteger(0);
		FalsifiedSample[] currentBest = new FalsifiedSample[]{originalSample};
		Consumer sampleShrunkConsumer = shrunkSample -> {
			// Remember current best because shrinking can be interrupted with RecreationDone exception
			currentBest[0] = shrunkSample;
			shrinkingSteps.incrementAndGet();
		};
		ShrinkingAlgorithm plainShrinker = new ShrinkingAlgorithm(
			originalSample,
			sampleShrunkConsumer,
			ignore -> {}
		);

		try {
			// Shrunk falsified sample has already been grabbed in sampleShrunkConsumer
			FalsifiedSample ignore = plainShrinker.shrink(recreatingFalsifier);
		} catch (RecreationDone ignore) {}

		if (recreatingSequence.isEmpty()) {
			return Optional.of(currentBest[0].shrinkables());
		} else {
			return Optional.empty();
		}
	}

	private FalsifiedSample createFalsifiedSample() {
		List