Added test to Impl

InseeFr · May 3, 2024 · c3fdd46 · c3fdd46
1 parent 6749f82
commit c3fdd46
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/src/test/java/fr/insee/genesis/domain/service/SurveyUnitUpdateImplTest.java b/src/test/java/fr/insee/genesis/domain/service/SurveyUnitUpdateImplTest.java
@@ -11,6 +11,7 @@
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.DisplayName;
 import org.junit.jupiter.api.Test;
+import org.springframework.http.ResponseEntity;
 
 import java.time.LocalDateTime;
 import java.time.Month;
@@ -153,6 +154,14 @@ void findIdUEsByIdQuestionnaireTest(){
         ).isNotEmpty();
     }
 
+    @Test
+    void getQuestionnairesByCampaignTest() {
+        addAdditionnalDtoToMongoStub("TESTQUESTIONNAIRE2");
+
+        Assertions.assertThat(surveyUnitUpdateImplStatic.findIdQuestionnairesByIdCampaign("TESTIDCAMPAIGN")).isNotEmpty().hasSize(2);
+
+    }
+
     private void addAdditionnalDtoToMongoStub(){
         List<VariableDto> externalVariableDtoList = new ArrayList<>();
         VariableDto variableDto = VariableDto.builder().idVar("TESTIDVAR").values(List.of(new String[]{"V1", "V2"})).build();
@@ -176,4 +185,27 @@ private void addAdditionnalDtoToMongoStub(){
         surveyUnitUpdatePersistencePortStub.getMongoStub().add(recentDTO);
     }
 
+    private void addAdditionnalDtoToMongoStub(String idQuestionnaire) {
+        List<VariableDto> externalVariableDtoList = new ArrayList<>();
+        VariableDto variableDto = VariableDto.builder().idVar("TESTIDVAR").values(List.of(new String[]{"V1", "V2"})).build();
+        externalVariableDtoList.add(variableDto);
+
+        List<CollectedVariableDto> collectedVariableDtoList = new ArrayList<>();
+        CollectedVariableDto collectedVariableDto = new CollectedVariableDto("TESTIDVAR", List.of(new String[]{"V1", "V2"}), "TESTIDLOOP", "TESTIDPARENT");
+        collectedVariableDtoList.add(collectedVariableDto);
+
+        SurveyUnitUpdateDto recentDTO = SurveyUnitUpdateDto.builder()
+                .idCampaign("TESTIDCAMPAIGN")
+                .mode(Mode.WEB)
+                .idUE("TESTIDUE")
+                .idQuest(idQuestionnaire)
+                .state(DataState.COLLECTED)
+                .fileDate(LocalDateTime.of(2023, 2, 2, 0, 0, 0))
+                .recordDate(LocalDateTime.of(2024, 2, 2, 0, 0, 0))
+                .externalVariables(externalVariableDtoList)
+                .collectedVariables(collectedVariableDtoList)
+                .build();
+        surveyUnitUpdatePersistencePortStub.getMongoStub().add(recentDTO);
+    }
+
 }