109 lines
3.4 KiB
Python
109 lines
3.4 KiB
Python
|
import random
|
||
|
|
import pytest
|
||
|
|
import jsonschema
|
||
|
|
import json
|
||
|
|
import time
|
||
|
|
from conftest import option, user_1, user_2
|
||
|
|
from test_cases import RpcTestCase, TransactionTestCase
|
||
|
|
from dataclasses import dataclass, field
|
||
|
|
from typing import List, Optional
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.rpc
|
||
|
|
@pytest.mark.wakuext
|
||
|
|
class TestRpc(RpcTestCase):
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"method, params",
|
||
|
|
[
|
||
|
|
("wakuext_peers", []),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_(self, method, params):
|
||
|
|
_id = str(random.randint(1, 8888))
|
||
|
|
|
||
|
|
response = self.rpc_request(method, params, _id, url=option.rpc_url_2)
|
||
|
|
self.verify_is_valid_json_rpc_response(response)
|
||
|
|
self.verify_json_schema(response, method)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.rpc
|
||
|
|
@pytest.mark.accounts
|
||
|
|
@pytest.mark.wakuext
|
||
|
|
class TestRpcMessaging(RpcTestCase):
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class User:
|
||
|
|
rpc_url: str
|
||
|
|
chat_public_key: Optional[str] = None
|
||
|
|
chat_id: Optional[str] = None
|
||
|
|
|
||
|
|
def test_add_contact(self):
|
||
|
|
_id = str(random.randint(1, 8888))
|
||
|
|
|
||
|
|
self.user_1 = self.User(rpc_url=option.rpc_url)
|
||
|
|
self.user_2 = self.User(rpc_url=option.rpc_url_2)
|
||
|
|
|
||
|
|
# get chat public key
|
||
|
|
for user in self.user_1, self.user_2:
|
||
|
|
response = self.rpc_request(
|
||
|
|
"accounts_getAccounts", [], _id, url=user.rpc_url
|
||
|
|
)
|
||
|
|
self.verify_is_valid_json_rpc_response(response)
|
||
|
|
|
||
|
|
user.chat_public_key = next(
|
||
|
|
(
|
||
|
|
item["public-key"]
|
||
|
|
for item in response.json()["result"]
|
||
|
|
if item["chat"]
|
||
|
|
),
|
||
|
|
None,
|
||
|
|
)
|
||
|
|
|
||
|
|
# send contact requests
|
||
|
|
for sender in self.user_1, self.user_2:
|
||
|
|
for receiver in self.user_1, self.user_2:
|
||
|
|
if sender != receiver:
|
||
|
|
response = self.rpc_request(
|
||
|
|
method="wakuext_sendContactRequest",
|
||
|
|
params=[
|
||
|
|
{
|
||
|
|
"id": receiver.chat_public_key,
|
||
|
|
"message": f"contact request from {sender.chat_public_key}: sent at {time.time()}",
|
||
|
|
}
|
||
|
|
],
|
||
|
|
_id=99,
|
||
|
|
url=sender.rpc_url,
|
||
|
|
)
|
||
|
|
|
||
|
|
self.verify_is_valid_json_rpc_response(response)
|
||
|
|
sender.chat_id = response.json()["result"]["chats"][0]["lastMessage"]["id"]
|
||
|
|
|
||
|
|
# accept contact requests
|
||
|
|
for user in self.user_1, self.user_2:
|
||
|
|
response = self.rpc_request(
|
||
|
|
method="wakuext_acceptContactRequest",
|
||
|
|
params=[
|
||
|
|
{
|
||
|
|
"id": user.chat_id,
|
||
|
|
}
|
||
|
|
],
|
||
|
|
_id=99,
|
||
|
|
url=user.rpc_url,
|
||
|
|
)
|
||
|
|
self.verify_is_valid_json_rpc_response(response)
|
||
|
|
|
||
|
|
# verify contacts
|
||
|
|
for user in (self.user_1, self.user_2), (self.user_2, self.user_1):
|
||
|
|
response = self.rpc_request(
|
||
|
|
method="wakuext_contacts",
|
||
|
|
params=[],
|
||
|
|
_id=99,
|
||
|
|
url=user[0].rpc_url,
|
||
|
|
)
|
||
|
|
self.verify_is_valid_json_rpc_response(response)
|
||
|
|
|
||
|
|
response = response.json()
|
||
|
|
assert response["result"][0]["added"] == True
|
||
|
|
assert response["result"][0]["id"] == user[1].chat_public_key
|