Issue jmprathab#257 update test

DDmit04 · Mar 10, 2021 · 53767d1 · 53767d1
1 parent 2b39fd7
commit 53767d1
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/service/src/test/java/com/myhome/services/unit/AmenitySDJpaServiceTest.java b/service/src/test/java/com/myhome/services/unit/AmenitySDJpaServiceTest.java
@@ -68,6 +68,8 @@ class AmenitySDJpaServiceTest {
   @Mock
   private AmenityApiMapper amenityApiMapper;
   @Mock
+  private AmenityBookingService amenityBookingService;
+  @Mock
   private AmenityBookingItemRepository bookingItemRepository;
 
   @InjectMocks
@@ -84,30 +86,31 @@ void deleteAmenity() {
     Amenity testAmenity =
         TestUtils.AmenityHelpers.getTestAmenity(TEST_AMENITY_ID, TEST_AMENITY_DESCRIPTION);
 
-    given(amenityRepository.findByAmenityIdWithCommunity(TEST_AMENITY_ID))
+    given(amenityRepository.findByAmenityIdWithCommunityAndHouse(TEST_AMENITY_ID))
         .willReturn(Optional.of(testAmenity));
 
     // when
     boolean amenityDeleted = amenitySDJpaService.deleteAmenity(TEST_AMENITY_ID);
 
     // then
     assertTrue(amenityDeleted);
-    verify(amenityRepository).findByAmenityIdWithCommunity(TEST_AMENITY_ID);
+    verify(amenityRepository).findByAmenityIdWithCommunityAndHouse(TEST_AMENITY_ID);
+    verify(amenityBookingService).removeAllAmenityBookings(TEST_AMENITY_ID);
     verify(amenityRepository).delete(testAmenity);
   }
 
   @Test
   void deleteAmenityNotExists() {
     // given
-    given(amenityRepository.findByAmenityIdWithCommunity(TEST_AMENITY_ID))
+    given(amenityRepository.findByAmenityIdWithCommunityAndHouse(TEST_AMENITY_ID))
         .willReturn(Optional.empty());
 
     // when
     boolean amenityDeleted = amenitySDJpaService.deleteAmenity(TEST_AMENITY_ID);
 
     // then
     assertFalse(amenityDeleted);
-    verify(amenityRepository).findByAmenityIdWithCommunity(TEST_AMENITY_ID);
+    verify(amenityRepository).findByAmenityIdWithCommunityAndHouse(TEST_AMENITY_ID);
     verify(amenityRepository, never()).delete(any());
   }