OAM-260: Added not null constraints for units in stock cards, event i…

…tems and inventory items
OpenLMIS · Jul 31, 2024 · a7e1ab6 · a7e1ab6
1 parent 9d71b22
commit a7e1ab6
Show file tree

Hide file tree

Showing 7 changed files with 392 additions and 376 deletions.
diff --git a/...test/java/org/openlmis/stockmanagement/repository/StockCardRepositoryIntegrationTest.java b/...test/java/org/openlmis/stockmanagement/repository/StockCardRepositoryIntegrationTest.java
@@ -74,7 +74,7 @@ CrudRepository<StockCard, UUID> getRepository() {
 
   @Override
   StockCard generateInstance() throws Exception {
-    return generateInstance(randomUUID(), randomUUID(), randomUUID(), randomUUID());
+    return generateInstance(randomUUID(), randomUUID(), randomUUID(), randomUUID(), randomUUID());
   }
 
   private StockCard generateInstance(UUID facility, UUID program, UUID product, UUID lot) {
@@ -221,9 +221,9 @@ public void findByIds() throws Exception {
   public void findByLotIdInShouldOnlyFindMatchingStockCards() throws Exception {
     UUID matchingLotId = UUID.randomUUID();
     stockCard1 = stockCardRepository.save(generateInstance(randomUUID(), randomUUID(), randomUUID(),
-        matchingLotId));
+        matchingLotId, randomUUID()));
     stockCard2 = stockCardRepository.save(generateInstance(randomUUID(), randomUUID(), randomUUID(),
-        matchingLotId));
+        matchingLotId, randomUUID()));
     stockCard3 = stockCardRepository.save(generateInstance());
     stockCard4 = stockCardRepository.save(generateInstance());
 
@@ -236,22 +236,28 @@ public void findByLotIdInShouldOnlyFindMatchingStockCards() throws Exception {
 
   private void initializeStockCardsForMultipleFacilitiesAndPrograms() {
     stockCard1 = stockCardRepository
-        .save(generateInstance(facilityId1, programId1, randomUUID(), randomUUID()));
+        .save(generateInstance(facilityId1, programId1, randomUUID(), randomUUID(), randomUUID()));
     stockCard2 = stockCardRepository
-        .save(generateInstance(facilityId2, programId2, randomUUID(), randomUUID()));
+        .save(generateInstance(facilityId2, programId2, randomUUID(), randomUUID(), randomUUID()));
     stockCard3 = stockCardRepository
-        .save(generateInstance(facilityId2, programId1, randomUUID(), randomUUID()));
+        .save(generateInstance(facilityId2, programId1, randomUUID(), randomUUID(), randomUUID()));
     stockCard4 = stockCardRepository
-        .save(generateInstance(facilityId1, programId2, randomUUID(), randomUUID()));
+        .save(generateInstance(facilityId1, programId2, randomUUID(), randomUUID(), randomUUID()));
     stockCard5 = stockCardRepository
-        .save(generateInstance(randomUUID(), randomUUID(), randomUUID(), randomUUID()));
-
-    stockCardRepository.save(generateInstance(facilityId1, programId1, randomUUID(), randomUUID()));
-    stockCardRepository.save(generateInstance(facilityId2, programId2, randomUUID(), randomUUID()));
-    stockCardRepository.save(generateInstance(facilityId1, programId2, randomUUID(), randomUUID()));
+        .save(generateInstance(randomUUID(), randomUUID(), randomUUID(), randomUUID(),
+            randomUUID()));
+
+    stockCardRepository.save(generateInstance(facilityId1, programId1, randomUUID(), randomUUID(),
+        randomUUID()));
+    stockCardRepository.save(generateInstance(facilityId2, programId2, randomUUID(), randomUUID(),
+        randomUUID()));
+    stockCardRepository.save(generateInstance(facilityId1, programId2, randomUUID(), randomUUID(),
+        randomUUID()));
     stockCardRepository
-        .save(generateInstance(facilityId2, randomUUID(), randomUUID(), randomUUID()));
+        .save(generateInstance(facilityId2, randomUUID(), randomUUID(), randomUUID(),
+            randomUUID()));
     stockCardRepository
-        .save(generateInstance(randomUUID(), programId1, randomUUID(), randomUUID()));
+        .save(generateInstance(randomUUID(), programId1, randomUUID(), randomUUID(),
+            randomUUID()));
   }
 }
diff --git a/...t/java/org/openlmis/stockmanagement/service/StockCardSummariesServiceIntegrationTest.java b/...t/java/org/openlmis/stockmanagement/service/StockCardSummariesServiceIntegrationTest.java
@@ -124,8 +124,8 @@ public void shouldFindExistingStockCards()
     when(programReferenceDataService.findOne(any(UUID.class)))
         .thenReturn(new ProgramDto());
 
-    createStockCard(orderable1Id, randomUUID());
-    createStockCard(orderable3Id, randomUUID());
+    createStockCard(orderable1Id, randomUUID(), randomUUID());
+    createStockCard(orderable3Id, randomUUID(), randomUUID());
 
     //when
     List<StockCardDto> cardDtos = stockCardSummariesService
@@ -159,7 +159,7 @@ public void shouldReturnPageOfStockCards() throws Exception {
     UUID orderableId = randomUUID();
     OrderableDto orderable = createOrderableDto(orderableId, "");
 
-    createStockCard(orderableId, randomUUID());
+    createStockCard(orderableId, randomUUID(), randomUUID());
 
     doReturn(singletonList(orderable)).when(orderableReferenceDataService).findAll();
 
@@ -183,7 +183,7 @@ private OrderableDto createOrderableDto(UUID orderableId, String productName) {
         .build();
   }
 
-  private StockCard createStockCard(UUID orderableId, UUID cardId) {
+  private StockCard createStockCard(UUID orderableId, UUID cardId, UUID unitId) {
     StockEvent stockEvent = stockEventsRepository.save(new StockEventDataBuilder()
         .withoutId().build());
 
@@ -192,6 +192,7 @@ private StockCard createStockCard(UUID orderableId, UUID cardId) {
     stockCard.setFacilityId(facilityId);
     stockCard.setProgramId(programId);
     stockCard.setOriginEvent(stockEvent);
+    stockCard.setUnitOfOrderableId(unitId);
     stockCard.setId(cardId);
     Map<String, String> oldExtraData = new HashMap<>();
     oldExtraData.put("vvmStatus", "STAGE_1");