diff --git a/tests/tests/test_api.py b/tests/tests/test_api.py index ff38065..3d2f88f 100644 --- a/tests/tests/test_api.py +++ b/tests/tests/test_api.py @@ -374,7 +374,8 @@ def test_streamfield_with_null_page(self): data = json.loads(response.content) # result should have a mapping for the page we just created, and its parent - self.assertEqual(len(data['mappings']), 2) + page_mappings = filter(lambda mapping: mapping[0] == 'wagtailcore.page', data['mappings']) + self.assertEqual(len(list(page_mappings)), 2) def test_parental_many_to_many(self): page = PageWithParentalManyToMany(title="This page has lots of ads!")