/**
 * 
 */
package org.gcube.vremanagement.executor.pluginmanager;

import java.util.Calendar;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;

import org.gcube.vremanagement.executor.ContextTest;
import org.gcube.vremanagement.executor.api.types.LaunchParameter;
import org.gcube.vremanagement.executor.api.types.Scheduling;
import org.gcube.vremanagement.executor.exception.PluginInstanceNotFoundException;
import org.gcube.vremanagement.executor.exception.UnableToInterruptTaskException;
import org.gcube.vremanagement.executor.persistence.SmartExecutorPersistenceConnector;
import org.gcube.vremanagement.executor.persistence.SmartExecutorPersistenceFactory;
import org.gcube.vremanagement.executor.plugin.PluginState;
import org.gcube.vremanagement.executor.plugin.PluginStateEvolution;
import org.gcube.vremanagement.executor.scheduler.SmartExecutorScheduler;
import org.gcube.vremanagement.executor.scheduler.SmartExecutorSchedulerFactory;
import org.gcube.vremanagement.helloworld.HelloWorldPlugin;
import org.junit.Assert;
import org.junit.Test;
import org.quartz.CronExpression;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author Luca Frosini (ISTI - CNR)
 */
public class SmartExecutorSchedulerTest extends ContextTest {

	private static Logger logger = LoggerFactory.getLogger(SmartExecutorSchedulerTest.class);

	public static final String START = "START";
	public static final String END = "END";

	public static SmartExecutorPersistenceConnector pc;

	public UUID scheduleTest(Scheduling scheduling, Long sleepTime) throws Exception {
		Map<String, Object> inputs = new HashMap<String, Object>();
		if (sleepTime == null) {
			sleepTime = new Long(10 * 1000); // 10 sec = 10 * 1000 millisec
		}
		inputs.put(HelloWorldPlugin.SLEEP_TIME, sleepTime);
		inputs.put("Test UUID", UUID.randomUUID());
		logger.debug("Inputs : {}", inputs);

		LaunchParameter parameter = new LaunchParameter(new HelloWorldPlugin().getName(), inputs);
		parameter.setScheduling(scheduling);

		SmartExecutorScheduler smartExecutorScheduler = SmartExecutorSchedulerFactory.getSmartExecutorScheduler();
		UUID uuid = smartExecutorScheduler.schedule(parameter, null);
		logger.debug("Scheduled Job with ID {}", uuid);

		pc = SmartExecutorPersistenceFactory.getPersistenceConnector();

		return uuid;
	}

	/*
	 * DeprecatedHelloWorldPlugin dependency needed
	 * 
	 * @Test public void deprecatedConstructorTest() throws Exception {
	 * Map<String, Object> inputs = new HashMap<String, Object>(); Long
	 * sleepTime = new Long(10*1000); // 10 sec = 10 * 1000 millisec
	 * inputs.put(DeprecatedHelloWorldPlugin.SLEEP_TIME, sleepTime);
	 * inputs.put("Test UUID", UUID.randomUUID()); logger.debug("Inputs : {}",
	 * inputs);
	 * 
	 * LaunchParameter parameter = new
	 * LaunchParameter(DeprecatedHelloWorldPluginDeclaration.class.newInstance()
	 * , inputs); parameter.setScheduling(null);
	 * 
	 * SmartExecutorScheduler smartExecutorScheduler =
	 * SmartExecutorScheduler.getInstance(); UUID uuid =
	 * smartExecutorScheduler.schedule(parameter);
	 * logger.debug("Scheduled Job with ID {}", uuid);
	 * 
	 * long startTime = Calendar.getInstance().getTimeInMillis(); long endTime =
	 * startTime; while(endTime <= (startTime + 12000)){ endTime =
	 * Calendar.getInstance().getTimeInMillis(); } PluginState pluginState =
	 * pc.getLastPluginInstanceState(uuid);
	 * Assert.assertEquals(PluginState.DONE, pluginState); }
	 */

	@Test
	public void schedulingTest() throws Exception {
		UUID uuid = scheduleTest(null, null);
		long startTime = Calendar.getInstance().getTimeInMillis();
		long endTime = startTime;
		while (endTime <= (startTime + 12000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}
		PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, null);
		Assert.assertEquals(PluginState.DONE, pluginStateEvolution.getPluginState());
	}

