Add integration test for duplicate

Author: BartChris

Kitodo/src/main/java/org/kitodo/production/forms/WorkflowForm.java

@@ -83,12 +83,15 @@ public class WorkflowForm extends BaseForm {
     private Integer roleId;
     private boolean migration;
     private static final String MIGRATION_FORM_PATH = MessageFormat.format(REDIRECT_PATH,"system");
+    private ExternalContext externalContext;
 
     /**
      * Constructor.
      */
     public WorkflowForm() {
         super.setLazyDTOModel(new LazyDTOModel(ServiceManager.getWorkflowService()));
+        this.externalContext = FacesContext.getCurrentInstance() != null
+                ? FacesContext.getCurrentInstance().getExternalContext() : null;
     }
 
     /**
@@ -118,6 +121,10 @@ public void setWorkflowStatus(WorkflowStatus workflowStatus) {
         this.workflowStatus = workflowStatus;
     }
 
+    public void setExternalContext(ExternalContext externalContext) {
+        this.externalContext = externalContext; // Allow test injection
+    }
+
     /**
      * Read XML for file chosen out of the select list.
      */
@@ -420,10 +427,9 @@ public String duplicate(Integer itemId) {
                 this.svgDiagram = IOUtils.toString(svgInputStream, StandardCharsets.UTF_8);
             }
             // Store duplicated workflow in Flash scope
-            ExternalContext externalContext = FacesContext.getCurrentInstance().getExternalContext();
-            externalContext.getFlash().put("duplicatedWorkflow", this.workflow);
-            externalContext.getFlash().put("xmlDiagram", this.xmlDiagram);
-            externalContext.getFlash().put("svgDiagram", this.svgDiagram);
+            storeInFlashScope("duplicatedWorkflow", this.workflow);
+            storeInFlashScope("xmlDiagram", this.xmlDiagram);
+            storeInFlashScope("svgDiagram", this.svgDiagram);
 
             return workflowEditPath + "&id=0";
 
@@ -492,6 +498,12 @@ public void load(int id) {
         }
     }
 
+    private void storeInFlashScope(String key, Object value) {
+        if (externalContext != null) {
+            externalContext.getFlash().put(key, value);
+        }
+    }
+
     /**
      * Get role id.
      *

Kitodo/src/test/java/org/kitodo/production/forms/WorkflowFormIT.java

@@ -11,18 +11,19 @@
 
 package org.kitodo.production.forms;
 
-import static org.junit.jupiter.api.Assertions.assertEquals;
-import static org.junit.jupiter.api.Assertions.assertFalse;
-import static org.junit.jupiter.api.Assertions.assertTrue;
-
+import java.io.File;
 import java.io.IOException;
+import java.net.URI;
+import java.nio.charset.StandardCharsets;
 import java.util.Arrays;
 import java.util.List;
 
+import org.apache.commons.io.FileUtils;
 import org.junit.jupiter.api.AfterAll;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
 import org.kitodo.MockDatabase;
+import org.kitodo.config.ConfigCore;
 import org.kitodo.data.database.beans.DataEditorSetting;
 import org.kitodo.data.database.beans.Task;
 import org.kitodo.data.database.beans.Template;
@@ -35,12 +36,25 @@
 import org.kitodo.production.services.ServiceManager;
 import org.kitodo.production.services.data.DataEditorSettingService;
 import org.kitodo.production.services.data.TaskService;
+import org.mockito.ArgumentCaptor;
+
+import javax.faces.context.ExternalContext;
+import javax.faces.context.FacesContext;
+import javax.faces.context.Flash;
+
+import static org.mockito.Mockito.*;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertTrue;
 
 public class WorkflowFormIT {
 
     private WorkflowForm currentWorkflowForm = new WorkflowForm();
     private static final TaskService taskService = ServiceManager.getTaskService();
     private static final DataEditorSettingService dataEditorSettingService = ServiceManager.getDataEditorSettingService();
+    private ExternalContext mockExternalContext;
+    private Flash mockFlash;
 
     /**
      * Setup Database and start elasticsearch.
@@ -122,6 +136,45 @@ public void shouldUpdateTemplateTasksAndDeleteOnlyAffectedDataEditorSettings() t
         assertEquals(0.6f, dataEditorSettingForTaskOfSecondTemplate.get(0).getGalleryWidth(), 0);
     }
 
+    @Test
+    public void shouldDuplicateWorkflowAndStoreInFlash() throws Exception {
+        // Arrange: Mock ExternalContext and Flash
+        mockExternalContext = mock(ExternalContext.class);
+        mockFlash = mock(Flash.class);
+        currentWorkflowForm.setExternalContext(mockExternalContext);
+
+        when(mockExternalContext.getFlash()).thenReturn(mockFlash); // Ensure Flash scope is returned
+
+        // Act: Call the duplicate method
+        String resultUrl = currentWorkflowForm.duplicate(1);
+
+        //Assert: Ensure redirection to the edit page
+        assertTrue(resultUrl.contains("&id=0"));
+
+        //Capture arguments that were passed to Flash
+        ArgumentCaptor<Workflow> workflowCaptor = ArgumentCaptor.forClass(Workflow.class);
+        ArgumentCaptor<String> xmlCaptor = ArgumentCaptor.forClass(String.class);
+        ArgumentCaptor<String> svgCaptor = ArgumentCaptor.forClass(String.class);
+
+        // ✅ Verify and capture actual method calls
+        verify(mockFlash, times(1)).put(eq("duplicatedWorkflow"), workflowCaptor.capture());
+        verify(mockFlash, times(1)).put(eq("xmlDiagram"), xmlCaptor.capture());
+        verify(mockFlash, times(1)).put(eq("svgDiagram"), svgCaptor.capture());
+
+        // ✅ Assert the captured workflow is not null and has expected values
+        Workflow duplicatedWorkflow = workflowCaptor.getValue();
+        String xmlWorkflow = xmlCaptor.getValue();
+        assertNotNull(duplicatedWorkflow, "Duplicated workflow should not be null");
+        assertEquals("gateway-test1", duplicatedWorkflow.getTitle().replaceFirst("_[^_]*$", ""),
+                "Expected duplicated workflow title");
+
+        String diagramPath = ConfigCore.getKitodoDiagramDirectory() + "gateway-test1" + ".bpmn20.xml";
+        String data = FileUtils.readFileToString(new File(diagramPath), StandardCharsets.UTF_8);
+        assertEquals(data, xmlWorkflow, "Expected duplicated workflow title");
+        assertNotNull(xmlCaptor.getValue(), "XML Diagram should be stored");
+        assertNotNull(svgCaptor.getValue(), "SVG Diagram should be stored");
+    }
+
     private Task createAndSaveTemplateTask(TaskStatus taskStatus, int ordering, Template template) throws DataException {
         Task task = new Task();
         task.setProcessingStatus(taskStatus);