mirror of
https://github.com/Reckless-Satoshi/robosats.git
synced 2024-09-11 11:55:26 +03:00
853 lines
30 KiB
Python
853 lines
30 KiB
Python
import json
|
|
import random
|
|
from datetime import datetime
|
|
from decimal import Decimal
|
|
from unittest.mock import patch
|
|
|
|
from decouple import config
|
|
from django.contrib.auth.models import User
|
|
from django.urls import reverse
|
|
|
|
from api.management.commands.follow_invoices import Command as FollowInvoices
|
|
from api.models import Currency, Order
|
|
from api.tasks import cache_market, follow_send_payment
|
|
from control.models import BalanceLog
|
|
from control.tasks import compute_node_balance
|
|
from tests.node_utils import (
|
|
add_invoice,
|
|
create_address,
|
|
pay_invoice,
|
|
set_up_regtest_network,
|
|
)
|
|
from tests.pgp_utils import sign_message
|
|
from tests.test_api import BaseAPITestCase
|
|
|
|
|
|
def read_file(file_path):
|
|
"""
|
|
Read a file and return its content.
|
|
"""
|
|
with open(file_path, "r") as file:
|
|
return file.read()
|
|
|
|
|
|
class TradeTest(BaseAPITestCase):
|
|
su_pass = "12345678"
|
|
su_name = config("ESCROW_USERNAME", cast=str, default="admin")
|
|
|
|
maker_form_buy_with_range = {
|
|
"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,
|
|
}
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
"""
|
|
Set up initial data for the test case.
|
|
"""
|
|
# Create super user
|
|
User.objects.create_superuser(cls.su_name, "super@user.com", cls.su_pass)
|
|
|
|
# Fetch currency prices from external APIs
|
|
cache_market()
|
|
|
|
# Initialize bitcoin core, mine some blocks, connect nodes, open channel
|
|
set_up_regtest_network()
|
|
|
|
# Take the first node balances snapshot
|
|
compute_node_balance()
|
|
|
|
def test_login_superuser(self):
|
|
"""
|
|
Test the login functionality for the superuser.
|
|
"""
|
|
path = reverse("admin:login")
|
|
data = {"username": self.su_name, "password": self.su_pass}
|
|
response = self.client.post(path, data)
|
|
self.assertEqual(response.status_code, 302)
|
|
self.assertResponse(
|
|
response
|
|
) # should skip given that /coordinator/login is not documented
|
|
|
|
def test_cache_market(self):
|
|
"""
|
|
Test if the cache_market() call during test setup worked
|
|
"""
|
|
usd = Currency.objects.get(id=1)
|
|
self.assertIsInstance(
|
|
usd.exchange_rate,
|
|
Decimal,
|
|
f"Exchange rate is not a Decimal. Got {type(usd.exchange_rate)}",
|
|
)
|
|
self.assertGreater(
|
|
usd.exchange_rate, 0, "Exchange rate is not higher than zero"
|
|
)
|
|
self.assertIsInstance(
|
|
usd.timestamp, datetime, "External price timestamp is not a datetime"
|
|
)
|
|
|
|
def test_initial_balance_log(self):
|
|
"""
|
|
Test if the initial node BalanceLog is correct.
|
|
One channel should exist with 0.5BTC in local.
|
|
No onchain balance should exist.
|
|
"""
|
|
balance_log = BalanceLog.objects.latest()
|
|
|
|
self.assertIsInstance(balance_log.time, datetime)
|
|
self.assertTrue(balance_log.total > 0)
|
|
self.assertTrue(balance_log.ln_local > 0)
|
|
self.assertTrue(balance_log.ln_local_unsettled >= 0)
|
|
self.assertTrue(balance_log.ln_remote > 0)
|
|
self.assertEqual(balance_log.ln_remote_unsettled, 0)
|
|
self.assertTrue(balance_log.onchain_total > 0)
|
|
self.assertTrue(balance_log.onchain_confirmed > 0)
|
|
self.assertEqual(balance_log.onchain_unconfirmed, 0)
|
|
self.assertTrue(balance_log.onchain_fraction > 0)
|
|
|
|
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.
|
|
"""
|
|
|
|
b91_token = read_file(f"tests/robots/{robot_index}/b91_token")
|
|
pub_key = read_file(f"tests/robots/{robot_index}/pub_key")
|
|
enc_priv_key = read_file(f"tests/robots/{robot_index}/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
|
|
|
|
def assert_robot(self, response, robot_index):
|
|
"""
|
|
Assert that the robot is created correctly.
|
|
"""
|
|
nickname = read_file(f"tests/robots/{robot_index}/nickname")
|
|
pub_key = read_file(f"tests/robots/{robot_index}/pub_key")
|
|
enc_priv_key = read_file(f"tests/robots/{robot_index}/enc_priv_key")
|
|
|
|
data = json.loads(response.content.decode())
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertResponse(response)
|
|
|
|
self.assertEqual(
|
|
data["nickname"],
|
|
nickname,
|
|
f"Robot created nickname is not {nickname}",
|
|
)
|
|
self.assertEqual(
|
|
data["public_key"], pub_key, "Returned public Kky does not match"
|
|
)
|
|
self.assertEqual(
|
|
data["encrypted_private_key"],
|
|
enc_priv_key,
|
|
"Returned encrypted private key does not match",
|
|
)
|
|
self.assertEqual(
|
|
len(data["tg_token"]), 15, "String is not exactly 15 characters long"
|
|
)
|
|
self.assertEqual(
|
|
data["tg_bot_name"],
|
|
config(
|
|
"TELEGRAM_BOT_NAME", cast=str, default="RoboCoordinatorNotificationBot"
|
|
),
|
|
"Telegram bot name is not correct",
|
|
)
|
|
self.assertFalse(
|
|
data["tg_enabled"], "The new robot's telegram seems to be enabled"
|
|
)
|
|
self.assertEqual(data["earned_rewards"], 0, "The new robot's rewards are not 0")
|
|
|
|
def create_robot(self, robot_index):
|
|
"""
|
|
Creates the robots in /tests/robots/{robot_index}
|
|
"""
|
|
path = reverse("robot")
|
|
headers = self.get_robot_auth(robot_index, True)
|
|
|
|
return self.client.get(path, **headers)
|
|
|
|
def test_create_robots(self):
|
|
"""
|
|
Test the creation of two robots to be used in the trade tests
|
|
"""
|
|
for robot_index in [1, 2]:
|
|
response = self.create_robot(robot_index)
|
|
self.assert_robot(response, robot_index)
|
|
|
|
def make_order(self, maker_form, robot_index=1):
|
|
"""
|
|
Create an order for the test.
|
|
"""
|
|
path = reverse("make")
|
|
# Get valid robot auth headers
|
|
headers = self.get_robot_auth(robot_index, True)
|
|
|
|
response = self.client.post(path, maker_form, **headers)
|
|
return response
|
|
|
|
def test_make_order(self):
|
|
"""
|
|
Test the creation of an order.
|
|
"""
|
|
maker_form = self.maker_form_buy_with_range
|
|
response = self.make_order(maker_form, robot_index=1)
|
|
data = json.loads(response.content.decode())
|
|
|
|
# Checks
|
|
self.assertResponse(response)
|
|
|
|
self.assertIsInstance(data["id"], int, "Order ID is not an integer")
|
|
self.assertEqual(
|
|
data["status"],
|
|
Order.Status.WFB,
|
|
"Newly created order status is not 'Waiting for maker bond'",
|
|
)
|
|
self.assertIsInstance(
|
|
datetime.fromisoformat(data["created_at"]),
|
|
datetime,
|
|
"Order creation timestamp is not datetime",
|
|
)
|
|
self.assertIsInstance(
|
|
datetime.fromisoformat(data["expires_at"]),
|
|
datetime,
|
|
"Order expiry time is not datetime",
|
|
)
|
|
self.assertEqual(
|
|
data["type"], Order.Types.BUY, "Buy order is not of type value BUY"
|
|
)
|
|
self.assertEqual(data["currency"], 1, "Order for USD is not of currency USD")
|
|
self.assertIsNone(
|
|
data["amount"], "Order with range has a non-null simple amount"
|
|
)
|
|
self.assertTrue(data["has_range"], "Order with range has a False has_range")
|
|
self.assertAlmostEqual(
|
|
float(data["min_amount"]),
|
|
maker_form["min_amount"],
|
|
"Order min amount does not match",
|
|
)
|
|
self.assertAlmostEqual(
|
|
float(data["max_amount"]),
|
|
maker_form["max_amount"],
|
|
"Order max amount does not match",
|
|
)
|
|
self.assertEqual(
|
|
data["payment_method"],
|
|
maker_form["payment_method"],
|
|
"Order payment method does not match",
|
|
)
|
|
self.assertEqual(
|
|
data["escrow_duration"],
|
|
maker_form["escrow_duration"],
|
|
"Order escrow duration does not match",
|
|
)
|
|
self.assertAlmostEqual(
|
|
float(data["bond_size"]),
|
|
maker_form["bond_size"],
|
|
"Order bond size does not match",
|
|
)
|
|
self.assertAlmostEqual(
|
|
float(data["latitude"]),
|
|
maker_form["latitude"],
|
|
"Order latitude does not match",
|
|
)
|
|
self.assertAlmostEqual(
|
|
float(data["longitude"]),
|
|
maker_form["longitude"],
|
|
"Order longitude does not match",
|
|
)
|
|
self.assertAlmostEqual(
|
|
float(data["premium"]),
|
|
maker_form["premium"],
|
|
"Order premium does not match",
|
|
)
|
|
self.assertFalse(
|
|
data["is_explicit"], "Relative pricing order has True is_explicit"
|
|
)
|
|
self.assertIsNone(
|
|
data["satoshis"], "Relative pricing order has non-null Satoshis"
|
|
)
|
|
self.assertIsNone(data["taker"], "New order's taker is not null")
|
|
|
|
return data
|
|
|
|
def get_order(self, order_id, robot_index=1, first_encounter=False):
|
|
path = reverse("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 cancel_order(self, order_id, robot_index=1):
|
|
path = reverse("order")
|
|
params = f"?order_id={order_id}"
|
|
headers = self.get_robot_auth(robot_index)
|
|
body = {"action": "cancel"}
|
|
response = self.client.post(path + params, body, **headers)
|
|
|
|
return response
|
|
|
|
def pause_order(self, order_id, robot_index=1):
|
|
path = reverse("order")
|
|
params = f"?order_id={order_id}"
|
|
headers = self.get_robot_auth(robot_index)
|
|
body = {"action": "pause"}
|
|
response = self.client.post(path + params, body, **headers)
|
|
|
|
return response
|
|
|
|
def test_get_order_created(self):
|
|
"""
|
|
Tests the creation of an order and the first request to see details,
|
|
including, the creation of the maker bond invoice.
|
|
"""
|
|
maker_form = self.maker_form_buy_with_range
|
|
robot_index = 1
|
|
|
|
order_made_response = self.make_order(maker_form, robot_index)
|
|
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.assertResponse(response)
|
|
|
|
self.assertEqual(data["id"], order_made_data["id"])
|
|
self.assertIsInstance(datetime.fromisoformat(data["created_at"]), datetime)
|
|
self.assertIsInstance(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"], read_file(f"tests/robots/{robot_index}/nickname")
|
|
)
|
|
self.assertIsInstance(data["satoshis_now"], int)
|
|
self.assertFalse(data["maker_locked"])
|
|
self.assertFalse(data["taker_locked"])
|
|
self.assertFalse(data["escrow_locked"])
|
|
self.assertIsInstance(data["bond_satoshis"], int)
|
|
|
|
# Cancel order to avoid leaving pending HTLCs after a successful test
|
|
self.cancel_order(data["id"])
|
|
|
|
def follow_hold_invoices(self):
|
|
# A background thread checks every 5 second the status of invoices. We invoke directly during test.
|
|
follow_invoices = FollowInvoices()
|
|
follow_invoices.follow_hold_invoices()
|
|
|
|
def send_payments(self):
|
|
# A background thread checks every 5 second whether there are outgoing payments. We invoke directly during test.
|
|
follow_invoices = FollowInvoices()
|
|
follow_invoices.send_payments()
|
|
|
|
def make_and_publish_order(self, maker_form, robot_index=1):
|
|
# Make an order
|
|
order_made_response = self.make_order(maker_form, robot_index)
|
|
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"])
|
|
invoice = response.json()["bond_invoice"]
|
|
|
|
# Lock the invoice from the robot's node
|
|
pay_invoice("robot", invoice)
|
|
|
|
# Check for invoice locked (the mocked LND will return ACCEPTED)
|
|
self.follow_hold_invoices()
|
|
|
|
# Get order
|
|
response = self.get_order(order_made_data["id"])
|
|
return response
|
|
|
|
def test_publish_order(self):
|
|
"""
|
|
Tests a trade from order creation to published (maker bond locked).
|
|
"""
|
|
maker_form = self.maker_form_buy_with_range
|
|
# Get order
|
|
response = self.make_and_publish_order(maker_form)
|
|
data = json.loads(response.content.decode())
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertResponse(response)
|
|
|
|
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.assertIsInstance(public_data["price_now"], float)
|
|
self.assertIsInstance(data["satoshis_now"], int)
|
|
|
|
# Cancel order to avoid leaving pending HTLCs after a successful test
|
|
self.cancel_order(data["id"])
|
|
|
|
def test_pause_unpause_order(self):
|
|
"""
|
|
Tests pausing and unpausing a public order
|
|
"""
|
|
maker_form = self.maker_form_buy_with_range
|
|
# Get order
|
|
response = self.make_and_publish_order(maker_form)
|
|
|
|
# PAUSE
|
|
response = self.pause_order(response.json()["id"])
|
|
data = response.json()
|
|
|
|
self.assertResponse(response)
|
|
self.assertEqual(data["status_message"], Order.Status(Order.Status.PAU).label)
|
|
|
|
# UNPAUSE
|
|
response = self.pause_order(response.json()["id"])
|
|
data = response.json()
|
|
|
|
self.assertResponse(response)
|
|
self.assertEqual(data["status_message"], Order.Status(Order.Status.PUB).label)
|
|
|
|
# Cancel order to avoid leaving pending HTLCs after a successful test
|
|
self.cancel_order(data["id"])
|
|
|
|
def take_order(self, order_id, amount, robot_index=2):
|
|
path = reverse("order")
|
|
params = f"?order_id={order_id}"
|
|
headers = self.get_robot_auth(robot_index, first_encounter=True)
|
|
body = {"action": "take", "amount": amount}
|
|
response = self.client.post(path + params, body, **headers)
|
|
|
|
return response
|
|
|
|
def make_and_take_order(
|
|
self, maker_form, take_amount=80, maker_index=1, taker_index=2
|
|
):
|
|
response_published = self.make_and_publish_order(maker_form, maker_index)
|
|
data_publised = json.loads(response_published.content.decode())
|
|
response = self.take_order(data_publised["id"], take_amount, taker_index)
|
|
return response
|
|
|
|
def test_make_and_take_order(self):
|
|
"""
|
|
Tests a trade from order creation to taken.
|
|
"""
|
|
maker_index = 1
|
|
taker_index = 2
|
|
maker_form = self.maker_form_buy_with_range
|
|
|
|
response = self.make_and_take_order(maker_form, 80, maker_index, taker_index)
|
|
data = json.loads(response.content.decode())
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertResponse(response)
|
|
|
|
self.assertEqual(data["status_message"], Order.Status(Order.Status.TAK).label)
|
|
self.assertEqual(
|
|
data["ur_nick"], read_file(f"tests/robots/{taker_index}/nickname")
|
|
)
|
|
self.assertEqual(
|
|
data["taker_nick"], read_file(f"tests/robots/{taker_index}/nickname")
|
|
)
|
|
self.assertEqual(
|
|
data["maker_nick"], read_file(f"tests/robots/{maker_index}/nickname")
|
|
)
|
|
self.assertEqual(data["maker_status"], "Active")
|
|
self.assertEqual(data["taker_status"], "Active")
|
|
self.assertFalse(data["is_maker"])
|
|
self.assertFalse(data["is_buyer"])
|
|
self.assertTrue(data["is_seller"])
|
|
self.assertTrue(data["is_taker"])
|
|
self.assertTrue(data["is_participant"])
|
|
self.assertTrue(data["maker_locked"])
|
|
self.assertFalse(data["taker_locked"])
|
|
self.assertFalse(data["escrow_locked"])
|
|
|
|
# Cancel order to avoid leaving pending HTLCs after a successful test
|
|
self.cancel_order(data["id"])
|
|
|
|
def make_and_lock_contract(
|
|
self, maker_form, take_amount=80, maker_index=1, taker_index=2
|
|
):
|
|
# Make an order
|
|
order_taken_response = self.make_and_take_order(
|
|
maker_form, take_amount, maker_index, taker_index
|
|
)
|
|
order_taken_data = json.loads(order_taken_response.content.decode())
|
|
|
|
# Maker's first order fetch. Should trigger maker bond hold invoice generation.
|
|
response = self.get_order(order_taken_data["id"], taker_index)
|
|
invoice = response.json()["bond_invoice"]
|
|
|
|
# Lock the invoice from the robot's node
|
|
pay_invoice("robot", invoice)
|
|
|
|
# Check for invoice locked (the mocked LND will return ACCEPTED)
|
|
self.follow_hold_invoices()
|
|
|
|
# Get order
|
|
response = self.get_order(order_taken_data["id"], taker_index)
|
|
return response
|
|
|
|
def test_make_and_lock_contract(self):
|
|
"""
|
|
Tests a trade from order creation to taker bond locked.
|
|
"""
|
|
maker_index = 1
|
|
taker_index = 2
|
|
maker_form = self.maker_form_buy_with_range
|
|
|
|
# Taker GET
|
|
response = self.make_and_lock_contract(maker_form, 80, maker_index, taker_index)
|
|
data = json.loads(response.content.decode())
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertResponse(response)
|
|
|
|
self.assertEqual(data["status_message"], Order.Status(Order.Status.WF2).label)
|
|
self.assertEqual(data["maker_status"], "Active")
|
|
self.assertEqual(data["taker_status"], "Active")
|
|
self.assertTrue(data["is_participant"])
|
|
self.assertTrue(data["maker_locked"])
|
|
self.assertTrue(data["taker_locked"])
|
|
self.assertFalse(data["escrow_locked"])
|
|
|
|
# Maker GET
|
|
response = self.get_order(data["id"], maker_index)
|
|
data = json.loads(response.content.decode())
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertResponse(response)
|
|
|
|
self.assertEqual(data["status_message"], Order.Status(Order.Status.WF2).label)
|
|
self.assertTrue(data["swap_allowed"])
|
|
self.assertIsInstance(data["suggested_mining_fee_rate"], float)
|
|
self.assertIsInstance(data["swap_fee_rate"], float)
|
|
self.assertTrue(data["suggested_mining_fee_rate"] > 0)
|
|
self.assertTrue(data["swap_fee_rate"] > 0)
|
|
self.assertEqual(data["maker_status"], "Active")
|
|
self.assertEqual(data["taker_status"], "Active")
|
|
self.assertTrue(data["is_participant"])
|
|
self.assertTrue(data["maker_locked"])
|
|
self.assertTrue(data["taker_locked"])
|
|
self.assertFalse(data["escrow_locked"])
|
|
|
|
# Cancel order to avoid leaving pending HTLCs after a successful test
|
|
self.cancel_order(data["id"])
|
|
|
|
def trade_to_locked_escrow(
|
|
self, maker_form, take_amount=80, maker_index=1, taker_index=2
|
|
):
|
|
# Make an order
|
|
locked_taker_response = self.make_and_lock_contract(
|
|
maker_form, take_amount, maker_index, taker_index
|
|
)
|
|
locked_taker_response_data = json.loads(locked_taker_response.content.decode())
|
|
|
|
# Maker's first order fetch. Should trigger maker bond hold invoice generation.
|
|
response = self.get_order(locked_taker_response_data["id"], taker_index)
|
|
invoice = response.json()["escrow_invoice"]
|
|
|
|
# Lock the invoice from the robot's node
|
|
pay_invoice("robot", invoice)
|
|
|
|
# Check for invoice locked (the mocked LND will return ACCEPTED)
|
|
self.follow_hold_invoices()
|
|
|
|
# Get order
|
|
response = self.get_order(locked_taker_response_data["id"], taker_index)
|
|
return response
|
|
|
|
def test_trade_to_locked_escrow(self):
|
|
"""
|
|
Tests a trade from order creation until escrow locked, before
|
|
invoice/address is submitted by buyer.
|
|
"""
|
|
maker_index = 1
|
|
taker_index = 2
|
|
maker_form = self.maker_form_buy_with_range
|
|
|
|
response = self.trade_to_locked_escrow(maker_form, 80, maker_index, taker_index)
|
|
data = json.loads(response.content.decode())
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertResponse(response)
|
|
|
|
self.assertEqual(data["status_message"], Order.Status(Order.Status.WFI).label)
|
|
self.assertTrue(data["maker_locked"])
|
|
self.assertTrue(data["taker_locked"])
|
|
self.assertTrue(data["escrow_locked"])
|
|
|
|
# Cancel order to avoid leaving pending HTLCs after a successful test
|
|
self.cancel_order(data["id"], 2)
|
|
|
|
def submit_payout_address(self, order_id, robot_index=1):
|
|
path = reverse("order")
|
|
params = f"?order_id={order_id}"
|
|
headers = self.get_robot_auth(robot_index)
|
|
|
|
payout_address = create_address("robot")
|
|
signed_payout_address = sign_message(
|
|
payout_address,
|
|
passphrase_path=f"tests/robots/{robot_index}/token",
|
|
private_key_path=f"tests/robots/{robot_index}/enc_priv_key",
|
|
)
|
|
body = {
|
|
"action": "update_address",
|
|
"address": signed_payout_address,
|
|
"mining_fee_rate": 50,
|
|
}
|
|
response = self.client.post(path + params, body, **headers)
|
|
|
|
return response
|
|
|
|
def trade_to_submitted_address(
|
|
self, maker_form, take_amount=80, maker_index=1, taker_index=2
|
|
):
|
|
response_escrow_locked = self.trade_to_locked_escrow(
|
|
maker_form, take_amount, maker_index, taker_index
|
|
)
|
|
response = self.submit_payout_address(
|
|
response_escrow_locked.json()["id"], maker_index
|
|
)
|
|
return response
|
|
|
|
def test_trade_to_submitted_address(self):
|
|
"""
|
|
Tests a trade from order creation until escrow locked and
|
|
address is submitted by buyer.
|
|
"""
|
|
maker_index = 1
|
|
taker_index = 2
|
|
maker_form = self.maker_form_buy_with_range
|
|
take_amount = round(
|
|
random.uniform(maker_form["min_amount"], maker_form["max_amount"]), 2
|
|
)
|
|
|
|
response = self.trade_to_submitted_address(
|
|
maker_form, take_amount, maker_index, taker_index
|
|
)
|
|
data = response.json()
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertResponse(response)
|
|
|
|
self.assertEqual(data["status_message"], Order.Status(Order.Status.CHA).label)
|
|
|
|
self.assertFalse(data["is_fiat_sent"])
|
|
|
|
# Cancel order to avoid leaving pending HTLCs after a successful test
|
|
self.cancel_order(data["id"])
|
|
|
|
def submit_payout_invoice(
|
|
self, order_id, num_satoshis, routing_budget, robot_index=1
|
|
):
|
|
path = reverse("order")
|
|
params = f"?order_id={order_id}"
|
|
headers = self.get_robot_auth(robot_index)
|
|
|
|
payout_invoice = add_invoice("robot", num_satoshis)
|
|
signed_payout_invoice = sign_message(
|
|
payout_invoice,
|
|
passphrase_path=f"tests/robots/{robot_index}/token",
|
|
private_key_path=f"tests/robots/{robot_index}/enc_priv_key",
|
|
)
|
|
body = {
|
|
"action": "update_invoice",
|
|
"invoice": signed_payout_invoice,
|
|
"routing_budget_ppm": routing_budget,
|
|
}
|
|
|
|
response = self.client.post(path + params, body, **headers)
|
|
|
|
return response
|
|
|
|
def trade_to_submitted_invoice(
|
|
self, maker_form, take_amount=80, maker_index=1, taker_index=2
|
|
):
|
|
response_escrow_locked = self.trade_to_locked_escrow(
|
|
maker_form, take_amount, maker_index, taker_index
|
|
)
|
|
|
|
response_get = self.get_order(response_escrow_locked.json()["id"], maker_index)
|
|
|
|
response = self.submit_payout_invoice(
|
|
response_escrow_locked.json()["id"],
|
|
response_get.json()["trade_satoshis"],
|
|
0,
|
|
maker_index,
|
|
)
|
|
return response
|
|
|
|
def test_trade_to_submitted_invoice(self):
|
|
"""
|
|
Tests a trade from order creation until escrow locked and
|
|
invoice is submitted by buyer.
|
|
"""
|
|
maker_index = 1
|
|
taker_index = 2
|
|
maker_form = self.maker_form_buy_with_range
|
|
take_amount = round(
|
|
random.uniform(maker_form["min_amount"], maker_form["max_amount"]), 2
|
|
)
|
|
|
|
response = self.trade_to_submitted_invoice(
|
|
maker_form, take_amount, maker_index, taker_index
|
|
)
|
|
data = response.json()
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertResponse(response)
|
|
|
|
self.assertEqual(data["status_message"], Order.Status(Order.Status.CHA).label)
|
|
self.assertFalse(data["is_fiat_sent"])
|
|
|
|
# Cancel order to avoid leaving pending HTLCs after a successful test
|
|
self.cancel_order(data["id"])
|
|
|
|
def confirm_fiat(self, order_id, robot_index=1):
|
|
path = reverse("order")
|
|
params = f"?order_id={order_id}"
|
|
headers = self.get_robot_auth(robot_index)
|
|
|
|
body = {"action": "confirm"}
|
|
|
|
response = self.client.post(path + params, body, **headers)
|
|
return response
|
|
|
|
def trade_to_confirm_fiat_sent_LN(
|
|
self, maker_form, take_amount=80, maker_index=1, taker_index=2
|
|
):
|
|
response_submitted_invoice = self.trade_to_submitted_invoice(
|
|
maker_form, take_amount, maker_index, taker_index
|
|
)
|
|
response = self.confirm_fiat(
|
|
response_submitted_invoice.json()["id"], maker_index
|
|
)
|
|
return response
|
|
|
|
def test_trade_to_confirm_fiat_sent_LN(self):
|
|
"""
|
|
Tests a trade from order creation until fiat sent confirmed
|
|
"""
|
|
maker_index = 1
|
|
taker_index = 2
|
|
maker_form = self.maker_form_buy_with_range
|
|
take_amount = round(
|
|
random.uniform(maker_form["min_amount"], maker_form["max_amount"]), 2
|
|
)
|
|
|
|
response = self.trade_to_confirm_fiat_sent_LN(
|
|
maker_form, take_amount, maker_index, taker_index
|
|
)
|
|
data = response.json()
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertResponse(response)
|
|
|
|
self.assertEqual(data["status_message"], Order.Status(Order.Status.FSE).label)
|
|
self.assertTrue(data["is_fiat_sent"])
|
|
|
|
# Cancel order to avoid leaving pending HTLCs after a successful test
|
|
self.cancel_order(data["id"], maker_index)
|
|
self.cancel_order(data["id"], taker_index)
|
|
|
|
def trade_to_confirm_fiat_received_LN(
|
|
self, maker_form, take_amount=80, maker_index=1, taker_index=2
|
|
):
|
|
response_submitted_invoice = self.trade_to_confirm_fiat_sent_LN(
|
|
maker_form, take_amount, maker_index, taker_index
|
|
)
|
|
response = self.confirm_fiat(
|
|
response_submitted_invoice.json()["id"], taker_index
|
|
)
|
|
return response
|
|
|
|
def test_trade_to_confirm_fiat_received_LN(self):
|
|
"""
|
|
Tests a trade from order creation until fiat received is confirmed by seller/taker
|
|
"""
|
|
maker_index = 1
|
|
taker_index = 2
|
|
maker_form = self.maker_form_buy_with_range
|
|
take_amount = round(
|
|
random.uniform(maker_form["min_amount"], maker_form["max_amount"]), 2
|
|
)
|
|
|
|
response = self.trade_to_confirm_fiat_received_LN(
|
|
maker_form, take_amount, maker_index, taker_index
|
|
)
|
|
data = response.json()
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertResponse(response)
|
|
|
|
self.assertEqual(data["status_message"], Order.Status(Order.Status.PAY).label)
|
|
self.assertTrue(data["is_fiat_sent"])
|
|
self.assertFalse(data["is_disputed"])
|
|
self.assertFalse(data["maker_locked"])
|
|
self.assertFalse(data["taker_locked"])
|
|
self.assertFalse(data["escrow_locked"])
|
|
|
|
@patch("api.tasks.follow_send_payment.delay", follow_send_payment)
|
|
def test_successful_LN(self):
|
|
"""
|
|
Tests a trade from order creation until Sats sent to buyer
|
|
"""
|
|
maker_index = 1
|
|
taker_index = 2
|
|
maker_form = self.maker_form_buy_with_range
|
|
take_amount = round(
|
|
random.uniform(maker_form["min_amount"], maker_form["max_amount"]), 2
|
|
)
|
|
|
|
response = self.trade_to_confirm_fiat_received_LN(
|
|
maker_form, take_amount, maker_index, taker_index
|
|
)
|
|
|
|
# Invoke the background thread that will call the celery-worker to follow_send_payment()
|
|
self.send_payments()
|
|
|
|
response = self.get_order(response.json()["id"], maker_index)
|
|
data = response.json()
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertResponse(response)
|
|
|
|
self.assertEqual(data["status_message"], Order.Status(Order.Status.SUC).label)
|
|
self.assertTrue(data["is_fiat_sent"])
|
|
self.assertFalse(data["is_disputed"])
|
|
self.assertIsHash(data["maker_summary"]["preimage"])
|
|
self.assertIsHash(data["maker_summary"]["payment_hash"])
|