diff --git a/src/test/java/ssu/eatssu/domain/menu/service/MenuServiceTest.java b/src/test/java/ssu/eatssu/domain/menu/service/MenuServiceTest.java index 47dc9cd..2e7a83d 100644 --- a/src/test/java/ssu/eatssu/domain/menu/service/MenuServiceTest.java +++ b/src/test/java/ssu/eatssu/domain/menu/service/MenuServiceTest.java @@ -7,6 +7,7 @@ import java.util.ArrayList; import java.util.List; +import java.util.Map; import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; import org.springframework.beans.factory.annotation.Autowired; @@ -15,7 +16,9 @@ import ssu.eatssu.domain.menu.dto.MenuResponse.MenuInformationResponse; import ssu.eatssu.domain.menu.dto.MenuResponse.MenusInformationResponse; import ssu.eatssu.domain.menu.entity.Menu; +import ssu.eatssu.domain.menu.entity.MenuCategory; import ssu.eatssu.domain.menu.repository.MealRepository; +import ssu.eatssu.domain.menu.repository.MenuCategoryRepository; import ssu.eatssu.domain.menu.repository.MenuRepository; import ssu.eatssu.domain.restaurant.entity.Restaurant; @@ -31,6 +34,9 @@ class MenuServiceTest { @Autowired private MealRepository mealRepository; + @Autowired + private MenuCategoryRepository menuCategoryRepository; + @BeforeEach void setUp() { menuRepository.deleteAll(); @@ -42,15 +48,26 @@ void setUp() { // given List menus = new ArrayList<>(); Restaurant foodCourt = Restaurant.from("FOOD_COURT"); - menus.add(Menu.createFixed("라면", foodCourt, 3000, null)); - menus.add(Menu.createFixed("떡볶이", foodCourt, 5000, null)); - menus.add(Menu.createFixed("짜게치", foodCourt, 4000, null)); + MenuCategory category1 = MenuCategory.builder().name("분식").restaurant(foodCourt).build(); + MenuCategory category2 = MenuCategory.builder().name("한식").restaurant(foodCourt).build(); + menus.add(Menu.createFixed("라면", foodCourt, 3000, category1)); + menus.add(Menu.createFixed("떡볶이", foodCourt, 5000, category2)); + menus.add(Menu.createFixed("짜게치", foodCourt, 4000, category1)); + menuCategoryRepository.save(category1); + menuCategoryRepository.save(category2); menuRepository.saveAll(menus); - // when & then - assertThat(menuService.findMenusByRestaurant(foodCourt)) - .extracting(MenuInformationResponse::getName) - .containsExactly("라면", "떡볶이", "짜게치"); + // when + Map> response = menuService.findMenusByRestaurant( + foodCourt); + + // then + assertThat(response).hasSize(2); + for (String key : response.keySet()) { + System.out.print(key + " : "); + response.get(key).forEach(menu -> System.out.print(menu.getName() + " ")); + System.out.print("\n"); + } } @Test