fix: get context using either collectionInstrumentId or campaignId

Author: loichenninger

src/main/java/fr/insee/genesis/domain/service/context/DataProcessingContextService.java

@@ -18,7 +18,6 @@
 
 import java.time.LocalDateTime;
 import java.util.ArrayList;
-import java.util.Arrays;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Set;
@@ -149,26 +148,38 @@ public DataProcessingContextModel getContext(String interrogationId) throws Gene
             throw new GenesisException(404,"No interrogation in database with id %s".formatted(interrogationId));
         }
         Set<String> partitionIds = new HashSet<>();
+        Set<String> campaignIds = new HashSet<>();
+        Set<String> collectionInstrumentIds = new HashSet<>();
+
         for (SurveyUnitModel su : surveyUnitModels){
             if (su.getCampaignId()!=null){
-                partitionIds.add(su.getCampaignId());
-                break;
+                campaignIds.add(su.getCampaignId());
+            }
+            if (su.getCollectionInstrumentId()!=null){
+                collectionInstrumentIds.add(su.getCampaignId());
             }
-            partitionIds.add(su.getCollectionInstrumentId());
         }
-        if(partitionIds.isEmpty()){
+        if(campaignIds.size() > 1 || collectionInstrumentIds.size()>1){
+            throw new GenesisException(500,"Multiple partitions for interrogation %s".formatted(interrogationId));
+        }
+
+        if(campaignIds.isEmpty() && collectionInstrumentIds.isEmpty()){
             return null;
         }
-        if(partitionIds.size() > 1){
-            throw new GenesisException(500,"Multiple partitions for interrogation %s %n%s".formatted(
-                    interrogationId,
-                    Arrays.toString(partitionIds.toArray())
-            ));
+
+        DataProcessingContextModel contextModel = new DataProcessingContextModel();
+        if (!collectionInstrumentIds.isEmpty()){
+            contextModel = DataProcessingContextMapper.INSTANCE.documentToModel(
+                    dataProcessingContextPersistancePort.findByPartitionId(collectionInstrumentIds.stream().toList().getFirst())
+            );
         }
 
-        return DataProcessingContextMapper.INSTANCE.documentToModel(
-                dataProcessingContextPersistancePort.findByPartitionId(partitionIds.stream().toList().getFirst())
-        );
+        if (contextModel.getId()==null && !campaignIds.isEmpty()) {
+            contextModel = DataProcessingContextMapper.INSTANCE.documentToModel(
+                    dataProcessingContextPersistancePort.findByPartitionId(campaignIds.stream().toList().getFirst()));
+        }
+
+        return contextModel;
     }
 
     @Override

src/test/java/fr/insee/genesis/domain/service/context/DataProcessingContextServiceTest.java

@@ -209,7 +209,7 @@ void getContext_shouldThrow500IfMultiplePartitions() {
         //To ensure test is portable on Unix/Linux/macOS and windows systems
         String normalizedMessage = ex.getMessage().replaceAll("\\r?\\n", "");
         Assertions.assertThat(ex.getStatus()).isEqualTo(500);
-        Assertions.assertThat(normalizedMessage).isEqualTo("Multiple partitions for interrogation 00001 [CAMPAIGN2, CAMPAIGN1]");
+        Assertions.assertThat(normalizedMessage).isEqualTo("Multiple partitions for interrogation 00001");
     }
 
     @Test