def test_add_post(self):
user = UserFactory()
post1 = TextPostFactory()
post2 = PicturePostFactory()
user.add_post(post1)
user.add_post(post2)
self.assertEqual(len(user.posts), 2)
def test_user_get_posts_by_followers1(self):
"""Should only return posts from users I'm following"""
user1 = UserFactory()
user2 = UserFactory()
user3 = UserFactory()
user4 = UserFactory()
user2_post1 = TextPost('Hello there friends', timestamp=datetime(2017, 2, 2))
user2_post2 = TextPost('Hello there friends', timestamp=datetime(2017, 2, 3))
user3_post1 = TextPost('Hello there friends', timestamp=datetime(2017, 2, 1))
user4_post1 = TextPost('Hello there friends', timestamp=datetime(2017, 2, 3))
user2.add_post(user2_post1)
user2.add_post(user2_post2)
user3.add_post(user3_post1)
user4.add_post(user4_post1)
# user1 follows user2 and user3
user1.follow(user2)
user1.follow(user3)
# 2 posts from user2 and 1 from user3
# post from user4 is excluded
self.assertEqual(len(user1.get_timeline()), 3)
self.assertFalse(user4_post1 in user1.get_timeline())
# should be sorted by creation timestamp
self.assertEqual(user1.get_timeline(), [
user2_post2,
user2_post1,
user3_post1
])
