def test_create_page(self): user = create_user() page = create_page() now = timezone.now() self.assertEqual(page.author, user) self.assertFalse(page.is_draft) self.assertLessEqual(page.created_at, now) self.assertLessEqual(page.modified_at, now)
def handle(self, *args, **kwargs): post_count = kwargs["post_count"] if "post_count" in kwargs else 100 user_count = kwargs["user_count"] if "user_count" in kwargs else 10 comment_count = kwargs["comment_count"] if "comment_count" in kwargs else 100 page_count = kwargs["page_count"] if "page_count" in kwargs else 5 [create_user() for _ in xrange(user_count)] [create_post().save() for _ in xrange(post_count)] [create_comment().save() for _ in xrange(comment_count)] [create_page().save() for _ in xrange(page_count)]
def test_create_page_without_users_in_database(self): with self.assertRaises(IndexError): create_page()