import javax.inject.Inject;

/**
 * Allows configuring test rerun mechanics.
 * <p>
 * This extension is added with the name 'rerun' to all {@link Test} tasks.
 */
public class TestRerunTaskExtension {

    /**
     * The default number of reruns we allow for a test task.
     */
    public static final Integer DEFAULT_MAX_RERUNS = 1;

    /**
     * The name of the extension added to each test task.

        TestRerunTaskExtension extension = test.getExtensions()
            .create(TestRerunTaskExtension.NAME, TestRerunTaskExtension.class, objectFactory);
        test.doFirst(new InitTaskAction(extension));
    }

    private static RerunTestExecuter createRetryTestExecuter(Task task, TestRerunTaskExtension extension) {

import java.util.stream.Collectors;

public final class RerunTestExecuter implements TestExecuter<JvmTestExecutionSpec> {

    private final TestRerunTaskExtension extension;
    private final TestExecuter<JvmTestExecutionSpec> delegate;

    public RerunTestExecuter(TestRerunTaskExtension extension, TestExecuter<JvmTestExecutionSpec> delegate) {
        this.extension = extension;
        this.delegate = delegate;
    }

    @Override