fix a bug affecting duplication of non indexed cases

Signed-off-by: jamal-khey <[email protected]>
powsybl · Aug 21, 2024 · 256056c · 256056c
1 parent b9d7b67
commit 256056c
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 4 deletions.
diff --git a/src/main/java/com/powsybl/caseserver/CaseService.java b/src/main/java/com/powsybl/caseserver/CaseService.java
@@ -53,6 +53,7 @@
 import java.util.zip.ZipOutputStream;
 
 import static com.powsybl.caseserver.CaseException.createDirectoryNotFound;
+import static com.powsybl.caseserver.dto.CaseInfos.*;
 
 /**
  * @author Abdelsalem Hedhili <abdelsalem.hedhili at rte-france.com>
@@ -230,12 +231,14 @@ UUID duplicateCase(UUID sourceCaseUuid, boolean withExpiration) {
             newCaseFile = newCaseUuidDirectory.resolve(existingCaseFile.getFileName());
             Files.copy(existingCaseFile, newCaseFile, StandardCopyOption.COPY_ATTRIBUTES);
 
-            CaseInfos existingCaseInfos = caseInfosService.getCaseInfosByUuid(sourceCaseUuid.toString()).orElseThrow();
-            CaseInfos caseInfos = createInfos(existingCaseInfos.getName(), newCaseUuid, existingCaseInfos.getFormat());
-            caseInfosService.addCaseInfos(caseInfos);
-
             CaseMetadataEntity existingCase = getCaseMetaDataEntity(sourceCaseUuid);
+            CaseInfos caseInfos = createInfos(newCaseFile.getFileName().toString(), newCaseUuid, getFormat(newCaseFile));
+            if (existingCase.isIndexed()) {
+                caseInfosService.addCaseInfos(caseInfos);
+            }
+
             createCaseMetadataEntity(newCaseUuid, withExpiration, existingCase.isIndexed());
+
             sendImportMessage(caseInfos.createMessage());
             return newCaseUuid;
 

diff --git a/src/test/java/com/powsybl/caseserver/CaseControllerTest.java b/src/test/java/com/powsybl/caseserver/CaseControllerTest.java
@@ -474,6 +474,30 @@ public void test() throws Exception {
         assertTrue(response.contains("\"format\":\"XIIDM\""));
     }
 
+    @Test
+    public void testDuplicateNonIndexedCase() throws Exception {
+        // create the storage dir
+        createStorageDir();
+
+        // import IIDM test case
+        String caseUuid = mvc.perform(multipart("/v1/cases")
+                        .file(createMockMultipartFile(TEST_CASE))
+                        .param("withExpiration", "false")
+                        .param("withIndexation", "false"))
+                .andExpect(status().isOk())
+                .andReturn().getResponse().getContentAsString();
+        assertNotNull(outputDestination.receive(1000, caseImportDestination));
+        //duplicate an existing case
+        String duplicateCaseStr = mvc.perform(post("/v1/cases?duplicateFrom=" + caseUuid.substring(1, caseUuid.length() - 1)))
+                .andExpect(status().isOk())
+                .andReturn().getResponse().getContentAsString();
+        UUID duplicateCaseUuid = UUID.fromString(duplicateCaseStr.substring(1, duplicateCaseStr.length() - 1));
+        assertNotNull(outputDestination.receive(1000, caseImportDestination));
+
+        assertFalse(caseMetadataRepository.findById(duplicateCaseUuid).get().isIndexed());
+
+    }
+
     private UUID importCase(String testCase, Boolean withExpiration) throws Exception {
         String importedCase;
         if (withExpiration) {