Add test cases for Arrange question type

1 files changed, 13 insertions, 5 deletions

diff --git a/yaksh/test_models.py b/yaksh/test_models.py
index a0ccd49..bebc929 100644
--- a/yaksh/test_models.py
+++ b/yaksh/test_models.py
@@ -1372,12 +1372,20 @@ class AnswerPaperTestCases(unittest.TestCase):
 
     def test_get_question_answer(self):
         """ Test get_question_answer() method of Answer Paper"""
+        questions = self.answerpaper.questions.all()
         answered = self.answerpaper.get_question_answers()
-        first_answer = list(answered.values())[0][0]
-        first_answer_obj = first_answer['answer']
-        self.assertEqual(first_answer_obj.answer, 'Demo answer')
-        self.assertTrue(first_answer_obj.correct)
-        self.assertEqual(len(answered), 2)
+        for question in questions:
+            answers_saved = Answer.objects.filter(question=question)
+            error_list = [json.loads(ans.error) for ans in answers_saved]
+            if answers_saved:
+                self.assertEqual(len(answered[question]), len(answers_saved))
+                ans_list = []
+                err_list = []
+                for val in answered[question]:
+                    ans_list.append(val.get('answer'))
+                    err_list.append(val.get('error_list'))
+                self.assertEqual(set(ans_list), set(answers_saved))
+                self.assertEqual(error_list, err_list)
 
     def test_is_answer_correct(self):
         self.assertTrue(self.answerpaper.is_answer_correct(self.questions[0]))