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
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
