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={