	@Test
	public void earlyStopTest() throws Exception {
		SmartExecutorScheduler smartExecutorScheduler = SmartExecutorSchedulerFactory.getSmartExecutorScheduler();
		UUID uuid = scheduleTest(null, null);
		try {
			smartExecutorScheduler.stop(uuid, false);
		} catch (UnableToInterruptTaskException e) {
			logger.error("UnableToInterruptTaskException this is the normal behaviour.", e);
			return;
		}

		long startTime = Calendar.getInstance().getTimeInMillis();
		long endTime = startTime;
		while (endTime <= (startTime + 12000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}

		try {
			PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, 5);
			Assert.assertEquals(PluginState.STOPPED, pluginStateEvolution.getPluginState());
		} catch (PluginInstanceNotFoundException e) {
			// OK
			logger.error("PluginInstanceNotFoundException this can be acceptable in some tests", e);
		}

	}

	@Test
	public void middleStopTest() throws Exception {
		UUID uuid = scheduleTest(null, null);
		SmartExecutorScheduler smartExecutorScheduler = SmartExecutorSchedulerFactory.getSmartExecutorScheduler();

		long startTime = Calendar.getInstance().getTimeInMillis();
		long endTime = startTime;
		while (endTime <= (startTime + 2000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}

		smartExecutorScheduler.stop(uuid, false);

		startTime = Calendar.getInstance().getTimeInMillis();
		endTime = startTime;
		while (endTime <= (startTime + 10000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}

		PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, null);
		Assert.assertEquals(PluginState.STOPPED, pluginStateEvolution.getPluginState());
	}

	@Test
	public void lateStopTest() throws Exception {
		UUID uuid = scheduleTest(null, null);
		SmartExecutorScheduler smartExecutorScheduler = SmartExecutorSchedulerFactory.getSmartExecutorScheduler();
		long startTime = Calendar.getInstance().getTimeInMillis();
		long endTime = startTime;
		while (endTime <= (startTime + 12000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}
		smartExecutorScheduler.stop(uuid, false);

		PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, null);
		Assert.assertEquals(PluginState.DONE, pluginStateEvolution.getPluginState());
	}

	@Test
	public void doubleLaunchfirstStoppedSchedulingTest() throws Exception {
		UUID first = scheduleTest(null, null);
		logger.debug("First scheduled id {}", first);
		UUID second = scheduleTest(null, null);
		logger.debug("Second scheduled id {}", second);
		SmartExecutorScheduler smartExecutorScheduler = SmartExecutorSchedulerFactory.getSmartExecutorScheduler();
		long startTime = Calendar.getInstance().getTimeInMillis();
		long endTime = startTime;
		while (endTime <= (startTime + 2000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}
		smartExecutorScheduler.stop(first, false);

		startTime = Calendar.getInstance().getTimeInMillis();
		endTime = startTime;
		while (endTime <= (startTime + 12000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}

		PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(first, null);
		Assert.assertEquals(PluginState.STOPPED, pluginStateEvolution.getPluginState());

		pluginStateEvolution = pc.getPluginInstanceState(second, null);
		Assert.assertEquals(PluginState.DONE, pluginStateEvolution.getPluginState());
	}

	@Test
	public void delayed() throws Exception {
		Scheduling scheduling = new Scheduling(20);
		UUID uuid = scheduleTest(scheduling, new Long(10 * 1000));

		SmartExecutorScheduler smartExecutorScheduler = SmartExecutorSchedulerFactory.getSmartExecutorScheduler();
		long startTime = Calendar.getInstance().getTimeInMillis();
		long endTime = startTime;
		while (endTime <= (startTime + 83 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}
		smartExecutorScheduler.stop(uuid, false);

		startTime = Calendar.getInstance().getTimeInMillis();
		endTime = startTime;
		while (endTime <= (startTime + 30 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}

		for (int i = 1; i < 5; i++) {
			PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, i);
			Assert.assertEquals(PluginState.DONE, pluginStateEvolution.getPluginState());
		}

		PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, 5);
		Assert.assertEquals(PluginState.STOPPED, pluginStateEvolution.getPluginState());

	}

	@Test
	public void delayedPreviousMustBeTerminated() throws Exception {
		Scheduling scheduling = new Scheduling(20, true);
		UUID uuid = scheduleTest(scheduling, new Long(22 * 1000));

		SmartExecutorScheduler smartExecutorScheduler = SmartExecutorSchedulerFactory.getSmartExecutorScheduler();
		long startTime = Calendar.getInstance().getTimeInMillis();
		long endTime = startTime;
		while (endTime <= (startTime + 80 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}
		smartExecutorScheduler.stop(uuid, false);

		startTime = Calendar.getInstance().getTimeInMillis();
		endTime = startTime;
		while (endTime <= (startTime + 30 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}

		for (int i = 1; i < 5; i++) {
			PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, i);
			if (i % 2 != 0) {
				Assert.assertEquals(PluginState.DONE, pluginStateEvolution.getPluginState());
			} else {
				Assert.assertEquals(PluginState.DISCARDED, pluginStateEvolution.getPluginState());
			}
		}

		PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, 5);
		Assert.assertEquals(PluginState.STOPPED, pluginStateEvolution.getPluginState());

	}

	@Test
	public void delayedAllPreviousMustBeTerminated() throws Exception {
		Scheduling scheduling = new Scheduling(20, true);
		UUID uuid = scheduleTest(scheduling, new Long(45 * 1000));

		SmartExecutorScheduler smartExecutorScheduler = SmartExecutorSchedulerFactory.getSmartExecutorScheduler();
		long startTime = Calendar.getInstance().getTimeInMillis();
		long endTime = startTime;
		while (endTime <= (startTime + 1.5 * 60 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}
		smartExecutorScheduler.stop(uuid, false);

		startTime = Calendar.getInstance().getTimeInMillis();
		endTime = startTime;
		while (endTime <= (startTime + 30 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}

		PluginState[] expectedStates = new PluginState[] { PluginState.DONE, PluginState.DISCARDED,
				PluginState.DISCARDED, PluginState.STOPPED };

		for (int i = 0; i < expectedStates.length; i++) {
			PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, i + 1);
			Assert.assertEquals(expectedStates[i], pluginStateEvolution.getPluginState());
		}

		PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, 5);
		Assert.assertEquals(PluginState.DISCARDED, pluginStateEvolution.getPluginState());

	}

	@Test(expected = PluginInstanceNotFoundException.class)
	public void delayedExpMaxtimes() throws Exception {
		Scheduling scheduling = new Scheduling(20, 3);
		UUID uuid = scheduleTest(scheduling, new Long(10 * 1000));

		long startTime = Calendar.getInstance().getTimeInMillis();
		long endTime = startTime;
		while (endTime <= (startTime + 1.5 * 60 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}

		for (int i = 0; i < 3; i++) {
			PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, i + 1);
			Assert.assertEquals(PluginState.DONE, pluginStateEvolution.getPluginState());
		}

		pc.getPluginInstanceState(uuid, 4);

	}

	@Test(expected = PluginInstanceNotFoundException.class)
	public void delayedExpTimeLimits() throws Exception {

		Calendar firstStartTime = Calendar.getInstance();

		Calendar stopTime = Calendar.getInstance();
		stopTime.setTimeInMillis(firstStartTime.getTimeInMillis());
		stopTime.add(Calendar.SECOND, 10);
		firstStartTime.add(Calendar.SECOND, 5);

		Scheduling scheduling = new Scheduling(20, 0, firstStartTime, stopTime);
		logger.debug("Scheduling : {}", scheduling);

		UUID uuid = scheduleTest(scheduling, new Long(10 * 1000));

		long startTime = Calendar.getInstance().getTimeInMillis();
		long endTime = startTime;
		while (endTime <= (startTime + 30 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}

		PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, 1);
		Assert.assertEquals(PluginState.DONE, pluginStateEvolution.getPluginState());

		pc.getPluginInstanceState(uuid, 2);

	}

	@Test
	public void cronExp() throws Exception {
		CronExpression cronExpression = new CronExpression("0/20 * * ? * *");
		Scheduling scheduling = new Scheduling(cronExpression);
		UUID uuid = scheduleTest(scheduling, new Long(10 * 1000));

		SmartExecutorScheduler smartExecutorScheduler = SmartExecutorSchedulerFactory.getSmartExecutorScheduler();
		long startTime = Calendar.getInstance().getTimeInMillis();
		long endTime = startTime;
		while (endTime <= (startTime + 1.5 * 60 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}
		smartExecutorScheduler.stop(uuid, false);

		startTime = Calendar.getInstance().getTimeInMillis();
		endTime = startTime;
		while (endTime <= (startTime + 30 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}

		for (int i = 1; i < 5; i++) {
			PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, i);
			Assert.assertEquals(PluginState.DONE, pluginStateEvolution.getPluginState());
		}

		try {
			PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, 5);
			Assert.assertEquals(PluginState.STOPPED, pluginStateEvolution.getPluginState());
		} catch (PluginInstanceNotFoundException e) {
			// OK
		}
	}

	@Test
	public void cronExpPreviousMustBeTerminated() throws Exception {
		CronExpression cronExpression = new CronExpression("0/20 * * ? * *");
		Scheduling scheduling = new Scheduling(cronExpression, true);
		UUID uuid = scheduleTest(scheduling, new Long(30 * 1000));

		SmartExecutorScheduler smartExecutorScheduler = SmartExecutorSchedulerFactory.getSmartExecutorScheduler();
		long startTime = Calendar.getInstance().getTimeInMillis();
		long endTime = startTime;
		while (endTime <= (startTime + 1.5 * 60 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}
		smartExecutorScheduler.stop(uuid, false);

		startTime = Calendar.getInstance().getTimeInMillis();
		endTime = startTime;
		while (endTime <= (startTime + 30 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}

		for (int i = 1; i < 5; i++) {
			PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, i);
			if (i % 2 != 0) {
				Assert.assertEquals(PluginState.DONE, pluginStateEvolution.getPluginState());
			} else {
				Assert.assertEquals(PluginState.DISCARDED, pluginStateEvolution.getPluginState());
			}
		}
		try {
			PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, 5);
			Assert.assertEquals(PluginState.STOPPED, pluginStateEvolution.getPluginState());
		} catch (PluginInstanceNotFoundException e) {
			// OK
		}

	}

	@Test
	public void cronExpAllPreviousTerminated() throws Exception {
		CronExpression cronExpression = new CronExpression("0/20 * * ? * *");
		Scheduling scheduling = new Scheduling(cronExpression, true);
		UUID uuid = scheduleTest(scheduling, new Long(45 * 1000));

		SmartExecutorScheduler smartExecutorScheduler = SmartExecutorSchedulerFactory.getSmartExecutorScheduler();
		long startTime = Calendar.getInstance().getTimeInMillis();
		long endTime = startTime;
		while (endTime <= (startTime + 1.5 * 60 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}
		smartExecutorScheduler.stop(uuid, false);

		startTime = Calendar.getInstance().getTimeInMillis();
		endTime = startTime;
		while (endTime <= (startTime + 20 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}

		PluginState[] expectedStates = new PluginState[] { PluginState.DONE, PluginState.DISCARDED,
				PluginState.DISCARDED, PluginState.STOPPED };

		for (int i = 0; i < expectedStates.length; i++) {
			PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, i + 1);
			Assert.assertEquals(expectedStates[i], pluginStateEvolution.getPluginState());
		}

		try {
			PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, 5);
			Assert.assertEquals(PluginState.DISCARDED, pluginStateEvolution.getPluginState());
		} catch (PluginInstanceNotFoundException e) {
			// OK
		}
	}

	@Test(expected = PluginInstanceNotFoundException.class)
	public void cronExpMaxtimes() throws Exception {
		CronExpression cronExpression = new CronExpression("0/20 * * ? * *");
		Scheduling scheduling = new Scheduling(cronExpression, 3);
		UUID uuid = scheduleTest(scheduling, new Long(10 * 1000));

		long startTime = Calendar.getInstance().getTimeInMillis();
		long endTime = startTime;
		while (endTime <= (startTime + 1.5 * 60 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}

		for (int i = 0; i < 3; i++) {
			PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, i + 1);
			Assert.assertEquals(PluginState.DONE, pluginStateEvolution.getPluginState());
		}

		pc.getPluginInstanceState(uuid, 4);

	}

	@Test(expected = PluginInstanceNotFoundException.class)
	public void cronExpTimeLimits() throws Exception {
		CronExpression cronExpression = new CronExpression("0/20 * * ? * *");
		Calendar firstStartTime = Calendar.getInstance();

		Calendar stopTime = Calendar.getInstance();
		stopTime.setTimeInMillis(firstStartTime.getTimeInMillis());
		stopTime.add(Calendar.SECOND, 20);

		Scheduling scheduling = new Scheduling(cronExpression, 0, firstStartTime, stopTime);
		logger.debug("Scheduling : {}", scheduling);

		UUID uuid = scheduleTest(scheduling, new Long(10 * 1000));

		long startTime = Calendar.getInstance().getTimeInMillis();
		long endTime = startTime;
		while (endTime <= (startTime + 30 * 1000)) {
			endTime = Calendar.getInstance().getTimeInMillis();
		}

		PluginStateEvolution pluginStateEvolution = pc.getPluginInstanceState(uuid, 1);
		Assert.assertEquals(PluginState.DONE, pluginStateEvolution.getPluginState());

		pc.getPluginInstanceState(uuid, 2);

	}

}