11from django .core .paginator import Page as PaginatorPage
2- from faker import Faker
3- from wagtail .coreutils import get_dummy_request
4- from wagtail .models import PageViewRestriction
52
3+ from faker import Faker
64from tbx .blog .factories import BlogIndexPageFactory , BlogPageFactory
75from tbx .blog .models import BlogPage
86from tbx .taxonomy .factories import SectorFactory , ServiceFactory
7+ from wagtail .coreutils import get_dummy_request
8+ from wagtail .models import PageViewRestriction
99from wagtail .test .utils import WagtailPageTestCase
1010
1111fake = Faker (["en_GB" ])
@@ -16,11 +16,17 @@ class TestBlogIndexPageFactory(WagtailPageTestCase):
1616 def setUpTestData (cls ):
1717 cls .blog_index = BlogIndexPageFactory (title = "The Torchbox Blog" )
1818 cls .blog_post = BlogPageFactory (title = "The blog post" , parent = cls .blog_index )
19- cls .private_blog_post = BlogPageFactory (title = "Private blog post" , parent = cls .blog_index )
20- cls .draft_blog_post = BlogPageFactory (title = "Draft blog post" , live = False , parent = cls .blog_index )
19+ cls .private_blog_post = BlogPageFactory (
20+ title = "Private blog post" , parent = cls .blog_index
21+ )
22+ cls .draft_blog_post = BlogPageFactory (
23+ title = "Draft blog post" , live = False , parent = cls .blog_index
24+ )
2125
2226 PageViewRestriction .objects .create (
23- page = cls .private_blog_post , restriction_type = "password" , password = "password123"
27+ page = cls .private_blog_post ,
28+ restriction_type = "password" ,
29+ password = "password123" ,
2430 )
2531
2632 def test_get_context (self ):
@@ -34,7 +40,9 @@ def test_blog_posts_property(self):
3440 another_blog = BlogPageFactory (title = "Tech blog" )
3541 BlogPageFactory (title = "Tech blog" , parent = another_blog )
3642
37- self .assertQuerysetEqual (self .blog_index .blog_posts , BlogPage .objects .filter (pk = self .blog_post .pk ))
43+ self .assertQuerysetEqual (
44+ self .blog_index .blog_posts , BlogPage .objects .filter (pk = self .blog_post .pk )
45+ )
3846
3947
4048class TestBlogPageFactory (WagtailPageTestCase ):
0 commit comments