class PostTest(TestCase):
"""Tests for Post model."""
def setUp(self):
self.post_1 = PostFactory(published=False)
self.post_2 = PostFactory()
def test_post_str(self):
post_1 = Post.objects.get(id=self.post_1.id)
post_2 = Post.objects.get(id=self.post_2.id)
self.assertEqual(str(post_1), self.post_1.title)
self.assertNotEqual(str(post_2), self.post_1.id)
def test_post_url(self):
django = Post.objects.get(id=self.post_1.id)
self.assertEqual(f"/post/{django.id}/", django.get_absolute_url())
def test_post_published(self):
post = Post.objects.get(title=self.post_1.title)
self.post_1.published = True
self.post_1.save()
self.post_1.refresh_from_db()
self.assertNotEquals(post.pub_date, self.post_1.pub_date)
def test_image(self):
test_post = PostFactory(background=TEST_IMAGE)
self.assertTrue('default.jpg' in test_post.image)
setattr(test_post, 'background', None)
test_post.save()
self.assertTrue(test_post.image is None)
