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