diff --git a/server/books/tests/test_own_content_blocks.py b/server/books/tests/test_own_content_blocks.py index 3e36ecd5..c743eaaa 100644 --- a/server/books/tests/test_own_content_blocks.py +++ b/server/books/tests/test_own_content_blocks.py @@ -53,7 +53,7 @@ class OwnContentTestCase(SkillboxTestCase): custom_content_block = ContentBlock(title='own', slug='own', user_created=True, owner=self.user) self.chapter.specific.add_child(instance=custom_content_block) - result = self.client.execute(chapterQuery, variables={ + result = self.get_client().execute(chapterQuery, variables={ "id": self.chapter_id }) self.assertEqual(len(result.data.get('chapter').get('contentBlocks')), 2) @@ -61,7 +61,7 @@ class OwnContentTestCase(SkillboxTestCase): for school_class in self.user.school_classes.all(): custom_content_block.visible_for.add(school_class) - result = self.client.execute(chapterQuery, variables={ + result = self.get_client().execute(chapterQuery, variables={ "id": self.chapter_id }) self.assertEqual(len(result.data.get('chapter').get('contentBlocks')), 2)