示例#1
0
    def test_follow_unfollow(self):
        user1 = User(name='tester1', email='*****@*****.**', password='******')
        db.session.add(user1)
        user2 = User(name='tester2', email='*****@*****.**', password='******')
        db.session.add(user2)
        db.session.commit()

        user1.follow(user2)
        assert user1.num_following == 1
        assert user1.get_following_query().first().id == user2.id
        assert user2.num_followers == 1
        assert user2.get_followers_query().first().id == user1.id

        user1.unfollow(user2)
        assert user1.num_following == 0
        assert user2.num_followers == 0