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()
