test: 추천 경우 추가

GDSC-Hongik · Aug 14, 2024 · cdfaeb9 · cdfaeb9
1 parent 5b248f9
commit cdfaeb9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/reviews/views.py b/reviews/views.py
@@ -80,15 +80,15 @@ def evaluate_review(request, restaurant_id, review_id):
         )
 
         evaluation = request.data.get("evaluation")
-        if evaluation == "1":
+        if evaluation == "1" or evaluation == 1:
             # 좋아요
             if created or recommend_entry.recommend != 1:
                 review.recommend_count += 1
                 if recommend_entry.recommend == 0:  # 이전에 싫어요였던 경우
                     review.decommend_count -= 1
                 recommend_entry.recommend = 1
 
-        elif evaluation == "0":
+        elif evaluation == "0" or evaluation == 0:
             # 싫어요
             if created or recommend_entry.recommend != 0:
                 review.decommend_count += 1