Fix unit test

This commit is contained in:
Ramon Wenger 2021-01-28 13:57:25 +01:00
parent a47a41d351
commit 1b19d8e6eb
1 changed files with 17 additions and 11 deletions

View File

@ -56,36 +56,42 @@ class ModuleSolutionVisibilityTest(TestCase):
def test_hide_solutions_for_students_and_then_show_them(self): def test_hide_solutions_for_students_and_then_show_them(self):
self.assertEqual(self.student.selected_class(), self.teacher.selected_class()) 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 'id': self.content_block_id
}) })
self.assertIsNone(result.get('errors')) self.assertIsNone(student_result.get('errors'))
self.assertEqual(len(result.get('data').get('contentBlock').get('contents')), 0) 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 'id': self.content_block_id
}) })
self.assertIsNone(result.get('errors')) self.assertIsNone(teacher_result.get('errors'))
self.assertEqual(len(result.get('data').get('contentBlock').get('contents')), 0) 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': { 'input': {
'slug': self.module.slug, 'slug': self.module.slug,
'enabled': True '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) 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 'id': self.content_block_id
}) })
self.assertIsNone(result.get('errors')) self.assertIsNone(student_result.get('errors'))
self.assertEqual(len(result.get('data').get('contentBlock').get('contents')), 1) 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): def test_try_to_show_solutions_as_student_and_fail(self):
result = self.student_client.execute(self.query, variables={ result = self.student_client.execute(self.query, variables={