1 files changed, 6 insertions, 2 deletions

diff --git a/workflow-designer-be/src/test/java/org/onap/sdc/workflow/TestUtil.java b/workflow-designer-be/src/test/java/org/onap/sdc/workflow/TestUtil.java
index 0a8355c9..0eb5ee77 100644
--- a/workflow-designer-be/src/test/java/org/onap/sdc/workflow/TestUtil.java
+++ b/workflow-designer-be/src/test/java/org/onap/sdc/workflow/TestUtil.java
@@ -3,22 +3,25 @@ package org.onap.sdc.workflow;
 import static org.onap.sdc.workflow.services.impl.ItemType.WORKFLOW;
 
 import org.onap.sdc.workflow.services.types.Workflow;
+import org.onap.sdc.workflow.services.types.ArchivingStatus;
 import org.openecomp.sdc.versioning.types.Item;
+import org.openecomp.sdc.versioning.types.ItemStatus;
 
 public class TestUtil {
 
-    public static Workflow createWorkflow(int workflowNum, boolean createId) {
+    public static Workflow createWorkflow(int workflowNum, boolean createId, ArchivingStatus archivingStatus) {
         Workflow workflow = new Workflow();
         if (createId) {
             workflow.setId(String.valueOf(workflowNum));
         }
         workflow.setName("Workflow_" + workflowNum);
         workflow.setDescription("Description_" + workflowNum);
+        workflow.setArchiving(archivingStatus);
 
         return workflow;
     }
 
-    public static Item createItem(int itemNum, boolean setType, boolean setId) {
+    public static Item createItem(int itemNum, boolean setType, boolean setId, ItemStatus archivingStatus) {
         Item item = new Item();
         if (setId) {
             item.setId(String.valueOf(itemNum));
@@ -28,6 +31,7 @@ public class TestUtil {
         if (setType) {
             item.setType(WORKFLOW.name());
         }
+        item.setStatus(archivingStatus);
         return item;
     }