diff --git a/src/main/java/com/mangkyu/employment/interview/app/enums/controller/EnumsController.java b/src/main/java/com/mangkyu/employment/interview/app/enums/controller/EnumsController.java index a0d5525..5f104c8 100644 --- a/src/main/java/com/mangkyu/employment/interview/app/enums/controller/EnumsController.java +++ b/src/main/java/com/mangkyu/employment/interview/app/enums/controller/EnumsController.java @@ -9,6 +9,7 @@ import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.RestController; import java.util.List; +import java.util.stream.Collectors; @RestController @RequiredArgsConstructor @@ -18,7 +19,11 @@ public class EnumsController { @GetMapping("/enums/quiz-categories") public ResponseEntity> getQuizCategoryList() { - return ResponseEntity.ok(enumMapperFactory.get(EnumMapperKey.QUIZ_CATEGORY)); + final List quizCategoryList = enumMapperFactory.get(EnumMapperKey.QUIZ_CATEGORY); + + return ResponseEntity.ok(quizCategoryList.stream() + .filter(EnumMapperValue::isExpose) + .collect(Collectors.toList())); } @GetMapping("/enums/quiz-levels") diff --git a/src/test/java/com/mangkyu/employment/interview/app/enums/controller/EnumsControllerTest.java b/src/test/java/com/mangkyu/employment/interview/app/enums/controller/EnumsControllerTest.java index 98e6398..0e9ef39 100644 --- a/src/test/java/com/mangkyu/employment/interview/app/enums/controller/EnumsControllerTest.java +++ b/src/test/java/com/mangkyu/employment/interview/app/enums/controller/EnumsControllerTest.java @@ -15,6 +15,7 @@ import org.springframework.test.web.servlet.MockMvc; import org.springframework.test.web.servlet.ResultActions; import org.springframework.test.web.servlet.request.MockMvcRequestBuilders; +import java.util.Arrays; import java.util.List; import static org.assertj.core.api.Assertions.assertThat; @@ -43,7 +44,7 @@ class EnumsControllerTest { // then final ResultActions resultActions = result.andExpect(status().isOk()); final String stringResponse = resultActions.andReturn().getResponse().getContentAsString(); - assertThat(new Gson().fromJson(stringResponse, List.class).size()).isEqualTo(QuizCategory.values().length); + assertThat(new Gson().fromJson(stringResponse, List.class).size()).isEqualTo(Arrays.stream(QuizCategory.values()).filter(QuizCategory::isExpose).count()); } @Test