/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.oodt.cas.workflow.engine;
//JDK imports
import java.io.BufferedReader;
import java.io.File;
import java.util.Arrays;
import java.util.Calendar;
import java.util.Date;
//APACHE imports
import org.apache.commons.io.FileUtils;
//OODT imports
import org.apache.oodt.cas.metadata.Metadata;
import org.apache.oodt.cas.workflow.engine.processor.TaskProcessor;
import org.apache.oodt.cas.workflow.engine.runner.AsynchronousLocalEngineRunner;
import org.apache.oodt.cas.workflow.structs.Priority;
import org.apache.oodt.commons.date.DateUtils;
import org.apache.oodt.commons.util.DateConvert;
//JODA imports
import org.joda.time.DateTime;
import org.joda.time.Seconds;
//Junit imports
import junit.framework.TestCase;
/**
*
* A test case for the {@link AsynchronousLocalEngineRunner}.
*
* @author mattmann
* @version $Revision$
*
*/
public class TestAsynchronousLocalEngineRunner extends TestCase {
private AsynchronousLocalEngineRunner runner;
protected File testDir;
private QuerierAndRunnerUtils utils;
public void testRun() {
TaskProcessor taskProcessor1 = null;
TaskProcessor taskProcessor2 = null;
try {
taskProcessor1 = (TaskProcessor) utils.getProcessor(Priority.getDefault()
.getValue(), "Executing", "running");
taskProcessor2 = (TaskProcessor) utils.getProcessor(Priority.getDefault()
.getValue(), "Executing", "running");
} catch (Exception e) {
fail(e.getMessage());
}
Metadata met = new Metadata();
met.addMetadata("StartDateTime", DateUtils.toString(Calendar.getInstance()));
taskProcessor1.getWorkflowInstance().getSharedContext().addMetadata(met);
taskProcessor2.getWorkflowInstance().getSharedContext().addMetadata(met);
try {
runner.execute(taskProcessor1);
runner.execute(taskProcessor2);
assertTrue(ranFast());
} catch (Exception e) {
e.printStackTrace();
fail(e.getMessage());
}
}
private boolean ranFast() {
boolean ranFast = true;
int jobNum = 1;
for (File f : this.testDir.listFiles()) {
BufferedReader br = null;
try {
String line = FileUtils.readFileToString(f);
String[] toks = line.split(",");
assertEquals("Toks not equal to 2: toks=[" + Arrays.asList(toks) + "]",
2, toks.length);
Date dateTime = DateConvert.isoParse(toks[1]);
Seconds seconds = Seconds.secondsBetween(new DateTime(dateTime),
new DateTime());
if (seconds.getSeconds() > 30) {
fail("More than 30 seconds elapsed now and running job " + jobNum
+ ": seconds elapsed: [" + seconds.getSeconds() + "]");
}
} catch (Exception e) {
e.printStackTrace();
fail(e.getMessage());
ranFast = false;
} finally {
if (br != null) {
try {
br.close();
} catch (Exception ignore) {
}
br = null;
}
}
}
return ranFast;
}
/*
* (non-Javadoc)
*
* @see junit.framework.TestCase#setUp()
*/
@Override
protected void setUp() throws Exception {
String parentPath = File.createTempFile("test", "txt").getParentFile()
.getAbsolutePath();
parentPath = parentPath.endsWith("/") ? parentPath : parentPath + "/";
String testJobDirPath = parentPath + "jobs";
testDir = new File(testJobDirPath);
testDir.mkdirs();
this.runner = new AsynchronousLocalEngineRunner();
this.utils = new QuerierAndRunnerUtils();
}
/*
* (non-Javadoc)
*
* @see junit.framework.TestCase#tearDown()
*/
@Override
protected void tearDown() throws Exception {
// blow away test file
deleteAllFiles(testDir.getAbsolutePath());
testDir.delete();
testDir = null;
this.runner = null;
}
private void deleteAllFiles(String startDir) {
File startDirFile = new File(startDir);
File[] delFiles = startDirFile.listFiles();
if (delFiles != null && delFiles.length > 0) {
for (int i = 0; i < delFiles.length; i++) {
delFiles[i].delete();
}
}
startDirFile.delete();
}
}