TestTaskNotExecuted.java

/*
 * ProActive Parallel Suite(TM):
 * The Open Source library for parallel and distributed
 * Workflows & Scheduling, Orchestration, Cloud Automation
 * and Big Data Analysis on Enterprise Grids & Clouds.
 *
 * Copyright (c) 2007 - 2017 ActiveEon
 * Contact: contact@activeeon.com
 *
 * This library is free software: you can redistribute it and/or
 * modify it under the terms of the GNU Affero General Public License
 * as published by the Free Software Foundation: version 3 of
 * the License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 *
 * If needed, contact us to obtain a release under GPL Version 2 or 3
 * or a different license than the AGPL.
 */
package functionaltests.job.error;

import static functionaltests.utils.SchedulerTHelper.log;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;

import java.io.File;
import java.net.URL;

import org.junit.Assert;
import org.junit.Test;
import org.ow2.proactive.scheduler.common.exception.TaskCouldNotRestartException;
import org.ow2.proactive.scheduler.common.exception.TaskCouldNotStartException;
import org.ow2.proactive.scheduler.common.exception.TaskSkippedException;
import org.ow2.proactive.scheduler.common.job.JobId;
import org.ow2.proactive.scheduler.common.job.JobState;
import org.ow2.proactive.scheduler.common.task.TaskInfo;
import org.ow2.proactive.scheduler.common.task.TaskResult;
import org.ow2.proactive.scheduler.common.task.TaskStatus;

import functionaltests.utils.SchedulerFunctionalTestNoRestart;


public class TestTaskNotExecuted extends SchedulerFunctionalTestNoRestart {

    private static URL jobDescriptor1 = TestTaskNotExecuted.class.getResource("/functionaltests/descriptors/Job_TaskCouldNotStart.xml");

    private static URL jobDescriptor2 = TestTaskNotExecuted.class.getResource("/functionaltests/descriptors/Job_TaskCouldNotRestart.xml");

    private static URL jobDescriptor3 = TestTaskNotExecuted.class.getResource("/functionaltests/descriptors/Job_TaskSkipped.xml");

    private static URL jobDescriptor1_Schema33 = TestTaskNotExecuted.class.getResource("/functionaltests/descriptors/Job_TaskCouldNotStart_Schema33.xml");

    private static URL jobDescriptor2_Schema33 = TestTaskNotExecuted.class.getResource("/functionaltests/descriptors/Job_TaskCouldNotRestart_Schema33.xml");

    private static URL jobDescriptor3_Schema33 = TestTaskNotExecuted.class.getResource("/functionaltests/descriptors/Job_TaskSkipped_Schema33.xml");

    @Test
    public void testTaskNotExecuted() throws Throwable {
        String jobDescriptorPath1 = new File(jobDescriptor1.toURI()).getAbsolutePath();
        String jobDescriptorPath2 = new File(jobDescriptor2.toURI()).getAbsolutePath();
        String jobDescriptorPath3 = new File(jobDescriptor3.toURI()).getAbsolutePath();
        testTaskNotExecuted(jobDescriptorPath1, jobDescriptorPath2, jobDescriptorPath3);
    }

    @Test
    public void testTaskNotExecutedCompatibilitySchema33() throws Throwable {
        String jobDescriptorPath1 = new File(jobDescriptor1_Schema33.toURI()).getAbsolutePath();
        String jobDescriptorPath2 = new File(jobDescriptor2_Schema33.toURI()).getAbsolutePath();
        String jobDescriptorPath3 = new File(jobDescriptor3_Schema33.toURI()).getAbsolutePath();
        testTaskNotExecuted(jobDescriptorPath1, jobDescriptorPath2, jobDescriptorPath3);
    }

    private void testTaskNotExecuted(String jobDescriptorPath1, String jobDescriptorPath2, String jobDescriptorPath3)
            throws Exception {
        log("Submitting job 1");
        JobId id1 = schedulerHelper.submitJob(jobDescriptorPath1);
        log("Wait for event job 1 submitted");
        schedulerHelper.waitForEventJobSubmitted(id1);
        log("Wait for event t1 running");
        schedulerHelper.waitForEventTaskRunning(id1, "t0");
        schedulerHelper.waitForEventTaskFinished(id1, "t0");
        schedulerHelper.waitForEventJobFinished(id1);

        TaskResult tr1 = schedulerHelper.getSchedulerInterface().getTaskResult(id1, "t1");
        assertTrue(tr1.getException() instanceof TaskCouldNotStartException);

        log("Submitting job 2");
        JobId id2 = schedulerHelper.submitJob(jobDescriptorPath2);
        log("Wait for event job 2 submitted");
        schedulerHelper.waitForEventJobSubmitted(id2);
        log("Wait for event t2 running");
        schedulerHelper.waitForEventTaskRunning(id2, "t2");
        log("Restarting task");
        schedulerHelper.getSchedulerInterface().restartTask(id2, "t2", Integer.MAX_VALUE);

        TaskInfo ti2 = schedulerHelper.waitForEventTaskWaitingForRestart(id2, "t2");
        schedulerHelper.getSchedulerInterface().killJob(id2);

        schedulerHelper.waitForEventJobFinished(id2);

        JobState jobState = schedulerHelper.getSchedulerInterface().getJobState(id2);
        Assert.assertEquals(TaskStatus.NOT_RESTARTED, jobState.getTasks().get(0).getStatus());
        TaskResult tr2 = schedulerHelper.getSchedulerInterface().getTaskResult(id2, "t2");
        assertTrue(tr2.getException() instanceof TaskCouldNotRestartException);

        log("Submitting job 3");
        JobId id3 = schedulerHelper.submitJob(jobDescriptorPath3);
        log("Wait for event job 3 submitted");
        schedulerHelper.waitForEventJobSubmitted(id3);
        log("Wait for event T T1 T2 finished");
        schedulerHelper.waitForEventTaskFinished(id3, "T");
        schedulerHelper.waitForEventTaskFinished(id3, "T1");

        TaskResult tr3_1 = schedulerHelper.getSchedulerInterface().getTaskResult(id3, "T1");
        TaskResult tr3_2 = schedulerHelper.getSchedulerInterface().getTaskResult(id3, "T2");

        assertNull(tr3_1.getException());
        assertTrue(tr3_2.getException() instanceof TaskSkippedException);
    }

}