151 lines
4.7 KiB
Python
151 lines
4.7 KiB
Python
import json
|
|
import websocket
|
|
import threading
|
|
import logging
|
|
import jsonschema
|
|
import requests
|
|
from conftest import option, user_1, user_2
|
|
|
|
|
|
class RpcTestCase:
|
|
|
|
def setup_method(self):
|
|
self.network_id = 31337
|
|
|
|
def _try_except_JSONDecodeError_KeyError(self, response, key: str):
|
|
try:
|
|
return response.json()[key]
|
|
except json.JSONDecodeError:
|
|
raise AssertionError(f"Invalid JSON in response: {response.content}")
|
|
except KeyError:
|
|
raise AssertionError(f"Key '{key}' not found in the JSON response.")
|
|
|
|
def verify_is_valid_json_rpc_response(self, response, _id=None):
|
|
assert response.status_code == 200
|
|
assert response.content
|
|
self._try_except_JSONDecodeError_KeyError(response, "result")
|
|
|
|
if _id:
|
|
try:
|
|
if _id != response.json()["id"]:
|
|
raise AssertionError(
|
|
f"got id: {response.json()['id']} instead of expected id: {_id}"
|
|
)
|
|
except KeyError:
|
|
raise AssertionError(f"no id in response {response.json()}")
|
|
return response
|
|
|
|
def verify_is_json_rpc_error(self, response):
|
|
assert response.status_code == 200
|
|
assert response.content
|
|
self._try_except_JSONDecodeError_KeyError(response, "error")
|
|
|
|
|
|
def rpc_request(self, method, params=[], _id=None, client=None, url=None):
|
|
client = client if client else requests.Session()
|
|
url = url if url else option.rpc_url
|
|
|
|
data = {"jsonrpc": "2.0", "method": method}
|
|
if params:
|
|
data["params"] = params
|
|
data["id"] = _id if _id else 13
|
|
|
|
response = client.post(url, json=data)
|
|
|
|
return response
|
|
|
|
def verify_json_schema(self, response, method):
|
|
with open(f"{option.base_dir}/schemas/{method}", "r") as schema:
|
|
jsonschema.validate(instance=response.json(),
|
|
schema=json.load(schema))
|
|
|
|
|
|
|
|
class TransactionTestCase(RpcTestCase):
|
|
|
|
|
|
def wallet_create_multi_transaction(self, **kwargs):
|
|
method = "wallet_createMultiTransaction"
|
|
transferTx_data = {
|
|
"data": "",
|
|
"from": user_1.address,
|
|
"gas": "0x5BBF",
|
|
"input": "",
|
|
"maxFeePerGas": "0xbcc0f04fd",
|
|
"maxPriorityFeePerGas": "0xbcc0f04fd",
|
|
"to": user_2.address,
|
|
"type": "0x02",
|
|
"value": "0x5af3107a4000",
|
|
}
|
|
for key, new_value in kwargs.items():
|
|
if key in transferTx_data:
|
|
transferTx_data[key] = new_value
|
|
else:
|
|
print(f"Warning: The key '{key}' does not exist in the transferTx parameters and will be ignored.")
|
|
params = [
|
|
{
|
|
"fromAddress": user_1.address,
|
|
"fromAmount": "0x5af3107a4000",
|
|
"fromAsset": "ETH",
|
|
"multiTxType": "MultiTransactionSend",
|
|
"toAddress": user_2.address,
|
|
"toAsset": "ETH",
|
|
},
|
|
[
|
|
{
|
|
"bridgeName": "Transfer",
|
|
"chainID": 31337,
|
|
"transferTx": transferTx_data
|
|
}
|
|
],
|
|
f"{option.password}",
|
|
]
|
|
response = self.rpc_request(method, params, 13)
|
|
return response
|
|
|
|
def setup_method(self):
|
|
super().setup_method()
|
|
|
|
response = self.wallet_create_multi_transaction()
|
|
try:
|
|
self.tx_hash = response.json(
|
|
)["result"]["hashes"][str(self.network_id)][0]
|
|
except (KeyError, json.JSONDecodeError):
|
|
raise Exception(response.content)
|
|
|
|
|
|
class SignalTestCase(RpcTestCase):
|
|
|
|
received_signals = []
|
|
|
|
def _on_message(self, ws, signal):
|
|
self.received_signals.append(signal)
|
|
|
|
def _on_error(self, ws, error):
|
|
logging.info(f"Error: {error}")
|
|
|
|
def _on_close(self, ws, close_status_code, close_msg):
|
|
logging.info(f"Connection closed: {close_status_code}, {close_msg}")
|
|
|
|
def _on_open(self, ws):
|
|
logging.info("Connection opened")
|
|
|
|
def _connect(self):
|
|
self.url = f"{option.ws_url}/signals"
|
|
|
|
ws = websocket.WebSocketApp(self.url,
|
|
on_message=self._on_message,
|
|
on_error=self._on_error,
|
|
on_close=self._on_close)
|
|
|
|
ws.on_open = self._on_open
|
|
|
|
ws.run_forever()
|
|
|
|
def setup_method(self):
|
|
super().setup_method()
|
|
|
|
websocket_thread = threading.Thread(target=self._connect)
|
|
websocket_thread.daemon = True
|
|
websocket_thread.start()
|