def test_followers(self): user = User() assert user.followers == set() user.followers = set([1]) assert user.followers == set([1])
def test_get_followers(self): user = User(username="******", email="*****@*****.**") db.session.add(user) assert user.get_followers().count() == 0 user2 = User(username="******", email="*****@*****.**") db.session.add(user2) db.session.commit() user.followers = set([user2.id]) assert user.get_followers().count() == 1 assert user.get_followers().first().id == user2.id