RepeatedTestSlave.java

package com.carrotsearch.junitbenchmarks.h2;

import com.carrotsearch.junitbenchmarks.BenchmarkOptions;
import com.carrotsearch.junitbenchmarks.BenchmarkRule;
import com.carrotsearch.junitbenchmarks.Common;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.MethodRule;

import java.io.File;
import java.sql.SQLException;
import java.util.Random;

/**
 * Test H2 consumer's chart generation functionality. 
 */
@BenchmarkHistoryChart(labelWith = LabelType.CUSTOM_KEY)
@BenchmarkOptions(callgc = false, perfDiffToAverage = 10d, perfDiffToLastRun = 10d)
public class RepeatedTestSlave
{
    static boolean cleanup = true; 
    static final File dbFile = new File(RepeatedTestSlave.class.getName());
    static final File dbFileFull = new File(dbFile.getName() + ".h2.db");

    private static H2Consumer h2consumer;

    @Rule
    public MethodRule benchmarkRun = new BenchmarkRule(h2consumer);

    @BeforeClass
    public static void checkFile() throws SQLException
    {
        h2consumer = new H2Consumer(dbFile);
    }

    @Test
    public void testMethodA() throws Exception
    {
        Thread.sleep(new Random().nextInt(20));
    }

    @Test
    public void testMethodC() throws Exception
    {
        Thread.sleep(new Random().nextInt(30));
    }

    @AfterClass
    public static void verify() throws Exception
    {
        h2consumer.close();

        if (cleanup)
        {
            Common.existsAndDelete(RepeatedTestSlave.class.getName() + ".html");
            Common.existsAndDelete(RepeatedTestSlave.class.getName() + ".json");
            dbFileFull.delete();
        }
    }
}