Add delete method and unit test

neuefische · Jun 5, 2024 · a8645f3 · a8645f3
1 parent 2942910
commit a8645f3
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/backend/src/main/java/com/neuefische/team2/backend/restaurant/RestaurantService.java b/backend/src/main/java/com/neuefische/team2/backend/restaurant/RestaurantService.java
@@ -25,4 +25,8 @@ public Restaurant findRestaurantById(String id) {
     public Restaurant addRestaurant(Restaurant restaurant) {
         return restaurantRepository.save(restaurant);
     }
+
+    public void deleteRestaurant(String id) {
+        restaurantRepository.deleteById(id);
+    }
 }
diff --git a/backend/src/test/java/com/neuefische/team2/backend/restaurant/RestaurantServiceTest.java b/backend/src/test/java/com/neuefische/team2/backend/restaurant/RestaurantServiceTest.java
@@ -84,4 +84,16 @@ void addRestaurant_whenRestaurantToSave_thenReturnSavedRestaurantWithId() {
         verify(mockRestaurantRepository).save(restaurantToSave);
         assertEquals(savedRestaurant, actual);
     }
+
+    @Test
+    void deleteRestaurant_whenMethodCalled_thenDeleteMethodOnRepositoryWasCalledOnlyOnce() {
+        //GIVEN
+        String id = "123";
+
+        //WHEN
+        restaurantService.deleteRestaurant(id);
+
+        //THEN
+        verify(mockRestaurantRepository, times(1)).deleteById(id);
+    }
 }