Search in sources :

Example 6 with JavaTask

use of org.ow2.proactive.scheduler.common.task.JavaTask in project scheduling by ow2-proactive.

the class Job2XMLTransformerTest method checkTaskVariables.

@Test
public void checkTaskVariables() throws Exception {
    File xmlFile = tmpFolder.newFile();
    Map<String, TaskVariable> variablesMap = new HashMap<>();
    TaskVariable taskVariable1 = new TaskVariable();
    taskVariable1.setName("name1");
    taskVariable1.setValue("value1");
    taskVariable1.setJobInherited(false);
    taskVariable1.setModel("model1");
    TaskVariable taskVariable2 = new TaskVariable();
    taskVariable2.setName("name2");
    taskVariable2.setValue("value2");
    taskVariable2.setJobInherited(true);
    taskVariable2.setModel("model2");
    variablesMap.put(taskVariable1.getName(), taskVariable1);
    variablesMap.put(taskVariable2.getName(), taskVariable2);
    TaskFlowJob job = new TaskFlowJob();
    JavaTask task = new JavaTask();
    task.setName("task");
    task.setExecutableClassName("oo.Bar");
    task.setVariables(variablesMap);
    job.addTask(task);
    new Job2XMLTransformer().job2xmlFile(job, xmlFile);
    TaskFlowJob recreatedJob = (TaskFlowJob) (JobFactory.getFactory().createJob(xmlFile.getAbsolutePath()));
    Map<String, TaskVariable> resVariables = recreatedJob.getTask("task").getVariables();
    assertEquals(2, resVariables.size());
    assertEquals(taskVariable1, resVariables.get("name1"));
    assertEquals(taskVariable2, resVariables.get("name2"));
}
Also used : HashMap(java.util.HashMap) TaskFlowJob(org.ow2.proactive.scheduler.common.job.TaskFlowJob) TaskVariable(org.ow2.proactive.scheduler.common.task.TaskVariable) JavaTask(org.ow2.proactive.scheduler.common.task.JavaTask) File(java.io.File) Test(org.junit.Test)

Example 7 with JavaTask

use of org.ow2.proactive.scheduler.common.task.JavaTask in project scheduling by ow2-proactive.

the class Job2XMLTransformerTest method walltimeIsPreserved.

@Test
public void walltimeIsPreserved() throws Exception {
    File xmlFile = tmpFolder.newFile();
    String taskName = "walltimeTask";
    // tests for various values including one second, one minute, one minute and one second, big value, etc. miliseconds are discarded
    long[] walltimesToTest = { 0, 1000, 60000, 61000, 3600000, 3601000, 3660000, 3661000, 999999000 };
    for (int i = 1; i < walltimesToTest.length; i++) {
        TaskFlowJob job = new TaskFlowJob();
        JavaTask task = new JavaTask();
        task.setName(taskName);
        task.setExecutableClassName("oo.Bar");
        task.setWallTime(walltimesToTest[i]);
        job.addTask(task);
        new Job2XMLTransformer().job2xmlFile(job, xmlFile);
        TaskFlowJob recreatedJob = (TaskFlowJob) (JobFactory.getFactory().createJob(xmlFile.getAbsolutePath()));
        assertEquals("Walltimes between original and recreated job must be equal", walltimesToTest[i], recreatedJob.getTask(taskName).getWallTime());
    }
}
Also used : TaskFlowJob(org.ow2.proactive.scheduler.common.job.TaskFlowJob) JavaTask(org.ow2.proactive.scheduler.common.task.JavaTask) File(java.io.File) Test(org.junit.Test)

Example 8 with JavaTask

use of org.ow2.proactive.scheduler.common.task.JavaTask in project scheduling by ow2-proactive.

the class Job2XMLTransformerTest method forkEnvironmentIsPreserved.

