497 lines
17 KiB
Java
497 lines
17 KiB
Java
/**
|
|
*
|
|
*/
|
|
package org.gcube.vremanagement.executor.pluginmanager;
|
|
|
|
import java.util.Calendar;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
import java.util.UUID;
|
|
|
|
import org.acme.HelloWorldPlugin;
|
|
import org.acme.HelloWorldPluginDeclaration;
|
|
import org.gcube.vremanagement.executor.api.types.LaunchParameter;
|
|
import org.gcube.vremanagement.executor.api.types.Scheduling;
|
|
import org.gcube.vremanagement.executor.exception.PluginStateNotRetrievedException;
|
|
import org.gcube.vremanagement.executor.exception.UnableToInterruptTaskException;
|
|
import org.gcube.vremanagement.executor.persistence.SmartExecutorPersistenceConnector;
|
|
import org.gcube.vremanagement.executor.plugin.PluginState;
|
|
import org.gcube.vremanagement.executor.plugin.PluginStateEvolution;
|
|
import org.gcube.vremanagement.executor.scheduler.SmartExecutorScheduler;
|
|
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) http://www.lucafrosini.com/
|
|
*/
|
|
public class SmartExecutorSchedulerTest {
|
|
|
|
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(HelloWorldPluginDeclaration.NAME, inputs);
|
|
parameter.setScheduling(scheduling);
|
|
|
|
SmartExecutorScheduler smartExecutorScheduler = SmartExecutorScheduler.getInstance();
|
|
UUID uuid = smartExecutorScheduler.schedule(parameter);
|
|
logger.debug("Scheduled Job with ID {}", uuid);
|
|
|
|
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.getLastPluginInstanceState(uuid);
|
|
Assert.assertEquals(PluginState.DONE, pluginStateEvolution.getPluginState());
|
|
}
|
|
|
|
@Test
|
|
public void earlyStopTest() throws Exception {
|
|
SmartExecutorScheduler smartExecutorScheduler = SmartExecutorScheduler.getInstance();
|
|
UUID uuid = scheduleTest(null, null);
|
|
try {
|
|
smartExecutorScheduler.stop(uuid, true, 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(PluginStateNotRetrievedException e){
|
|
// OK
|
|
logger.error("PluginStateNotRetrievedException this can be acceptable in some tests", e);
|
|
}
|
|
|
|
}
|
|
|
|
@Test
|
|
public void middleStopTest() throws Exception {
|
|
UUID uuid = scheduleTest(null, null);
|
|
SmartExecutorScheduler smartExecutorScheduler = SmartExecutorScheduler.getInstance();
|
|
|
|
long startTime = Calendar.getInstance().getTimeInMillis();
|
|
long endTime = startTime;
|
|
while(endTime <= (startTime + 2000)){
|
|
endTime = Calendar.getInstance().getTimeInMillis();
|
|
}
|
|
|
|
smartExecutorScheduler.stop(uuid, true, false);
|
|
|
|
startTime = Calendar.getInstance().getTimeInMillis();
|
|
endTime = startTime;
|
|
while(endTime <= (startTime + 10000)){
|
|
endTime = Calendar.getInstance().getTimeInMillis();
|
|
}
|
|
PluginStateEvolution pluginStateEvolution = pc.getLastPluginInstanceState(uuid);
|
|
Assert.assertEquals(PluginState.STOPPED, pluginStateEvolution.getPluginState());
|
|
}
|
|
|
|
@Test
|
|
public void lateStopTest() throws Exception {
|
|
UUID uuid = scheduleTest(null, null);
|
|
SmartExecutorScheduler smartExecutorScheduler = SmartExecutorScheduler.getInstance();
|
|
long startTime = Calendar.getInstance().getTimeInMillis();
|
|
long endTime = startTime;
|
|
while(endTime <= (startTime + 12000)){
|
|
endTime = Calendar.getInstance().getTimeInMillis();
|
|
}
|
|
smartExecutorScheduler.stop(uuid, true, false);
|
|
|
|
PluginStateEvolution pluginStateEvolution = pc.getLastPluginInstanceState(uuid);
|
|
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 = SmartExecutorScheduler.getInstance();
|
|
long startTime = Calendar.getInstance().getTimeInMillis();
|
|
long endTime = startTime;
|
|
while(endTime <= (startTime + 2000)){
|
|
endTime = Calendar.getInstance().getTimeInMillis();
|
|
}
|
|
smartExecutorScheduler.stop(first, true, false);
|
|
|
|
startTime = Calendar.getInstance().getTimeInMillis();
|
|
endTime = startTime;
|
|
while(endTime <= (startTime + 12000)){
|
|
endTime = Calendar.getInstance().getTimeInMillis();
|
|
}
|
|
|
|
PluginStateEvolution pluginStateEvolution = pc.getLastPluginInstanceState(first);
|
|
Assert.assertEquals(PluginState.STOPPED, pluginStateEvolution.getPluginState());
|
|
|
|
pluginStateEvolution = pc.getLastPluginInstanceState(second);
|
|
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 = SmartExecutorScheduler.getInstance();
|
|
long startTime = Calendar.getInstance().getTimeInMillis();
|
|
long endTime = startTime;
|
|
while(endTime <= (startTime + 83 * 1000)){
|
|
endTime = Calendar.getInstance().getTimeInMillis();
|
|
}
|
|
smartExecutorScheduler.stop(uuid, true, 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 = SmartExecutorScheduler.getInstance();
|
|
long startTime = Calendar.getInstance().getTimeInMillis();
|
|
long endTime = startTime;
|
|
while(endTime <= (startTime + 80 * 1000)){
|
|
endTime = Calendar.getInstance().getTimeInMillis();
|
|
}
|
|
smartExecutorScheduler.stop(uuid, true, 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 = SmartExecutorScheduler.getInstance();
|
|
long startTime = Calendar.getInstance().getTimeInMillis();
|
|
long endTime = startTime;
|
|
while(endTime <= (startTime + 1.5 * 60 * 1000)){
|
|
endTime = Calendar.getInstance().getTimeInMillis();
|
|
}
|
|
smartExecutorScheduler.stop(uuid, true, 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=PluginStateNotRetrievedException.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=PluginStateNotRetrievedException.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 = SmartExecutorScheduler.getInstance();
|
|
long startTime = Calendar.getInstance().getTimeInMillis();
|
|
long endTime = startTime;
|
|
while(endTime <= (startTime + 1.5 * 60 * 1000)){
|
|
endTime = Calendar.getInstance().getTimeInMillis();
|
|
}
|
|
smartExecutorScheduler.stop(uuid, true, 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(PluginStateNotRetrievedException 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 = SmartExecutorScheduler.getInstance();
|
|
long startTime = Calendar.getInstance().getTimeInMillis();
|
|
long endTime = startTime;
|
|
while(endTime <= (startTime + 1.5 * 60 * 1000)){
|
|
endTime = Calendar.getInstance().getTimeInMillis();
|
|
}
|
|
smartExecutorScheduler.stop(uuid, true, 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(PluginStateNotRetrievedException 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 = SmartExecutorScheduler.getInstance();
|
|
long startTime = Calendar.getInstance().getTimeInMillis();
|
|
long endTime = startTime;
|
|
while(endTime <= (startTime + 1.5 * 60 * 1000)){
|
|
endTime = Calendar.getInstance().getTimeInMillis();
|
|
}
|
|
smartExecutorScheduler.stop(uuid, true, 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(PluginStateNotRetrievedException e){
|
|
// OK
|
|
}
|
|
}
|
|
|
|
@Test(expected=PluginStateNotRetrievedException.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=PluginStateNotRetrievedException.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);
|
|
|
|
}
|
|
|
|
}
|