Create test data in specific test cases

Author: dizballanze

bar/blog/tests/base_test_case.py

@@ -6,16 +6,17 @@ class BaseBlogTestCase(TestCase):
 
     def setUp(self):
         super().setUp()
-        tags = [Tag.objects.create(name='Foo'), Tag.objects.create(name='Bar'), Tag.objects.create(name='Spam')]
-        self.authors = []
-        for author_num in range(5):
-            author = Author.objects.create(
-                bio='Bio #{}'.format(author_num), email='author{}@e.co'.format(author_num),
-                username='author-{}'.format(author_num))
-            author.set_password('v3rys3cr31')
-            for article_num in range(10):
-                article = Article.objects.create(
-                    title='Article #{}'.format(author_num * 10 + article_num), content='foo bar', author=author,
-                    comments_on=(article_num % 2 == 0))
-                article.tags.add(*tags)
-            self.authors.append(author)
+
+    def _create_tags(self):
+        return [Tag.objects.create(name='Foo'), Tag.objects.create(name='Bar'), Tag.objects.create(name='Spam')]
+
+    def _create_author(self, username, email, bio, password):
+        author = Author.objects.create(bio=bio, email=email, username=username)
+        author.set_password(password)
+        return author
+
+    def _create_article(self, title, content, author, comments_on, tags):
+        article = Article.objects.create(title=title, content=content, author=author, comments_on=comments_on)
+        if tags:
+            article.tags.add(*tags)
+        return article

bar/blog/tests/tests_articles_list.py

@@ -11,6 +11,20 @@ def test_foo_bar(self):
     def test_url_resolving(self):
         self.assertEqual(reverse('articles_list'), '/blog/')
 
+
+class ArticlesListPaginationTestCase(BaseBlogTestCase):
+
+    def setUp(self):
+        super().setUp()
+        self.tags = self._create_tags()
+        for author_num in range(5):
+            author = self._create_author(
+                'author-{}'.format(author_num), 'author-{}@e.co'.format(author_num), 'Bio #{}'.format(author_num),
+                'v3rys3cr31')
+            for article_num in range(10):
+                self._create_article(
+                    'Article #{}'.format(author_num*10+article_num), 'foo bar', author, True, self.tags)
+
     def test_first_20_articles_are_on_the_page(self):
         resp = self.client.get(reverse('articles_list'))
         for i in range(20):

bar/blog/tests/tests_articles_tags.py

@@ -5,6 +5,12 @@
 
 class ArticlesTagsTestCase(BaseBlogTestCase):
 
+    def setUp(self):
+        super().setUp()
+        tags = self._create_tags()
+        author = self._create_author('foobar', 'foobar@e.co', 'spam', 'v3rys3cr31')
+        self._create_article('Spam', 'foo bar', author, True, tags)
+
     def test_foo_bar(self):
         self.assertEqual(3, 2+1)
 

bar/blog/tests/tests_author_page.py

@@ -12,6 +12,6 @@ def test_url_resolving(self):
         self.assertEqual(reverse('author_page', kwargs=dict(username='foo-bar')), '/blog/authors/foo-bar')
 
     def test_author_name_on_its_page(self):
-        author = self.authors[0]
+        author = self._create_author('foobar', 'foobar@e.co', 'spam', 'v3rys3cr31')
         resp = self.client.get(reverse('author_page', kwargs=dict(username=author.username)))
         self.assertIn(author.username, str(resp.content))

bar/blog/views.py

@@ -11,7 +11,7 @@ class ArticlesListView(ListView):
     model = Article
     context_object_name = 'articles'
     paginate_by = 20
-    queryset = Article.objects.all()
+    queryset = Article.objects.all().order_by('id')
 
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)