From 1b19d8e6eb22f79190cb2f1760025c49a0e64eaf Mon Sep 17 00:00:00 2001 From: Ramon Wenger Date: Thu, 28 Jan 2021 13:57:25 +0100 Subject: [PATCH] Fix unit test --- .../tests/test_module_solution_visibility.py | 28 +++++++++++-------- 1 file changed, 17 insertions(+), 11 deletions(-) diff --git a/server/books/tests/test_module_solution_visibility.py b/server/books/tests/test_module_solution_visibility.py index e9b769b9..a27c1417 100644 --- a/server/books/tests/test_module_solution_visibility.py +++ b/server/books/tests/test_module_solution_visibility.py @@ -56,36 +56,42 @@ class ModuleSolutionVisibilityTest(TestCase): def test_hide_solutions_for_students_and_then_show_them(self): self.assertEqual(self.student.selected_class(), self.teacher.selected_class()) - result = self.student_client.execute(self.query, variables={ + student_result = self.student_client.execute(self.query, variables={ 'id': self.content_block_id }) - self.assertIsNone(result.get('errors')) - self.assertEqual(len(result.get('data').get('contentBlock').get('contents')), 0) + self.assertIsNone(student_result.get('errors')) + self.assertEqual(len(student_result.get('data').get('contentBlock').get('contents')), 0) - result = self.teacher_client.execute(self.query, variables={ + teacher_result = self.teacher_client.execute(self.query, variables={ 'id': self.content_block_id }) - self.assertIsNone(result.get('errors')) - self.assertEqual(len(result.get('data').get('contentBlock').get('contents')), 0) + self.assertIsNone(teacher_result.get('errors')) + self.assertEqual(len(teacher_result.get('data').get('contentBlock').get('contents')), 1) - result = self.teacher_client.execute(self.update_mutation, variables={ + teacher_result = self.teacher_client.execute(self.update_mutation, variables={ 'input': { 'slug': self.module.slug, 'enabled': True } }) - self.assertEqual(result.get('data').get('updateSolutionVisibility').get('success'), True) + self.assertEqual(teacher_result.get('data').get('updateSolutionVisibility').get('success'), True) self.assertEqual(self.module.solutions_enabled_for.filter(pk=self.selected_class.pk).count(), 1) - result = self.student_client.execute(self.query, variables={ + student_result = self.student_client.execute(self.query, variables={ 'id': self.content_block_id }) - self.assertIsNone(result.get('errors')) - self.assertEqual(len(result.get('data').get('contentBlock').get('contents')), 1) + self.assertIsNone(student_result.get('errors')) + self.assertEqual(len(student_result.get('data').get('contentBlock').get('contents')), 1) + + teacher_result = self.teacher_client.execute(self.query, variables={ + 'id': self.content_block_id + }) + self.assertIsNone(teacher_result.get('errors')) + self.assertEqual(len(teacher_result.get('data').get('contentBlock').get('contents')), 1) def test_try_to_show_solutions_as_student_and_fail(self): result = self.student_client.execute(self.query, variables={