コード例 #1
0
def test_add_post():
    user = UserFactory()
    post1 = TextPostFactory()
    post2 = PicturePostFactory()

    user.add_post(post1)
    user.add_post(post2)

    assert len(user.posts) == 2
コード例 #2
0
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]