diff --git a/server/books/tests/test_module_mutations.py b/server/books/tests/test_module_mutations.py index ffb5fd57..372f1c18 100644 --- a/server/books/tests/test_module_mutations.py +++ b/server/books/tests/test_module_mutations.py @@ -117,4 +117,6 @@ class NewContentBlockMutationTest(TestCase): new_content_block = result.get('data').get('addContentBlock').get('newContentBlock') self.assertEqual(new_content_block.get('title'), title) self.assertEqual(len(new_content_block['contents']), 1) - self.assertEqual(new_content_block['contents'], [{'type': 'image_url_block', 'value': {'url': '/test.png'}}]) + content = new_content_block['contents'][0] + self.assertEqual(content.get('type'), 'image_url_block') + self.assertEqual(content.get('value'), {'url': '/test.png'})