def test_post_string_representation():
user = UserFactory(first_name='Kevin', last_name='Watson')
post1 = TextPostFactory()
post2 = PicturePostFactory()
post3 = CheckInPostFactory()
post1.set_user(user)
post2.set_user(user)
post3.set_user(user)
assert str(post1) == '@Kevin Watson: "Sample post text"\n\tTuesday, Jan 10, 2017'
assert str(post2) == '@Kevin Watson: "Sample post text"\n\thttp://fake-domain.com/images/sample.jpg\n\tTuesday, Jan 10, 2017'
assert str(post3) == '@Kevin Checked In: "Sample post text"\n\t-34.603722, -58.381592\n\tTuesday, Jan 10, 2017'
def test_user_timeline():
"""Should only return posts from users I'm following"""
user1 = UserFactory()
user2 = UserFactory()
user3 = UserFactory()
user4 = UserFactory()
user2_post1 = TextPostFactory()
user2_post2 = TextPostFactory()
user3_post1 = PicturePostFactory()
user4_post1 = TextPostFactory()
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
assert len(user1.get_timeline()) == 3
assert user4_post1 not in user1.get_timeline()
# should be sorted by creation timestamp
assert user1.get_timeline() == [user2_post1, user2_post2, user3_post1]
def test_add_post():
user = UserFactory()
post1 = TextPostFactory()
post2 = PicturePostFactory()
user.add_post(post1)
user.add_post(post2)
assert len(user.posts) == 2