Add delete method and unit test

neuefische · Jun 3, 2024 · dc38f2d · dc38f2d
1 parent a4955a1
commit dc38f2d
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
@@ -17,4 +17,8 @@ public RestaurantService(RestaurantRepository restaurantRepository) {
     public List<Restaurant> getRestaurants() {
         return restaurantRepository.findAll();
     }
+
+    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
@@ -42,4 +42,16 @@ void getRestaurants_whenOneRestaurantsInDB_thenReturnListOfOne() {
         List<Restaurant> expected = List.of(restaurant);
         assertEquals(expected, actual);
     }
+
+    @Test
+    void deleteRestaurant_whenMethodCalled_thenDeleteMethodOnRepositoryWasCalledOnlyOnce() {
+        //GIVEN
+        String id = "123";
+
+        //WHEN
+        restaurantService.deleteRestaurant(id);
+
+        //THEN
+        verify(mockRestaurantRepository, times(1)).deleteById(id);
+    }
 }