| /* |
| * 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.instrepo; |
| |
| //OODT imports |
| import org.apache.oodt.cas.metadata.Metadata; |
| import org.apache.oodt.cas.workflow.structs.Workflow; |
| import org.apache.oodt.cas.workflow.structs.WorkflowCondition; |
| import org.apache.oodt.cas.workflow.structs.WorkflowInstance; |
| import org.apache.oodt.cas.workflow.structs.WorkflowStatus; |
| import org.apache.oodt.cas.workflow.structs.WorkflowTask; |
| import org.apache.oodt.cas.workflow.structs.WorkflowTaskConfiguration; |
| import org.apache.oodt.cas.workflow.structs.exceptions.InstanceRepositoryException; |
| |
| //JDK imports |
| import java.io.File; |
| import java.io.FileInputStream; |
| import java.io.IOException; |
| import java.util.Iterator; |
| import java.util.List; |
| import java.util.Vector; |
| |
| //Apache Imports |
| import org.apache.commons.io.FileUtils; |
| |
| //Junit imports |
| import junit.framework.TestCase; |
| |
| /** |
| * @author mattmann |
| * @version $Revision$ |
| * @since OODT-55 |
| * |
| * <p> |
| * Test Case for the {@link LuceneWorkflowInstaceRepository} |
| * </p>. |
| */ |
| public class TestLuceneWorkflowInstanceRepository extends TestCase implements |
| WorkflowStatus { |
| |
| private LuceneWorkflowInstanceRepository repo = null; |
| |
| private WorkflowInstance testInst = null; |
| |
| private Workflow testWkflw; |
| |
| private WorkflowTask testTask; |
| |
| private WorkflowCondition testCond; |
| |
| private String tmpDirPath = null; |
| |
| public TestLuceneWorkflowInstanceRepository() { |
| testInst = new WorkflowInstance(); |
| testWkflw = new Workflow(); |
| testTask = new WorkflowTask(); |
| testCond = new WorkflowCondition(); |
| testWkflw.setName("test.workflow"); |
| testWkflw.setId("test.id"); |
| List tasks = new Vector(); |
| List conds = new Vector(); |
| |
| testCond.setConditionId("test.cond.id"); |
| testCond.setConditionInstanceClassName("test.class"); |
| testCond.setConditionName("test.cond.name"); |
| testCond.setOrder(1); |
| conds.add(testCond); |
| |
| testTask.setTaskConfig(new WorkflowTaskConfiguration()); |
| testTask.setTaskId("test.task.id"); |
| testTask.setConditions(conds); |
| testTask.setOrder(1); |
| testTask.setTaskInstanceClassName("test.class"); |
| testTask.setTaskName("test.task.name"); |
| tasks.add(testTask); |
| testWkflw.setTasks(tasks); |
| |
| testInst = new WorkflowInstance(); |
| testInst.setCurrentTaskId("test.task"); |
| testInst.setStatus(STARTED); |
| testInst.setWorkflow(testWkflw); |
| |
| Metadata sharedContext = new Metadata(); |
| sharedContext.addMetadata("TestKey1", "TestVal1"); |
| sharedContext.addMetadata("TestKey1", "TestVal2"); |
| sharedContext.addMetadata("TestKey2", "TestVal3"); |
| testInst.setSharedContext(sharedContext); |
| |
| // first load the example configuration |
| try { |
| System.getProperties().load( |
| new FileInputStream("./src/main/resources/workflow.properties")); |
| } catch (Exception e) { |
| fail(e.getMessage()); |
| } |
| |
| // get a temp directory |
| |
| File tempDir = null; |
| File tempFile = null; |
| |
| try { |
| tempFile = File.createTempFile("foo", "bar"); |
| tempFile.deleteOnExit(); |
| tempDir = tempFile.getParentFile(); |
| } catch (Exception e) { |
| fail(e.getMessage()); |
| } |
| |
| tmpDirPath = tempDir.getAbsolutePath(); |
| if (!tmpDirPath.endsWith("/")) { |
| tmpDirPath += "/"; |
| } |
| |
| tmpDirPath += "testInstRepo/"; |
| |
| // now override the catalog ones |
| System.setProperty( |
| "org.apache.oodt.cas.workflow.instanceRep.lucene.idxPath", |
| tmpDirPath); |
| |
| System.setProperty( |
| "org.apache.oodt.cas.workflow.instanceRep.pageSize", "20"); |
| |
| } |
| |
| /* |
| * (non-Javadoc) |
| * |
| * @see junit.framework.TestCase#setUp() |
| */ |
| protected void setUp() throws Exception { |
| repo = (LuceneWorkflowInstanceRepository) new LuceneWorkflowInstanceRepositoryFactory().createInstanceRepository(); |
| } |
| |
| /* |
| * (non-Javadoc) |
| * |
| * @see junit.framework.TestCase#tearDown() |
| */ |
| protected void tearDown() throws Exception { |
| // now remove the temporary directory used |
| if (tmpDirPath != null) { |
| FileUtils.forceDelete(new File(tmpDirPath)); |
| } |
| if (repo != null) { |
| repo = null; |
| } |
| |
| } |
| |
| /** |
| * @since OODT-389 |
| **/ |
| public void testInstanceRepoInitialization() { |
| // Getting the number of workflow instances should not fail even on an empty index |
| try { |
| int count = repo.getNumWorkflowInstances(); |
| // There should be no instances in the index at this point |
| assertEquals(0, count); |
| } catch (Exception e) { |
| fail(e.getMessage()); |
| } |
| } |
| |
| public void testUpdateDocumentAndPreserveId() { |
| try { |
| repo.addWorkflowInstance(testInst); |
| } catch (InstanceRepositoryException e) { |
| fail(e.getMessage()); |
| } |
| |
| // preserve its id |
| String wInstId = testInst.getId(); |
| |
| // modify it |
| try { |
| repo.updateWorkflowInstance(testInst); |
| } catch (InstanceRepositoryException e) { |
| fail(e.getMessage()); |
| } |
| |
| // make sure that the new id is the same |
| assertEquals(wInstId, testInst.getId()); |
| |
| // make sure that there is only 1 workflow instance |
| List wInsts = null; |
| try { |
| wInsts = repo.getWorkflowInstances(); |
| } catch (InstanceRepositoryException e) { |
| fail(e.getMessage()); |
| } |
| |
| assertNotNull(wInsts); |
| assertEquals(1, wInsts.size()); |
| |
| // make sure that we can look up that workflow inst by its id |
| WorkflowInstance foundInst = null; |
| try { |
| foundInst = repo.getWorkflowInstanceById(wInstId); |
| } catch (InstanceRepositoryException e) { |
| fail(e.getMessage()); |
| } |
| |
| assertNotNull(foundInst); |
| assertEquals(foundInst.getId(), wInstId); |
| assertNotNull(foundInst.getSharedContext()); |
| assertNotNull(foundInst.getSharedContext().getHashtable()); |
| assertEquals(2, foundInst.getSharedContext().getHashtable().keySet() |
| .size()); |
| assertNotNull(foundInst.getSharedContext().getAllMetadata("TestKey1")); |
| assertEquals(2, foundInst.getSharedContext().getAllMetadata("TestKey1") |
| .size()); |
| |
| boolean gotVal1 = false, gotVal2 = false; |
| |
| for (Iterator i = foundInst.getSharedContext().getAllMetadata( |
| "TestKey1").iterator(); i.hasNext();) { |
| String val = (String) i.next(); |
| if (val.equals("TestVal1")) { |
| gotVal1 = true; |
| } else if (val.equals("TestVal2")) { |
| gotVal2 = true; |
| } |
| } |
| |
| assert (gotVal1 && gotVal2); |
| |
| assertNotNull(foundInst.getSharedContext().getMetadata("TestKey2")); |
| assertEquals("TestVal3", foundInst.getSharedContext().getMetadata( |
| "TestKey2")); |
| } |
| |
| } |