@Test
public void forkEnvironmentIsPreserved() throws Exception {
    File xmlFile = tmpFolder.newFile();
    TaskFlowJob job = new TaskFlowJob();
    JavaTask task = new JavaTask();
    task.setName("forkedTask");
    task.setExecutableClassName("oo.Bar");
    task.setForkEnvironment(new ForkEnvironment());
    job.addTask(task);
    new Job2XMLTransformer().job2xmlFile(job, xmlFile);
    TaskFlowJob recreatedJob = (TaskFlowJob) (JobFactory.getFactory().createJob(xmlFile.getAbsolutePath()));
    assertNotNull(recreatedJob.getTask("forkedTask").getForkEnvironment());
}
Also used : TaskFlowJob(org.ow2.proactive.scheduler.common.job.TaskFlowJob) JavaTask(org.ow2.proactive.scheduler.common.task.JavaTask) ForkEnvironment(org.ow2.proactive.scheduler.common.task.ForkEnvironment) File(java.io.File) Test(org.junit.Test)

Example 9 with JavaTask

use of org.ow2.proactive.scheduler.common.task.JavaTask in project scheduling by ow2-proactive.

the class TestJobFactory method run_.

private void run_(String impl) throws Throwable {
    String scriptFolder = new File(getClass().getResource("scripts/").toURI()).getAbsolutePath();
    System.setProperty("jobName", "Job_TaskFlow");
    log("Test Job TASK_FLOW");
    TaskFlowJob tfJob = getJob(jobTaskFlowDescriptor, impl, scriptFolder);
    // Check job properties
    assertEquals("No paquit in its HostName.", tfJob.getDescription());
    assertEquals("Job_TaskFlow", tfJob.getName());
    assertEquals("My_project", tfJob.getProjectName());
    assertEquals(JobPriority.HIGH, tfJob.getPriority());
    assertEquals(OnTaskError.CANCEL_JOB, tfJob.getOnTaskErrorProperty().getValue());
    assertEquals(2, tfJob.getMaxNumberOfExecution());
    assertEquals(RestartMode.ELSEWHERE, tfJob.getRestartTaskOnError());
    assertEquals(JobType.TASKSFLOW, tfJob.getType());
    assertEquals(4, tfJob.getTasks().size());
    assertEquals("input/space", tfJob.getInputSpace());
    assertEquals("output/space", tfJob.getOutputSpace());
    // Check task 1 properties
    Task task1 = tfJob.getTask("task1");
    assertEquals("task1", task1.getName());
    // The job factory overwrites the task settings with the job settings if they are set to none.
    assertEquals(OnTaskError.NONE, task1.getOnTaskErrorProperty().getValue());
    assertEquals(false, task1.isPreciousResult());
    assertEquals(RestartMode.ANYWHERE, task1.getRestartTaskOnError());
    assertEquals(1, task1.getMaxNumberOfExecution());
    assertEquals("Parallel Tasks - Task 1", task1.getDescription());
    assertEquals(2, task1.getSelectionScripts().size());
    assertEquals(false, task1.getSelectionScripts().get(0).isDynamic());
    Assert.assertTrue(task1.getSelectionScripts().get(0).getScript() != null);
    assertEquals(1, task1.getSelectionScripts().get(0).getParameters().length);
    assertEquals("paquit", task1.getSelectionScripts().get(0).getParameters()[0]);
    assertEquals(true, task1.getSelectionScripts().get(1).isDynamic());
    Assert.assertTrue(task1.getSelectionScripts().get(1).getScript() != null);
    assertEquals(2, task1.getSelectionScripts().get(1).getParameters().length);
    assertEquals("test1", task1.getSelectionScripts().get(1).getParameters()[0]);
    assertEquals("test2", task1.getSelectionScripts().get(1).getParameters()[1]);
    Assert.assertTrue(task1.getPreScript().getScript().contains("Beginning of Pre-Script"));
    Assert.assertTrue(task1.getPostScript().getScript().contains("Content is equals to " + scriptFolder + "/unset.js"));
    assertNull(task1.getPostScript().getParameters());
    Assert.assertTrue(task1.getCleaningScript().getScript().contains("Beginning of clean script"));
    assertNull(task1.getCleaningScript().getParameters());
    assertEquals(null, task1.getDependencesList());
    assertEquals(1, task1.getNumberOfNodesNeeded());
    assertEquals(12 * 1000, task1.getWallTime());
    assertEquals(true, task1.isWallTimeSet());
    assertEquals(0, task1.getGenericInformation().size());
    assertNull(task1.getInputFilesList());
    assertNull(task1.getOutputFilesList());
    assertEquals("1", ((JavaTask) task1).getArgument("sleepTime"));
    assertEquals("1", ((JavaTask) task1).getArgument("number"));
    assertEquals("org.ow2.proactive.scheduler.examples.WaitAndPrint", ((JavaTask) task1).getExecutableClassName());
    assertEquals(true, ((JavaTask) task1).isFork());
    assertEquals(null, task1.getForkEnvironment());
    // Check task 2 properties
    Task task2 = tfJob.getTask("task2");
    assertEquals("task2", task2.getName());
    assertEquals(OnTaskError.NONE, task2.getOnTaskErrorProperty().getValue());
    // the following commented check fails, it is what we expect, because replacement is done in the internal factory.
    // it avoids complex changes during user job creation process.
    // this property is tested in the TestDatabaseCRUD
    // Assert.assertEquals(tfJob.getTask("task2").isCancelJobOnError(),true);
    assertEquals(false, task2.isPreciousResult());
    assertEquals(RestartMode.ELSEWHERE, task2.getRestartTaskOnError());
    assertEquals(2, task2.getMaxNumberOfExecution());
    assertEquals("Parallel Tasks - Task 2", task2.getDescription());
    assertEquals(null, task2.getSelectionScripts());
    Assert.assertTrue(task2.getPreScript().getScript().contains("Beginning of Pre-Script"));
    assertEquals(null, task2.getPostScript());
    assertEquals(null, task2.getCleaningScript());
    assertEquals(null, task2.getDependencesList());
    assertEquals(1, task2.getNumberOfNodesNeeded());
    assertEquals(0, task2.getWallTime());
    assertEquals(false, task2.isWallTimeSet());
    assertEquals(0, task2.getGenericInformation().size());
    Assert.assertTrue(task2.getInputFilesList().get(0).getInputFiles().getIncludes().contains("tata*"));
    Assert.assertTrue(task2.getInputFilesList().get(0).getInputFiles().getExcludes().contains("tata*1"));
    assertEquals(InputAccessMode.TransferFromInputSpace, task2.getInputFilesList().get(0).getMode());
    Assert.assertTrue(task2.getInputFilesList().get(1).getInputFiles().getIncludes().contains("toto*.txt"));
    Assert.assertTrue(task2.getInputFilesList().get(1).getInputFiles().getExcludes().contains("toto*2.txt"));
    assertEquals(InputAccessMode.TransferFromOutputSpace, task2.getInputFilesList().get(1).getMode());
    Assert.assertTrue(task2.getOutputFilesList().get(0).getOutputFiles().getIncludes().contains("titi*"));
    Assert.assertTrue(task2.getOutputFilesList().get(0).getOutputFiles().getExcludes().contains("titi*1"));
    assertEquals(OutputAccessMode.TransferToOutputSpace, task2.getOutputFilesList().get(0).getMode());
    Assert.assertTrue(task2.getOutputFilesList().get(1).getOutputFiles().getIncludes().contains("titi*.txt"));
    Assert.assertTrue(task2.getOutputFilesList().get(1).getOutputFiles().getExcludes().contains("titi*3.txt"));
    assertEquals(OutputAccessMode.TransferToOutputSpace, task2.getOutputFilesList().get(1).getMode());
    assertEquals("12", ((JavaTask) task2).getArgument("sleepTime"));
    assertEquals("21", ((JavaTask) task2).getArgument("number"));
    assertEquals("/bin/java/jdk1.5", ((JavaTask) task2).getArgument("test"));
    assertEquals("org.ow2.proactive.scheduler.examples.WaitAndPrint", ((JavaTask) task2).getExecutableClassName());
    assertEquals(true, ((JavaTask) task2).isFork());
    assertEquals(false, task2.isWallTimeSet());
    assertEquals("/bin/java/jdk1.5", task2.getForkEnvironment().getJavaHome());
    assertEquals("/bin/java/jdk1.5/toto", task2.getForkEnvironment().getWorkingDir());
    assertEquals("-dparam=12", task2.getForkEnvironment().getJVMArguments().get(0));
    assertEquals("-djhome=/bin/java/jdk1.5", task2.getForkEnvironment().getJVMArguments().get(1));
    Map<String, String> props = task2.getForkEnvironment().getSystemEnvironment();
    assertEquals(2, props.size());
    assertEquals("ioi", props.get("toto"));
    assertEquals("456", props.get("tata"));
    assertEquals("ioi", task2.getForkEnvironment().getSystemEnvironmentVariable("toto"));
    assertEquals("456", task2.getForkEnvironment().getSystemEnvironmentVariable("tata"));
    List<String> addcp = task2.getForkEnvironment().getAdditionalClasspath();
    assertEquals(2, addcp.size());
    assertEquals("a", addcp.get(0));
    assertEquals("b", addcp.get(1));
    Assert.assertNotNull(task2.getForkEnvironment().getEnvScript());
    // Check task 3 properties
    Task task3 = tfJob.getTask("task3");
    assertEquals("task3", task3.getName());
    assertEquals(OnTaskError.NONE, task3.getOnTaskErrorProperty().getValue());
    // the following commented check fails, it is what we expect, because replacement is done in the
    // internal factory.
    // it avoids complex changes during user job creation process.
    // this property is tested in the TestDatabaseCRUD
    // Assert.assertEquals(tfJob.getTask("task3").isCancelJobOnError(),true);
    assertEquals(false, task3.isPreciousResult());
    // the following commented check fails, it is what we expect, because replacement is done in the
    // internal factory.
    // it avoids complex changes during user job creation process.
    // this property is tested in the TestDatabaseCRUD
    // Assert.assertEquals(tfJob.getTask("task3").getRestartTaskOnError(),RestartMode.ELSEWHERE);
    // the following commented check fails, it is what we expect, because replacement is done in the internal factory.
    // it avoids complex changes during user job creation process.
    // this property is tested in the TestDatabaseCRUD
    // Assert.assertEquals(tfJob.getTask("task3").getMaxNumberOfExecution(),2);
    assertEquals("Dependent Tasks - Task 3", task3.getDescription());
    assertEquals(null, task3.getSelectionScripts());
    assertEquals(null, task3.getPreScript());
    Assert.assertTrue(task3.getPostScript().getScript().contains("Unsetting system property user.property1"));
    assertNull(task3.getPostScript().getParameters());
    assertEquals(null, task3.getCleaningScript());
    assertEquals(2, task3.getDependencesList().size());
    assertEquals("task1", task3.getDependencesList().get(0).getName());
    assertEquals("task2", task3.getDependencesList().get(1).getName());
    assertEquals(10 * 60 * 1000 + 53 * 1000, task3.getWallTime());
    assertEquals(true, task3.isWallTimeSet());
    assertEquals(0, task3.getGenericInformation().size());
    assertEquals(1, task3.getInputFilesList().size());
    Assert.assertTrue(task3.getInputFilesList().get(0).getInputFiles().getIncludes().contains("tata*"));
    Assert.assertTrue(task3.getInputFilesList().get(0).getInputFiles().getExcludes().isEmpty());
    assertEquals(InputAccessMode.none, task3.getInputFilesList().get(0).getMode());
    assertNull(task3.getOutputFilesList());
    assertEquals(5, ((NativeTask) task3).getCommandLine().length);
    assertEquals("1", ((NativeTask) task3).getCommandLine()[1]);
    assertEquals("2 2", ((NativeTask) task3).getCommandLine()[2]);
    assertEquals("3", ((NativeTask) task3).getCommandLine()[3]);
    assertEquals("12", ((NativeTask) task3).getCommandLine()[4]);
    assertNull(((NativeTask) task3).getWorkingDir());
    assertEquals(3, task3.getNumberOfNodesNeeded());
    // Check task 4 properties
    Task task4 = tfJob.getTask("task4");
    assertEquals("task4", task4.getName());
    assertEquals(OnTaskError.CANCEL_JOB, task4.getOnTaskErrorProperty().getValue());
    assertEquals(true, task4.isPreciousResult());
    assertEquals(RestartMode.ANYWHERE, task4.getRestartTaskOnError());
    assertEquals(3, task4.getMaxNumberOfExecution());
    assertEquals(null, task4.getDescription());
    assertEquals(null, task4.getSelectionScripts());
    assertEquals(null, task4.getPreScript());
    assertEquals(null, task4.getPostScript());
    assertEquals(null, task4.getCleaningScript());
    assertEquals(1, task4.getDependencesList().size());
    assertEquals("task3", task4.getDependencesList().get(0).getName());
    assertEquals(0, task4.getWallTime());
    assertEquals(false, task4.isWallTimeSet());
    assertEquals("v11", task4.getGenericInformation().get("n11"));
    assertEquals("v22", task4.getGenericInformation().get("n22"));
    assertNull(task4.getInputFilesList());
    assertEquals(5, task4.getOutputFilesList().size());
    Assert.assertTrue(task4.getOutputFilesList().get(0).getOutputFiles().getIncludes().contains("a"));
    Assert.assertTrue(task4.getOutputFilesList().get(1).getOutputFiles().getIncludes().contains("b"));
    Assert.assertTrue(task4.getOutputFilesList().get(2).getOutputFiles().getIncludes().contains("c"));
    Assert.assertTrue(task4.getOutputFilesList().get(3).getOutputFiles().getIncludes().contains("d"));
    Assert.assertTrue(task4.getOutputFilesList().get(4).getOutputFiles().getIncludes().contains("e"));
    Assert.assertTrue(task4.getOutputFilesList().get(0).getOutputFiles().getExcludes().contains("f"));
    Assert.assertTrue(task4.getOutputFilesList().get(1).getOutputFiles().getExcludes().contains("g"));
    Assert.assertTrue(task4.getOutputFilesList().get(2).getOutputFiles().getExcludes().isEmpty());
    Assert.assertTrue(task4.getOutputFilesList().get(3).getOutputFiles().getExcludes().contains("h"));
    Assert.assertTrue(task4.getOutputFilesList().get(4).getOutputFiles().getExcludes().contains("i"));
    assertEquals(OutputAccessMode.TransferToOutputSpace, task4.getOutputFilesList().get(0).getMode());
    assertEquals(OutputAccessMode.none, task4.getOutputFilesList().get(1).getMode());
    assertEquals(OutputAccessMode.none, task4.getOutputFilesList().get(2).getMode());
    assertEquals(OutputAccessMode.TransferToOutputSpace, task4.getOutputFilesList().get(3).getMode());
    assertEquals(OutputAccessMode.none, task4.getOutputFilesList().get(4).getMode());
    assertEquals(null, ((NativeTask) task4).getWorkingDir());
    assertEquals(10, task4.getNumberOfNodesNeeded());
    log("Test Job MULTI_NODES");
    TaskFlowJob mnJob = getJob(jobMultiNodesDescriptor, impl, scriptFolder);
    // Check job properties
    assertEquals("No description", mnJob.getDescription());
    assertEquals("job_multiNodes", mnJob.getName());
    assertEquals(JobPriority.LOW, mnJob.getPriority());
    assertEquals(OnTaskError.NONE, mnJob.getOnTaskErrorProperty().getValue());
    assertEquals(1, mnJob.getMaxNumberOfExecution());
    assertEquals(RestartMode.ANYWHERE, mnJob.getRestartTaskOnError());
    assertEquals(JobType.TASKSFLOW, mnJob.getType());
    assertEquals("v1", mnJob.getGenericInformation().get("n1"));
    assertEquals("v2", mnJob.getGenericInformation().get("n2"));
    // Check task properties
    JavaTask jt = (JavaTask) mnJob.getTask("Controller");
    assertEquals("100", jt.getArgument("numberToFind"));
    assertEquals(OnTaskError.NONE, jt.getOnTaskErrorProperty().getValue());
    assertEquals(null, jt.getCleaningScript());
    assertEquals(null, jt.getDependencesList());
    assertEquals("Will control the workers in order to find the prime number", jt.getDescription());
    assertEquals("org.ow2.proactive.scheduler.examples.MultiNodeExample", jt.getExecutableClassName());
    assertEquals("v11", jt.getGenericInformation().get("n11"));
    assertEquals("v22", jt.getGenericInformation().get("n22"));
    assertEquals(1, jt.getMaxNumberOfExecution());
    assertEquals("Controller", jt.getName());
    assertEquals(3, jt.getNumberOfNodesNeeded());
    assertEquals(null, jt.getPreScript());
    assertEquals(null, jt.getPostScript());
    assertEquals(RestartMode.ANYWHERE, jt.getRestartTaskOnError());
    assertEquals(null, jt.getSelectionScripts());
    Assert.assertTrue(jt.isPreciousResult());
    log("Test generated task name");
    TaskFlowJob job = new TaskFlowJob();
    for (int i = 0; i < 4; i++) {
        Task t = new NativeTask();
        job.addTask(t);
    }
    Task t1, t2, t3, t4;
    assertNull(job.getTask(SchedulerConstants.TASK_NAME_IFNOTSET + "0"));
    Assert.assertNotNull(t1 = job.getTask(SchedulerConstants.TASK_NAME_IFNOTSET + "1"));
    Assert.assertNotNull(t2 = job.getTask(SchedulerConstants.TASK_NAME_IFNOTSET + "2"));
    Assert.assertNotNull(t3 = job.getTask(SchedulerConstants.TASK_NAME_IFNOTSET + "3"));
    Assert.assertNotNull(t4 = job.getTask(SchedulerConstants.TASK_NAME_IFNOTSET + "4"));
    assertNull(job.getTask(SchedulerConstants.TASK_NAME_IFNOTSET + "5"));
    Assert.assertTrue(t1 != t2);
    Assert.assertTrue(t1 != t3);
    Assert.assertTrue(t1 != t4);
    Assert.assertTrue(t2 != t3);
    Assert.assertTrue(t2 != t4);
    Assert.assertTrue(t3 != t4);
}
Also used : Task(org.ow2.proactive.scheduler.common.task.Task) JavaTask(org.ow2.proactive.scheduler.common.task.JavaTask) NativeTask(org.ow2.proactive.scheduler.common.task.NativeTask) TaskFlowJob(org.ow2.proactive.scheduler.common.job.TaskFlowJob) JavaTask(org.ow2.proactive.scheduler.common.task.JavaTask) NativeTask(org.ow2.proactive.scheduler.common.task.NativeTask) File(java.io.File)

