def test_CancelFriendRequest(db):
user1, token1 = generate_user("user1")
user2, token2 = generate_user("user2")
with api_session(token1) as api:
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user2.id))
res = api.ListFriendRequests(empty_pb2.Empty())
assert res.sent[0].user_id == user2.id
fr_id = res.sent[0].friend_request_id
api.CancelFriendRequest(
api_pb2.CancelFriendRequestReq(friend_request_id=fr_id))
res = api.ListFriendRequests(empty_pb2.Empty())
assert len(res.sent) == 0
def test_CancelFriendRequest(db):
user1, token1 = generate_user()
user2, token2 = generate_user()
with api_session(token1) as api:
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user2.id))
res = api.ListFriendRequests(empty_pb2.Empty())
assert res.sent[0].user_id == user2.id
fr_id = res.sent[0].friend_request_id
api.CancelFriendRequest(
api_pb2.CancelFriendRequestReq(friend_request_id=fr_id))
# check it's gone
res = api.ListFriendRequests(empty_pb2.Empty())
assert len(res.sent) == 0
assert len(res.received) == 0
# check not friends
res = api.ListFriends(empty_pb2.Empty())
assert len(res.user_ids) == 0
with api_session(token2) as api:
# check it's gone
res = api.ListFriendRequests(empty_pb2.Empty())
assert len(res.sent) == 0
assert len(res.received) == 0
# check we're not friends
res = api.ListFriends(empty_pb2.Empty())
assert len(res.user_ids) == 0
with api_session(token1) as api:
# check we can send another friend req
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user2.id))
res = api.ListFriendRequests(empty_pb2.Empty())
assert res.sent[
0].state == api_pb2.FriendRequest.FriendRequestStatus.PENDING
assert res.sent[0].user_id == user2.id