Reckless-Satoshi/robosats
1
1
Fork 0
mirror of https://github.com/Reckless-Satoshi/robosats.git synced 2024-08-16 15:20:55 +03:00
Code Issues Projects Releases Wiki Activity

Add test publish and view orders

Browse Source
This commit is contained in:
Reckless_Satoshi 2023-11-09 10:33:53 +00:00 committed by Reckless_Satoshi
parent bece7c7d4e
commit edd9455d7b
3 changed files with 189 additions and 68 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/mocks/cln.py
View File

@ -4,7 +4,7 @@ from unittest.mock import MagicMock
class MockNodeStub:
def __init__(channel, other):
def __init__(self, channel):
pass
def Getinfo(self, request):
@ -34,7 +34,7 @@ class MockNodeStub:
class MockHoldStub:
def __init__(channel, other):
def __init__(self, channel):
pass
def HoldInvoiceLookup(self, request):

26
tests/mocks/lnd.py
View File

@ -4,6 +4,9 @@ from unittest.mock import MagicMock
class MockLightningStub:
def __init__(self, channel):
pass
def GetInfo(self, request):
response = MagicMock()
response.testnet = True
@ -35,6 +38,8 @@ class MockLightningStub:
response.payment_addr = '\275\205\224\016\363\325\262\201\306"8\022e\343\215\355\277\304\021\r\037l\202\023\314\353\334\265\002\036h\322'
response.num_msat = 1731000
return response
def CancelInvoice(self, request):
response = MagicMock()
if (
@ -76,12 +81,16 @@ class MockLightningStub:
class MockInvoicesStub:
def __init__(self, channel):
pass
def AddHoldInvoice(self, request):
response = MagicMock()
if request.value == 1731:
response.payment_request = "lntb17310n1pj552mdpp50p2utzh7mpsf3uq7u7cws4a96tj3kyq54hchdkpw8zecamx9klrqd2j2pshjmt9de6zqun9vejhyetwvdjn5gphxs6nsvfe893z6wphvfsj6dryvymj6wp5xvuz6wp5xcukvdec8yukgcf49cs9g6rfwvs8qcted4jkuapq2ay5cnpqgefy2326g5syjn3qt984253q2aq5cnz92skzqcmgv43kkgr0dcs9ymmzdafkzarnyp5kvgr5dpjjqmr0vd4jqampwvs8xatrvdjhxumxw4kzugzfwss8w6tvdssxyefqw4hxcmmrddjkggpgveskjmpfyp6kumr9wdejq7t0w5sxx6r9v96zqmmjyp3kzmnrv4kzqatwd9kxzar9wfskcmre9ccqz52xqzwzsp5hkzegrhn6kegr33z8qfxtcudaklugygdrakgyy7va0wt2qs7drfq9qyyssqc6rztchzl4m7mlulrhlcajszcl9fan8908k9n5x7gmz8g8d6ht5pj4l8r0dushq6j5s8x7yv9a5klz0kfxwy8v6ze6adyrrp4wu0q0sq3t604x"
response.add_index = 1
response.payment_addr = b'\275\205\224\016\363\325\262\201\306"8\022e\343\215\355\277\304\021\r\037l\202\023\314\353\334\265\002\036h\322'
# if request.value == 1731:
response.payment_request = "lntb17310n1pj552mdpp50p2utzh7mpsf3uq7u7cws4a96tj3kyq54hchdkpw8zecamx9klrqd2j2pshjmt9de6zqun9vejhyetwvdjn5gphxs6nsvfe893z6wphvfsj6dryvymj6wp5xvuz6wp5xcukvdec8yukgcf49cs9g6rfwvs8qcted4jkuapq2ay5cnpqgefy2326g5syjn3qt984253q2aq5cnz92skzqcmgv43kkgr0dcs9ymmzdafkzarnyp5kvgr5dpjjqmr0vd4jqampwvs8xatrvdjhxumxw4kzugzfwss8w6tvdssxyefqw4hxcmmrddjkggpgveskjmpfyp6kumr9wdejq7t0w5sxx6r9v96zqmmjyp3kzmnrv4kzqatwd9kxzar9wfskcmre9ccqz52xqzwzsp5hkzegrhn6kegr33z8qfxtcudaklugygdrakgyy7va0wt2qs7drfq9qyyssqc6rztchzl4m7mlulrhlcajszcl9fan8908k9n5x7gmz8g8d6ht5pj4l8r0dushq6j5s8x7yv9a5klz0kfxwy8v6ze6adyrrp4wu0q0sq3t604x"
response.add_index = 1
response.payment_addr = b'\275\205\224\016\363\325\262\201\306"8\022e\343\215\355\277\304\021\r\037l\202\023\314\353\334\265\002\036h\322'
return response
def CancelInvoice(self, request):
response = MagicMock()
@ -93,7 +102,12 @@ class MockInvoicesStub:
def LookupInvoiceV2(self, request):
response = MagicMock()
return response
if request.payment_hash == bytes.fromhex(
"7855c58afed86098f01ee7b0e857a5d2e51b1014adf176d82e38b38eecc5b7c6"
):
response.memo = "Payment reference: ..."
response.state = 3 # "ACCEPTED"
return response
class MockRouterStub:
@ -117,7 +131,7 @@ class MockSignerStub:
class MockVersionerStub:
def __init__(channel, other):
def __init__(self, channel):
pass
def GetVersion(self, request):

227
tests/test_trade_pipeline.py
View File

@ -7,15 +7,13 @@ from decouple import config
from django.contrib.auth.models import User
from django.test import Client, TestCase
from api.management.commands.follow_invoices import Command as FollowInvoices
from api.models import Currency, Order
from api.tasks import cache_market
from tests.mocks.cln import MockHoldStub, MockNodeStub
from tests.mocks.lnd import (
from tests.mocks.cln import MockHoldStub # , MockNodeStub
from tests.mocks.lnd import ( # MockRouterStub,; MockSignerStub,; MockVersionerStub,
MockInvoicesStub,
MockLightningStub,
MockRouterStub,
MockSignerStub,
MockVersionerStub,
)
@ -39,7 +37,7 @@ class TradeTest(TestCase):
response = self.client.post(path, data)
self.assertEqual(response.status_code, 302)
def get_robot_auth(self, robot_index):
def get_robot_auth(self, robot_index, first_encounter=False):
"""
Create an AUTH header that embeds token, pub_key, and enc_priv_key into a single string
as requested by the robosats token middleware.
@ -51,9 +49,14 @@ class TradeTest(TestCase):
with open(f"tests/robots/{robot_index}/enc_priv_key", "r") as file:
enc_priv_key = file.read()
headers = {
"HTTP_AUTHORIZATION": f"Token {b91_token} | Public {pub_key} | Private {enc_priv_key}"
}
# First time a robot authenticated, it is registered by the backend, so pub_key and enc_priv_key is needed
if first_encounter:
headers = {
"HTTP_AUTHORIZATION": f"Token {b91_token} | Public {pub_key} | Private {enc_priv_key}"
}
else:
headers = {"HTTP_AUTHORIZATION": f"Token {b91_token}"}
return headers, pub_key, enc_priv_key
def assert_robot(self, response, pub_key, enc_priv_key, expected_nickname):
@ -93,7 +96,7 @@ class TradeTest(TestCase):
Creates the robots in /tests/robots/{robot_index}
"""
path = "/api/robot/"
headers, pub_key, enc_priv_key = self.get_robot_auth(robot_index)
headers, pub_key, enc_priv_key = self.get_robot_auth(robot_index, True)
response = self.client.get(path, **headers)
@ -125,44 +128,34 @@ class TradeTest(TestCase):
usd.timestamp, datetime, "External price timestamp is not a datetime"
)
def test_create_order(
self,
robot_index=1,
payment_method="Advcash Cash F2F",
min_amount=21,
max_amount=101.7,
premium=3.34,
public_duration=69360,
escrow_duration=8700,
bond_size=3.5,
latitude=34.7455,
longitude=135.503,
):
def create_order(self, maker_form, robot_index=1):
# Requisites
# Cache market prices
self.test_cache_market()
path = "/api/make/"
# Get valid robot auth headers
headers, _, _ = self.get_robot_auth(robot_index)
headers, _, _ = self.get_robot_auth(robot_index, True)
# Prepare request body
response = self.client.post(path, maker_form, **headers)
return response
def test_create_order(self):
maker_form = {
"type": Order.Types.BUY,
"currency": 1,
"has_range": True,
"min_amount": min_amount,
"max_amount": max_amount,
"payment_method": payment_method,
"min_amount": 21,
"max_amount": 101.7,
"payment_method": "Advcash Cash F2F",
"is_explicit": False,
"premium": premium,
"public_duration": public_duration,
"escrow_duration": escrow_duration,
"bond_size": bond_size,
"latitude": latitude,
"longitude": longitude,
"premium": 3.34,
"public_duration": 69360,
"escrow_duration": 8700,
"bond_size": 3.5,
"latitude": 34.7455,
"longitude": 135.503,
}
response = self.client.post(path, maker_form, **headers)
response = self.create_order(maker_form, robot_index=1)
data = json.loads(response.content.decode())
# Checks
@ -191,32 +184,44 @@ class TradeTest(TestCase):
)
self.assertTrue(data["has_range"], "Order with range has a False has_range")
self.assertAlmostEqual(
float(data["min_amount"]), min_amount, "Order min amount does not match"
float(data["min_amount"]),
maker_form["min_amount"],
"Order min amount does not match",
)
self.assertAlmostEqual(
float(data["max_amount"]), max_amount, "Order max amount does not match"
float(data["max_amount"]),
maker_form["max_amount"],
"Order max amount does not match",
)
self.assertEqual(
data["payment_method"],
payment_method,
maker_form["payment_method"],
"Order payment method does not match",
)
self.assertEqual(
data["escrow_duration"],
escrow_duration,
maker_form["escrow_duration"],
"Order escrow duration does not match",
)
self.assertAlmostEqual(
float(data["bond_size"]), bond_size, "Order bond size does not match"
float(data["bond_size"]),
maker_form["bond_size"],
"Order bond size does not match",
)
self.assertAlmostEqual(
float(data["latitude"]), latitude, "Order latitude does not match"
float(data["latitude"]),
maker_form["latitude"],
"Order latitude does not match",
)
self.assertAlmostEqual(
float(data["longitude"]), longitude, "Order longitude does not match"
float(data["longitude"]),
maker_form["longitude"],
"Order longitude does not match",
)
self.assertAlmostEqual(
float(data["premium"]), premium, "Order premium does not match"
float(data["premium"]),
maker_form["premium"],
"Order premium does not match",
)
self.assertFalse(
data["is_explicit"], "Relative pricing order has True is_explicit"
@ -226,23 +231,125 @@ class TradeTest(TestCase):
)
self.assertIsNone(data["taker"], "New order's taker is not null")
@patch("api.lightning.cln.node_pb2_grpc.NodeStub", MockNodeStub)
return data
@patch("api.lightning.cln.hold_pb2_grpc.HoldStub", MockHoldStub)
@patch("api.lightning.lnd.verrpc_pb2_grpc.VersionerStub", MockVersionerStub)
@patch("api.lightning.lnd.lightning_pb2_grpc.LightningStub", MockLightningStub)
@patch("api.lightning.lnd.invoices_pb2_grpc.InvoicesStub", MockInvoicesStub)
@patch("api.lightning.lnd.router_pb2_grpc.RouterStub", MockRouterStub)
@patch("api.lightning.lnd.signer_pb2_grpc.SignerStub", MockSignerStub)
def test_maker_bond_locked(self):
self.test_create_order(
robot_index=1,
payment_method="Cash F2F",
min_amount=80,
max_amount=500,
premium=5,
public_duration=86000,
escrow_duration=8000,
bond_size=2,
latitude=0,
longitude=0,
def get_order(self, order_id, robot_index=1, first_encounter=False):
path = "/api/order/"
params = f"?order_id={order_id}"
headers, _, _ = self.get_robot_auth(robot_index, first_encounter)
response = self.client.get(path + params, **headers)
return response
def test_get_order_created(self):
# Make an order
maker_form = {
"type": Order.Types.BUY,
"currency": 1,
"has_range": True,
"min_amount": 21,
"max_amount": 101.7,
"payment_method": "Advcash Cash F2F",
"is_explicit": False,
"premium": 3.34,
"public_duration": 69360,
"escrow_duration": 8700,
"bond_size": 3.5,
"latitude": 34.7455,
"longitude": 135.503,
}
order_made_response = self.create_order(maker_form, robot_index=1)
order_made_data = json.loads(order_made_response.content.decode())
# Maker's first order fetch. Should trigger maker bond hold invoice generation.
response = self.get_order(order_made_data["id"])
data = json.loads(response.content.decode())
self.assertEqual(response.status_code, 200)
self.assertEqual(data["id"], order_made_data["id"])
self.assertTrue(
isinstance(datetime.fromisoformat(data["created_at"]), datetime)
)
self.assertTrue(
isinstance(datetime.fromisoformat(data["expires_at"]), datetime)
)
self.assertTrue(data["is_maker"])
self.assertTrue(data["is_participant"])
self.assertTrue(data["is_buyer"])
self.assertFalse(data["is_seller"])
self.assertEqual(data["maker_status"], "Active")
self.assertEqual(data["status_message"], Order.Status(Order.Status.WFB).label)
self.assertFalse(data["is_fiat_sent"])
self.assertFalse(data["is_disputed"])
self.assertEqual(data["ur_nick"], "MyopicRacket333")
self.assertTrue(isinstance(data["satoshis_now"], int))
self.assertFalse(data["maker_locked"])
self.assertFalse(data["taker_locked"])
self.assertFalse(data["escrow_locked"])
self.assertEqual(
data["bond_invoice"],
"lntb17310n1pj552mdpp50p2utzh7mpsf3uq7u7cws4a96tj3kyq54hchdkpw8zecamx9klrqd2j2pshjmt9de6zqun9vejhyetwvdjn5gphxs6nsvfe893z6wphvfsj6dryvymj6wp5xvuz6wp5xcukvdec8yukgcf49cs9g6rfwvs8qcted4jkuapq2ay5cnpqgefy2326g5syjn3qt984253q2aq5cnz92skzqcmgv43kkgr0dcs9ymmzdafkzarnyp5kvgr5dpjjqmr0vd4jqampwvs8xatrvdjhxumxw4kzugzfwss8w6tvdssxyefqw4hxcmmrddjkggpgveskjmpfyp6kumr9wdejq7t0w5sxx6r9v96zqmmjyp3kzmnrv4kzqatwd9kxzar9wfskcmre9ccqz52xqzwzsp5hkzegrhn6kegr33z8qfxtcudaklugygdrakgyy7va0wt2qs7drfq9qyyssqc6rztchzl4m7mlulrhlcajszcl9fan8908k9n5x7gmz8g8d6ht5pj4l8r0dushq6j5s8x7yv9a5klz0kfxwy8v6ze6adyrrp4wu0q0sq3t604x",
)
self.assertTrue(isinstance(data["bond_satoshis"], int))
@patch("api.lightning.lnd.invoices_pb2_grpc.InvoicesStub", MockInvoicesStub)
def check_for_locked_bonds(self):
# A background thread checks every 5 second the status of invoices. We invoke directly during test.
# It will ask LND via gRPC. In our test, the request/response from LND is mocked, and it will return fake invoice status "ACCEPTED"
follow_invoices = FollowInvoices()
follow_invoices.follow_hold_invoices()
def create_and_publish_order(self, maker_form, robot_index=1):
# Make an order
order_made_response = self.create_order(maker_form, robot_index=1)
order_made_data = json.loads(order_made_response.content.decode())
# Maker's first order fetch. Should trigger maker bond hold invoice generation.
self.get_order(order_made_data["id"])
# Check for invoice locked (the mocked LND will return ACCEPTED)
self.check_for_locked_bonds()
# Get order
response = self.get_order(order_made_data["id"])
return response
def test_publish_order(self):
maker_form = {
"type": Order.Types.BUY,
"currency": 1,
"has_range": True,
"min_amount": 21,
"max_amount": 101.7,
"payment_method": "Advcash Cash F2F",
"is_explicit": False,
"premium": 3.34,
"public_duration": 69360,
"escrow_duration": 8700,
"bond_size": 3.5,
"latitude": 34.7455,
"longitude": 135.503,
}
# Get order
response = self.create_and_publish_order(maker_form)
data = json.loads(response.content.decode())
self.assertEqual(response.status_code, 200)
self.assertEqual(data["id"], data["id"])
self.assertEqual(data["status_message"], Order.Status(Order.Status.PUB).label)
self.assertTrue(data["maker_locked"])
self.assertFalse(data["taker_locked"])
self.assertFalse(data["escrow_locked"])
# Test what we can see with newly created robot 2 (only for public status)
public_response = self.get_order(
data["id"], robot_index=2, first_encounter=True
)
public_data = json.loads(public_response.content.decode())
self.assertFalse(public_data["is_participant"])
self.assertTrue(isinstance(public_data["price_now"], float))
self.assertTrue(isinstance(data["satoshis_now"], int))