def test_is_friend(self): user = User(username="******", email="*****@*****.**") db.session.add(user) assert user.get_following().count() == 0 user2 = User(username="******", email="*****@*****.**") db.session.add(user2) db.session.commit() assert not user.is_friend(user2) assert not user2.is_friend(user) user.follow(user2) db.session.commit() assert not user.is_friend(user2) assert not user2.is_friend(user)
def test_is_friend(self): user = User(username="******", email="*****@*****.**") db.session.add(user) assert user.get_following().count() == 0 user2 = User(username="******", email="*****@*****.**") db.session.add(user2) db.session.commit() assert not user.is_friend(user2) assert not user2.is_friend(user) user.follow(user2) db.session.commit() assert not user.is_friend(user2) assert not user2.is_friend(user) user2.follow(user) assert user.is_friend(user2) assert user2.is_friend(user)
def test_get_friends(self): user = User(username="******", email="*****@*****.**") db.session.add(user) assert user.get_friends().count() == 0 user2 = User(username="******", email="*****@*****.**") db.session.add(user2) assert user2.get_friends().count() == 0 db.session.commit() assert not user.is_friend(user2) assert not user2.is_friend(user) assert user.get_friends().count() == 0 assert user2.get_friends().count() == 0 user.follow(user2) db.session.commit() assert not user.is_friend(user2) assert not user2.is_friend(user) assert user.get_friends().count() == 0 assert user2.get_friends().count() == 0 user2.follow(user) assert user.is_friend(user2) assert user2.is_friend(user) assert user.get_friends().count() == 1 assert user2.get_friends().count() == 1 assert user.get_friends().first().id == user2.id assert user2.get_friends().first().id == user.id