def test_cant_friend_request_pending(db):
user1, token1 = generate_user("user1")
user2, token2 = generate_user("user2")
user3, token3 = generate_user("user3")
# send friend request from user1 to user2
with api_session(token1) as api:
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user2.id))
with api_session(token2) as api, pytest.raises(grpc.RpcError):
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user1.id))
def test_cant_friend_request_twice(db):
user1, token1 = generate_user()
user2, token2 = generate_user()
# send friend request from user1 to user2
with api_session(token1) as api:
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user2.id))
with pytest.raises(grpc.RpcError) as e:
api.SendFriendRequest(
api_pb2.SendFriendRequestReq(user_id=user2.id))
assert e.value.code() == grpc.StatusCode.FAILED_PRECONDITION
assert e.value.details() == errors.FRIENDS_ALREADY_OR_PENDING
def test_reject_friend_request(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].state == api_pb2.FriendRequest.FriendRequestStatus.PENDING
assert res.sent[0].user_id == user2.id
with api_session(token2) as api:
res = api.ListFriendRequests(empty_pb2.Empty())
assert res.received[
0].state == api_pb2.FriendRequest.FriendRequestStatus.PENDING
assert res.received[0].user_id == user1.id
fr_id = res.received[0].friend_request_id
# reject it
api.RespondFriendRequest(
api_pb2.RespondFriendRequestReq(friend_request_id=fr_id,
accept=False))
# 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(token1) 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
# 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
def test_cant_friend_request_already_friends(db):
user1, token1 = generate_user()
user2, token2 = generate_user()
make_friends(user1, user2)
with api_session(token1) as api:
with pytest.raises(grpc.RpcError) as e:
api.SendFriendRequest(
api_pb2.SendFriendRequestReq(user_id=user2.id))
assert e.value.code() == grpc.StatusCode.FAILED_PRECONDITION
assert e.value.details() == errors.FRIENDS_ALREADY_OR_PENDING
with api_session(token2) as api:
with pytest.raises(grpc.RpcError) as e:
api.SendFriendRequest(
api_pb2.SendFriendRequestReq(user_id=user1.id))
assert e.value.code() == grpc.StatusCode.FAILED_PRECONDITION
assert e.value.details() == errors.FRIENDS_ALREADY_OR_PENDING
def test_friend_request_flow(db):
user1, token1 = generate_user("user1")
user2, token2 = generate_user("user2")
user3, token3 = generate_user("user3")
# send friend request from user1 to user2
with api_session(token1) as api:
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user2.id))
# check it went through
res = api.ListFriendRequests(empty_pb2.Empty())
assert len(res.sent) == 1
assert len(res.received) == 0
assert res.sent[
0].state == api_pb2.FriendRequest.FriendRequestStatus.PENDING
assert res.sent[0].user_id == user2.id
with api_session(token2) as api:
# check it's there
res = api.ListFriendRequests(empty_pb2.Empty())
assert len(res.sent) == 0
assert len(res.received) == 1
assert res.received[
0].state == api_pb2.FriendRequest.FriendRequestStatus.PENDING
assert res.received[0].user_id == user1.id
fr_id = res.received[0].friend_request_id
# accept it
api.RespondFriendRequest(
api_pb2.RespondFriendRequestReq(friend_request_id=fr_id,
accept=True))
# check it's gone
res = api.ListFriendRequests(empty_pb2.Empty())
assert len(res.sent) == 0
assert len(res.received) == 0
# check we're friends now
res = api.ListFriends(empty_pb2.Empty())
assert len(res.user_ids) == 1
assert res.user_ids[0] == user1.id
with api_session(token1) 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 friends now
res = api.ListFriends(empty_pb2.Empty())
assert len(res.user_ids) == 1
assert res.user_ids[0] == user2.id
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))
# 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
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_ListMutualFriends(db):
user1, token1 = generate_user()
user2, token2 = generate_user()
user3, token3 = generate_user()
user4, token4 = generate_user()
user5, token5 = generate_user()
# arrange friends like this: 1<->2, 1<->3, 1<->4, 1<->5, 3<->2, 3<->4,
# so 1 and 2 should have mutual friend 3 only
make_friends(user1, user2)
make_friends(user1, user3)
make_friends(user1, user4)
make_friends(user1, user5)
make_friends(user3, user2)
make_friends(user3, user4)
with api_session(token1) as api:
mutual_friends = api.ListMutualFriends(
api_pb2.ListMutualFriendsReq(user_id=user2.id)).mutual_friends
assert len(mutual_friends) == 1
assert mutual_friends[0].user_id == user3.id
# and other way around same
with api_session(token2) as api:
mutual_friends = api.ListMutualFriends(
api_pb2.ListMutualFriendsReq(user_id=user1.id)).mutual_friends
assert len(mutual_friends) == 1
assert mutual_friends[0].user_id == user3.id
# Check pending request doesn't have effect
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user5.id))
mutual_friends = api.ListMutualFriends(
api_pb2.ListMutualFriendsReq(user_id=user1.id)).mutual_friends
assert len(mutual_friends) == 1
assert mutual_friends[0].user_id == user3.id
# both ways
with api_session(token1) as api:
mutual_friends = api.ListMutualFriends(
api_pb2.ListMutualFriendsReq(user_id=user2.id)).mutual_friends
assert len(mutual_friends) == 1
assert mutual_friends[0].user_id == user3.id
def test_pending_friend_request_count(db):
user1, token1 = generate_user("user1")
user2, token2 = generate_user("user2")
user3, token3 = generate_user("user3")
with api_session(token2) as api:
res = api.Ping(api_pb2.PingReq())
assert res.pending_friend_request_count == 0
with api_session(token1) as api:
res = api.Ping(api_pb2.PingReq())
assert res.pending_friend_request_count == 0
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user2.id))
res = api.Ping(api_pb2.PingReq())
assert res.pending_friend_request_count == 0
with api_session(token2) as api:
res = api.Ping(api_pb2.PingReq())
assert res.pending_friend_request_count == 1
with api_session(token2) as api:
# check it's there
res = api.ListFriendRequests(empty_pb2.Empty())
assert len(res.sent) == 0
assert len(res.received) == 1
assert res.received[
0].state == api_pb2.FriendRequest.FriendRequestStatus.PENDING
assert res.received[0].user_id == user1.id
fr_id = res.received[0].friend_request_id
# accept it
api.RespondFriendRequest(
api_pb2.RespondFriendRequestReq(friend_request_id=fr_id,
accept=True))
res = api.Ping(api_pb2.PingReq())
assert res.pending_friend_request_count == 0
def test_ListFriends(db):
user1, token1 = generate_user()
user2, token2 = generate_user()
user3, token3 = generate_user()
# send friend request from user1 to user2 and user3
with api_session(token1) as api:
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user2.id))
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user3.id))
with api_session(token3) as api:
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user2.id))
with api_session(token2) as api:
res = api.ListFriendRequests(empty_pb2.Empty())
assert len(res.received) == 2
# order is an implementation detail
user1_req = [req for req in res.received if req.user_id == user1.id][0]
user3_req = [req for req in res.received if req.user_id == user3.id][0]
assert user1_req.state == api_pb2.FriendRequest.FriendRequestStatus.PENDING
assert user1_req.user_id == user1.id
api.RespondFriendRequest(
api_pb2.RespondFriendRequestReq(
friend_request_id=user1_req.friend_request_id, accept=True))
assert user3_req.state == api_pb2.FriendRequest.FriendRequestStatus.PENDING
assert user3_req.user_id == user3.id
api.RespondFriendRequest(
api_pb2.RespondFriendRequestReq(
friend_request_id=user3_req.friend_request_id, accept=True))
# check we now have two friends
res = api.ListFriends(empty_pb2.Empty())
assert len(res.user_ids) == 2
assert user1.id in res.user_ids
assert user3.id in res.user_ids
with api_session(token3) as api:
res = api.ListFriends(empty_pb2.Empty())
assert len(res.user_ids) == 1
assert user2.id in res.user_ids
res = api.ListFriendRequests(empty_pb2.Empty())
assert len(res.received) == 1
assert res.received[
0].state == api_pb2.FriendRequest.FriendRequestStatus.PENDING
assert res.received[0].user_id == user1.id
fr_id = res.received[0].friend_request_id
api.RespondFriendRequest(
api_pb2.RespondFriendRequestReq(friend_request_id=fr_id,
accept=True))
res = api.ListFriends(empty_pb2.Empty())
assert len(res.user_ids) == 2
assert user1.id in res.user_ids
assert user2.id in res.user_ids
with api_session(token1) as api:
res = api.ListFriends(empty_pb2.Empty())
assert len(res.user_ids) == 2
assert user2.id in res.user_ids
assert user3.id in res.user_ids
def test_mutual_friends(db):
user1, token1 = generate_user("user1")
user2, token2 = generate_user("user2")
user3, token3 = generate_user("user3")
user4, token4 = generate_user("user4")
user5, token5 = generate_user("user5")
# arrange friends like this: 1<->2, 1<->3, 1<->4, 1<->5, 3<->2, 3<->4,
# 2<->5 pending
# so 1 and 2 should have mutual friend 3 only
with api_session(token1) as api:
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user2.id))
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user3.id))
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user4.id))
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user5.id))
with api_session(token3) as api:
res = api.ListFriendRequests(empty_pb2.Empty())
assert res.received[0].user_id == user1.id
fr_id = res.received[0].friend_request_id
api.RespondFriendRequest(
api_pb2.RespondFriendRequestReq(friend_request_id=fr_id,
accept=True))
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user2.id))
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user4.id))
with api_session(token5) as api:
res = api.ListFriendRequests(empty_pb2.Empty())
assert res.received[0].user_id == user1.id
fr_id = res.received[0].friend_request_id
api.RespondFriendRequest(
api_pb2.RespondFriendRequestReq(friend_request_id=fr_id,
accept=True))
with api_session(token2) as api:
res = api.ListFriendRequests(empty_pb2.Empty())
assert res.received[0].user_id == user1.id
fr_id = res.received[0].friend_request_id
api.RespondFriendRequest(
api_pb2.RespondFriendRequestReq(friend_request_id=fr_id,
accept=True))
assert res.received[1].user_id == user3.id
fr_id = res.received[1].friend_request_id
api.RespondFriendRequest(
api_pb2.RespondFriendRequestReq(friend_request_id=fr_id,
accept=True))
with api_session(token4) as api:
res = api.ListFriendRequests(empty_pb2.Empty())
assert res.received[0].user_id == user1.id
fr_id = res.received[0].friend_request_id
api.RespondFriendRequest(
api_pb2.RespondFriendRequestReq(friend_request_id=fr_id,
accept=True))
assert res.received[1].user_id == user3.id
fr_id = res.received[1].friend_request_id
api.RespondFriendRequest(
api_pb2.RespondFriendRequestReq(friend_request_id=fr_id,
accept=True))
api.SendFriendRequest(api_pb2.SendFriendRequestReq(user_id=user5.id))
with api_session(token1) as api:
res = api.GetUser(api_pb2.GetUserReq(user=str(user2.username)))
assert len(res.mutual_friends) == 1
assert res.mutual_friends[0].user_id == user3.id
# and other way around same
with api_session(token2) as api:
res = api.GetUser(api_pb2.GetUserReq(user=str(user1.username)))
assert len(res.mutual_friends) == 1
assert res.mutual_friends[0].user_id == user3.id
