[feat] 시연용 테스트API 삭제 추가

TelePigeon · Aug 30, 2024 · 97c4e2a · 97c4e2a
1 parent 3926a61
commit 97c4e2a
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/main/java/com/telepigeon/server/controller/TestController.java b/src/main/java/com/telepigeon/server/controller/TestController.java
@@ -1,6 +1,5 @@
 package com.telepigeon.server.controller;
 
-import com.telepigeon.server.domain.Answer;
 import com.telepigeon.server.domain.Profile;
 import com.telepigeon.server.domain.Question;
 import com.telepigeon.server.domain.Token;
@@ -96,10 +95,10 @@ public ResponseEntity<Void> testQuestion(
         Profile profile2 = profileRetriever.findById(93L);
         Question question1 = questionRetriever.findFirstByProfile(profile1);
         Question question2 = questionRetriever.findFirstByProfile(profile2);
-        Answer answer1 = answerRetriever.findByQuestion(question1);
-        Answer answer2 = answerRetriever.findByQuestion(question2);
-        answerRepository.delete(answer1);
-        answerRepository.delete(answer2);
+        if (answerRetriever.existsByQuestion(question1))
+            answerRepository.delete(answerRetriever.findByQuestion(question1));
+        if (answerRetriever.existsByQuestion(question2))
+            answerRepository.delete(answerRetriever.findByQuestion(question2));
         questionRepository.delete(question1);
         questionRepository.delete(question2);
         questionService.create(profile1);