/*************************************************************************
 *                                                                       *
 *  SignServer: The OpenSource Automated Signing Server                  *
 *                                                                       *
 *  This software is free software; you can redistribute it and/or       *
 *  modify it under the terms of the GNU Lesser General Public           *
 *  License as published by the Free Software Foundation; either         *
 *  version 2.1 of the License, or any later version.                    *
 *                                                                       *
 *  See terms of license at gnu.org.                                     *
 *                                                                       *
 *************************************************************************/
package org.signserver.server.statistics.nonpersistent;

import java.util.Calendar;
import java.util.Date;

import org.signserver.common.NonPersistentStatisticsConstants;
import org.signserver.common.StatisticsConstants;
import org.signserver.common.WorkerConfig;
import org.signserver.server.statistics.Event;

import junit.framework.TestCase;

/**
 * TODO: Document me!
 * 
 * @version $Id: SecondStatisticsCollectorTest.java 3375 2013-02-20 11:53:25Z malu9369 $
 */
public class SecondStatisticsCollectorTest extends TestCase {

    @Override
    protected void setUp() throws Exception {
        super.setUp();
    }

    public void testBasics() throws Exception {
        SecondStatisticsCollector mc = genSecondStatisticsCollector(null);

        Calendar currentTime = Calendar.getInstance();
        currentTime.setTimeInMillis(System.currentTimeMillis());

        assertNotNull(mc.genCurrentStartPeriod());
        Calendar currentStartTime = Calendar.getInstance();
        currentStartTime.setTime(mc.genCurrentStartPeriod());
        assertTrue(currentTime.get(Calendar.DAY_OF_MONTH) == currentStartTime.get(Calendar.DAY_OF_MONTH));
        assertTrue(currentTime.get(Calendar.HOUR) == currentStartTime.get(Calendar.HOUR));
        assertTrue(currentTime.get(Calendar.MINUTE) == currentStartTime.get(Calendar.MINUTE));
        assertTrue(currentTime.get(Calendar.SECOND) == currentStartTime.get(Calendar.SECOND));
        assertTrue(currentStartTime.get(Calendar.MILLISECOND) == 0);
        assertNotNull(mc.genCurrentEndPeriod());

        Calendar currentEndTime = Calendar.getInstance();
        currentEndTime.setTime(mc.genCurrentEndPeriod());
        assertTrue(currentTime.get(Calendar.DAY_OF_MONTH) == currentEndTime.get(Calendar.DAY_OF_MONTH));
        assertTrue(currentTime.get(Calendar.HOUR) == currentEndTime.get(Calendar.HOUR));
        assertTrue(currentTime.get(Calendar.MINUTE) == currentEndTime.get(Calendar.MINUTE));
        assertTrue(currentTime.get(Calendar.SECOND) == currentEndTime.get(Calendar.SECOND));
        assertTrue(currentEndTime.get(Calendar.MILLISECOND) == 999);

        assertTrue(mc.getExpireTime() == (Long.parseLong(NonPersistentStatisticsConstants.DEFAULT_SECONDSTATISTICS_EXPIRETIME) * 1000));

        assertTrue(mc.fetchStatistics(StatisticsConstants.QUERYTYPE_ALL, null, null).size() == 0);

        // TODO: these tests fail randomly based on current system time, disable them for now...
        //mc.addEvent(getEvent());
        //mc.addEvent(getEvent());
        //assertEquals(1, mc.fetchStatistics(StatisticsConstants.QUERYTYPE_ALL, null, null).size());

        //Thread.sleep(1100);
        //mc.addEvent(getEvent());
        //assertTrue("" + mc.fetchStatistics(StatisticsConstants.QUERYTYPE_ALL, null, null).size(), mc.fetchStatistics(StatisticsConstants.QUERYTYPE_ALL, null, null).size() == 2);

        //mc.flush();
        //assertTrue(mc.fetchStatistics(StatisticsConstants.QUERYTYPE_ALL, null, null).size() == 0);
    }

    // TODO: investigate whether this test should remain, since it currently fails in some cases depending on time, disable it for now...
//    public void testFifoQueue() throws Exception {
//        SecondStatisticsCollector mc = genSecondStatisticsCollector("3");
//        mc.addEvent(getEvent());
//        Thread.sleep(1050);
//        mc.addEvent(getEvent());
//        Thread.sleep(1050);
//        mc.addEvent(getEvent());
//        assertTrue(mc.fetchStatistics(StatisticsConstants.QUERYTYPE_ALL, null, null).size() == 3);
//        assertTrue("" + mc.fetchStatistics(StatisticsConstants.QUERYTYPE_ALL, new Date(System.currentTimeMillis() - 1000), null).size(), mc.fetchStatistics(StatisticsConstants.QUERYTYPE_ALL, new Date(System.currentTimeMillis() - 1000), null).size() == 1);
//        assertTrue(mc.fetchStatistics(StatisticsConstants.QUERYTYPE_ALL, new Date(System.currentTimeMillis() - 3100), new Date(System.currentTimeMillis() - 1100)).size() > 1);
//        assertTrue(mc.fetchStatistics(StatisticsConstants.QUERYTYPE_ALL, null, new Date(System.currentTimeMillis() - 1000)).size() == 2);
//
//        Thread.sleep(1050);
//        assertTrue(mc.fetchStatistics(StatisticsConstants.QUERYTYPE_ALL, null, null).size() == 2);
//    }

    private SecondStatisticsCollector genSecondStatisticsCollector(String expireTime) {
        SecondStatisticsCollector ret = new SecondStatisticsCollector();
        WorkerConfig config = new WorkerConfig();
        if (expireTime != null) {
            config.setProperty(NonPersistentStatisticsConstants.SECONDSTATISTICS_EXPIRETIME, expireTime);
        }
        ret.init(123, config, null);

        return ret;
    }

    private Event getEvent() throws InterruptedException {
        Event event = new Event(123);
        event.start();
        Thread.sleep(10);
        event.stop();
        return event;
    }
}