#20840 refactoring the unit test

dotCMS · Jul 19, 2022 · cafe624 · cafe624
1 parent 87975f0
commit cafe624
Showing 1 changed file with 15 additions and 5 deletions.
diff --git a/...c/integration-test/java/com/dotmarketing/portlets/workflows/business/WorkflowAPITest.java b/...c/integration-test/java/com/dotmarketing/portlets/workflows/business/WorkflowAPITest.java
@@ -10,6 +10,7 @@
 import static org.junit.Assert.fail;
 
 import com.dotcms.IntegrationTestBase;
+import com.dotcms.business.WrapInTransaction;
 import com.dotcms.content.elasticsearch.business.event.ContentletCheckinEvent;
 import com.dotcms.contenttype.business.ContentTypeAPIImpl;
 import com.dotcms.contenttype.business.FieldAPI;
@@ -1459,18 +1460,27 @@ public void findAvailableActions() throws Exception {
             final Contentlet testContentleti = testContentlet;
             runNoLicense(()-> {
 
-                CacheLocator.getWorkFlowCache().clearCache();
-                List<WorkflowAction> foundActions_ = APILocator.getWorkflowAPI()
-                        .findAvailableActions(testContentleti, joeContributor);
-                Assert.assertFalse(foundActions_.isEmpty());
-                Assert.assertEquals(WorkflowAPI.SYSTEM_WORKFLOW_ID, foundActions_.get(0).getSchemeId());
+               final List<WorkflowStep> steps = getSteps(testContentleti);
+
+                Assert.assertFalse(steps.isEmpty());
+                Assert.assertEquals(WorkflowAPI.SYSTEM_WORKFLOW_ID, steps.get(0).getSchemeId());
             });
         } finally {
             contentletAPI.destroy(testContentlet, user, false);
         }
 
     }
 
+    @WrapInTransaction
+    private List<WorkflowStep>  getSteps (final Contentlet testContentleti) throws DotDataException {
+
+        final WorkFlowFactory workFlowFactory = FactoryLocator.getWorkFlowFactory();
+        final List<WorkflowScheme> schemes = Arrays.asList(workFlowFactory.findSystemWorkflow()) ;
+        CacheLocator.getWorkFlowCache().clearCache();
+        List<WorkflowStep> steps = FactoryLocator.getWorkFlowFactory().findStepsByContentlet(testContentleti, schemes);
+        return steps;
+    }
+
     /**
      * Test the find findAvailableActions methods
      */