fix: 심리테스트 수정-> colors쪽 에러

LikeLion-12th-SKHU · Aug 5, 2024 · 12ffc53 · 12ffc53
1 parent b829c1a
commit 12ffc53
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/src/main/java/net/skhu/likelion12thteam03be/survey/application/SurveyService.java b/src/main/java/net/skhu/likelion12thteam03be/survey/application/SurveyService.java
@@ -59,7 +59,6 @@ public void save(SurveySaveReqDto surveySaveReqDto, Principal principal) {
                         .build())
                         .toList();
         surveyColorRepository.saveAll(surveyColors);
-
     }
 
     public SurveyResDto findByLoginId(Principal principal) {
@@ -97,14 +96,15 @@ public void update(SurveySaveReqDto surveySaveReqDto, Principal principal) {
         Emotion emotion = emotionRepository.findById(surveySaveReqDto.emotionId())
                 .orElseThrow(() -> new IllegalArgumentException("해당 감정을 찾을 수 없습니다: " + surveySaveReqDto.emotionId()));
         List<Color> colors = colorRepository.findAllById(surveySaveReqDto.colorIds());
+
         List<SurveyColor> surveyColors = colors.stream()
                 .map(color -> SurveyColor.builder()
                         .survey(survey)
                         .color(color)
                         .build())
                 .toList();
-
         survey.update(surveySaveReqDto.score(), emotion, surveyColors);
+
         surveyRepository.save(survey);
     }
 }
diff --git a/src/main/java/net/skhu/likelion12thteam03be/survey/domain/Survey.java b/src/main/java/net/skhu/likelion12thteam03be/survey/domain/Survey.java
@@ -38,15 +38,14 @@ public class Survey {
     public Survey(int score, Emotion emotion, List<SurveyColor> colors, User user) {
         this.score = score;
         this.emotion = emotion;
-        this.colors = colors != null ? colors : new ArrayList<>();
+        this.colors = colors;
         this.user = user;
     }
 
     public void update(int score, Emotion emotion, List<SurveyColor> colors) {
         this.score = score;
         this.emotion = emotion;
-        if (colors != null) {
-            this.colors.addAll(colors);
-        }
+        this.colors.clear();
+        this.colors.addAll(colors);
     }
 }