Example 10 with JavaTask

use of org.ow2.proactive.scheduler.common.task.JavaTask in project scheduling by ow2-proactive.

the class TestStaxJobFactory method testJobCreationAttributeOrderDefinitionParameterXmlElement.

@Test
public void testJobCreationAttributeOrderDefinitionParameterXmlElement() throws URISyntaxException, JobCreationException, IOException, ClassNotFoundException {
    TaskFlowJob job = (TaskFlowJob) factory.createJob(getResource("job_attr_def_parameter_xml_element.xml"));
    Map<String, Serializable> arguments = ((JavaTask) job.getTask("task")).getArguments();
    assertExpectedKeyValueEntriesMatch(arguments);
}
Also used : Serializable(java.io.Serializable) TaskFlowJob(org.ow2.proactive.scheduler.common.job.TaskFlowJob) JavaTask(org.ow2.proactive.scheduler.common.task.JavaTask) Test(org.junit.Test)

Aggregations

JavaTask (org.ow2.proactive.scheduler.common.task.JavaTask)96 TaskFlowJob (org.ow2.proactive.scheduler.common.job.TaskFlowJob)92 Test (org.junit.Test)40 ForkEnvironment (org.ow2.proactive.scheduler.common.task.ForkEnvironment)22 InternalJob (org.ow2.proactive.scheduler.job.InternalJob)18 InternalTask (org.ow2.proactive.scheduler.task.internal.InternalTask)17 JobId (org.ow2.proactive.scheduler.common.job.JobId)14 NativeTask (org.ow2.proactive.scheduler.common.task.NativeTask)12 EmptyTask (org.ow2.proactive.scheduler.examples.EmptyTask)11 SelectionScript (org.ow2.proactive.scripting.SelectionScript)10 SimpleScript (org.ow2.proactive.scripting.SimpleScript)10 TaskResultImpl (org.ow2.proactive.scheduler.task.TaskResultImpl)9 File (java.io.File)8 HashMap (java.util.HashMap)7 Task (org.ow2.proactive.scheduler.common.task.Task)7 JobResult (org.ow2.proactive.scheduler.common.job.JobResult)6 EmptyExecutable (functionaltests.executables.EmptyExecutable)5 JobDescriptor (org.ow2.proactive.scheduler.common.JobDescriptor)5 Scheduler (org.ow2.proactive.scheduler.common.Scheduler)5 TaskDescriptor (org.ow2.proactive.scheduler.common.TaskDescriptor)4