TestH2HistoryChartOptions.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;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;

@AxisRange(max = 1)
@BenchmarkMethodChart(filePrefix = "methods-CLASSNAME")
@BenchmarkHistoryChart(filePrefix = "class-global")
public class TestH2HistoryChartOptions
{
    static final File dbFile = new File(RepeatedTestSlave.class.getName());
    static final File dbFileFull = new File(dbFile.getName() + ".h2.db");
    static final File chartsDir = new File("tmp-subdir");

    private static H2Consumer h2consumer;

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

    @BeforeClass
    public static void checkFile() throws SQLException
    {
        Common.deleteDir(chartsDir);
        chartsDir.mkdir();

        h2consumer = new H2Consumer(dbFile, chartsDir, null);
    }

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

    @AxisRange(max = 2)
    @BenchmarkHistoryChart(filePrefix = "method-level")
    @Test
    @BenchmarkOptions(perfUnits = 10)
    public void testMethodB() throws Exception
    {
        Thread.sleep(new Random().nextInt(30));
    }

    @BenchmarkHistoryChart(filePrefix = "method-level")
    @Test
    public void testMethodC() throws Exception
    {
        Thread.sleep(new Random().nextInt(30));
    }

    @BenchmarkHistoryChart(filePrefix = "testMethodD-CLASSNAME")
    @Test
    public void testMethodD() throws Exception
    {
        Thread.sleep(new Random().nextInt(30));
    }

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

        String c1 = Common.getAndDelete(new File(chartsDir, "class-global" + ".json"));
        assertTrue(c1.contains("testMethodA"));
        assertTrue(c1.contains("testMethodB"));
        assertTrue(c1.contains("testMethodC"));
        c1 = Common.getAndDelete(new File(chartsDir, "class-global" + ".html"));
        assertTrue(c1.contains("max:"));
        assertFalse(c1.contains("min:"));

        String c2 = Common.getAndDelete(new File(chartsDir, "method-level" + ".json"));
        assertFalse(c2.contains("testMethodA"));
        assertTrue(c2.contains("testMethodB"));
        assertTrue(c2.contains("testMethodC"));
        c2 = Common.getAndDelete(new File(chartsDir, "method-level" + ".html"));
        assertTrue(c2.contains("max: 2"));
        assertFalse(c2.contains("min:"));

        Common.existsAndDelete(
            new File(chartsDir, "methods-" + TestH2HistoryChartOptions.class.getName() + ".html"));
        Common.existsAndDelete(
            new File(chartsDir, "methods-" + TestH2HistoryChartOptions.class.getName() + ".json"));

        Common.existsAndDelete(
            new File(chartsDir, "testMethodD-" + TestH2HistoryChartOptions.class.getName() + ".html"));
        Common.existsAndDelete(
            new File(chartsDir, "testMethodD-" + TestH2HistoryChartOptions.class.getName() + ".json"));

        assertTrue(Common.deleteDir(chartsDir));
    }
}