package org.embulk.filter;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import mockit.Mocked;
import mockit.NonStrictExpectations;
import mockit.Verifications;

import org.embulk.spi.Exec;
import org.junit.Test;
import org.slf4j.Logger;

public class TestSpeedometerSpeedAggregator {
    @Mocked SpeedometerSpeedController controller;
    @Mocked Exec exec;

    @Test
    public void testGetInstance() {
        assertNotNull("Verify there is a singleton.", SpeedometerSpeedAggregator.getInstance());
    }

    @Test
    public void testSpeedometerSpeedAggregator() {
        SpeedometerSpeedAggregator aggregator = new SpeedometerSpeedAggregator();
        assertEquals("Verify the default global start time is zero.", 0, aggregator.getGlobalStartTime());
        assertEquals("Verify the default active count is zero.", 0, aggregator.getActiveControllerCount());
    }

    @Test
    public void testStartController() {
        SpeedometerSpeedAggregator aggregator = new SpeedometerSpeedAggregator();
        long nowTime = System.currentTimeMillis();
        aggregator.startController(controller, nowTime);
        assertEquals("Verify global start time is set.", nowTime, aggregator.getGlobalStartTime());
        assertEquals("Verify active controller count.", 1, aggregator.getActiveControllerCount());
        assertTrue("Verify there is a registered controller.", aggregator.getControllerList().contains(controller));
    }

    @Test
    public void testStopController() {
        new NonStrictExpectations() {{
            controller.getTotalBytes(); result = 11;
        }};

        SpeedometerSpeedAggregator aggregator = new SpeedometerSpeedAggregator();
        long nowTime = System.currentTimeMillis();
        aggregator.startController(controller, nowTime);
        aggregator.stopController(controller);
        assertEquals("Verify active controller count.", 0, aggregator.getActiveControllerCount());
        assertFalse("Verify there is a registered controller.", aggregator.getControllerList().contains(controller));
        assertEquals("Verify controller's total bytes are added to aggregator.", 11, aggregator.getGlobalTotalBytes());
    }

    @Test
    public void testStopControllerShowOverallMessage(@Mocked final Logger logger) {
        new NonStrictExpectations() {{
            Exec.getLogger(SpeedometerFilterPlugin.class); result = logger;
        }};

        SpeedometerSpeedAggregator aggregator = new SpeedometerSpeedAggregator();
        long nowTime = System.currentTimeMillis();
        aggregator.startController(controller, nowTime);
        aggregator.startController(controller, nowTime);
        aggregator.stopController(controller);
        aggregator.stopController(controller);

        new Verifications() {{
            logger.info(withAny("Overall message.")); times = 2;
        }};
   }

    @Test
    public void testGetSpeedLimitForController() {
        new NonStrictExpectations() {{
            controller.getSpeedLimit(); result = 10;
        }};

        SpeedometerSpeedAggregator aggregator = new SpeedometerSpeedAggregator();
        long nowTime = System.currentTimeMillis();
        aggregator.startController(controller, nowTime);
        assertEquals("Verify speed limit for one controller.", 10, aggregator.getSpeedLimitForController(controller));
        aggregator.startController(controller, nowTime);
        assertEquals("Verify speed limit for two controllers.", 10 / 2, aggregator.getSpeedLimitForController(controller));
    }

    @Test
    public void testCheckProgress() throws Exception {
        SpeedometerSpeedAggregator aggregator = new SpeedometerSpeedAggregator();
        long initPeriodTime = System.currentTimeMillis();
        final long startPeriodTime = initPeriodTime + 2;
        aggregator.startController(controller, initPeriodTime);
        aggregator.checkProgress(initPeriodTime, 1);
        aggregator.checkProgress(startPeriodTime, 1);
        aggregator.checkProgress(startPeriodTime, 1); // This call should not affect anything.

        new Verifications() {{
            controller.renewPeriod(); times = 1;
            controller.getTotalBytes(); times = 1;
            controller.getPeriodBytesPerSec(startPeriodTime); times = 1;
        }};
    }

    @Test
    public void testCheckProgressIsDisabled() throws Exception {
        SpeedometerSpeedAggregator aggregator = new SpeedometerSpeedAggregator();
        long initPeriodTime = System.currentTimeMillis();
        final long startPeriodTime = initPeriodTime + 2;
        aggregator.startController(controller, initPeriodTime);
        aggregator.checkProgress(initPeriodTime, 0);
        aggregator.checkProgress(startPeriodTime, 0);

        new Verifications() {{
            controller.renewPeriod(); times = 0;
            controller.getTotalBytes(); times = 0;
            controller.getPeriodBytesPerSec(startPeriodTime); times = 0;
        }};
    }
}