import org.junit.rules.TestRule
import org.junit.runner.Description
import org.junit.runners.model.Statement

class PerformanceTestIdProvider implements TestRule {

    private testSpec
    private String methodName
    private String className

    PerformanceTestIdProvider() {}

    @Override
    Statement apply(Statement base, Description description) {
        methodName = description.methodName

                super.defaultSpec(builder)
                builder.workingDirectory = temporaryFolder.testDirectory
            }
        }
        performanceTestIdProvider.setTestSpec(runner)
    }

    @Rule
    PerformanceTestIdProvider performanceTestIdProvider = new PerformanceTestIdProvider()

    static {
        // TODO - find a better way to cleanup
        System.addShutdownHook {
            ((Closeable) RESULT_STORE).close()

    protected final IntegrationTestBuildContext buildContext = new IntegrationTestBuildContext()

    @Rule
    PerformanceTestIdProvider performanceTestIdProvider = new PerformanceTestIdProvider()

    CrossBuildPerformanceTestRunner runner

    def setup() {
        Assume.assumeFalse(Boolean.getBoolean(CROSS_VERSION_ONLY_PROPERTY_NAME))

    private final IntegrationTestBuildContext buildContext = new IntegrationTestBuildContext()

    private CrossVersionPerformanceTestRunner runner

    @Rule
    PerformanceTestIdProvider performanceTestIdProvider = new PerformanceTestIdProvider()

    def setup() {
        runner = new CrossVersionPerformanceTestRunner(
                new GradleBuildExperimentRunner(gradleProfilerReporter, outputDirSelector),

        assert dir.exists() && dir.isDirectory()
        return dir
    }

    private static File pluginInfoDir = resolvePluginInfoDir()

    @Rule
    PerformanceTestIdProvider performanceTestIdProvider = new PerformanceTestIdProvider()

    @AutoCleanup
    @Shared
    def resultStore = new BuildScanResultsStore()

    protected final IntegrationTestBuildContext buildContext = new IntegrationTestBuildContext()