Updated survey so it will not add more responses after it has been cl…

…osed.
bardia-p · Nov 3, 2023 · d95b1d6 · d95b1d6
1 parent 0f1786f
commit d95b1d6
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/src/main/java/com/opinionowl/opinionowl/models/Survey.java b/src/main/java/com/opinionowl/opinionowl/models/Survey.java
@@ -61,8 +61,12 @@ public boolean removeQuestion(Long questionId){
      * Adds a response to the survey.
      * @param response the response to add to the survey.
      */
-    public void addResponse(Response response){
-        this.responses.add(response);
+    public boolean addResponse(Response response){
+        if (!closed){
+            this.responses.add(response);
+            return true;
+        }
+        return false;
     }
 
     /**

diff --git a/src/test/java/com/opinionowl/opinionowl/SurveyTest.java b/src/test/java/com/opinionowl/opinionowl/SurveyTest.java
@@ -36,6 +36,15 @@ public void testPersist() {
 
         surveyRepository.save(survey);
 
+        survey.setClosed(true);
+
+        Response r2 = new Response(new HashMap<Long, String>(){{
+            put(Long.valueOf(1), "yo");
+            put(Long.valueOf(2), "b");
+        }});
+
+        survey.addResponse(r2);
+
         List<Survey> results = surveyRepository.findAll();
         for (Survey r : results){
             System.out.println(r.toString());