Host-customized fork of https://github.com/tecnovert/basicswap/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
236 lines
10 KiB
236 lines
10 KiB
3 years ago
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
# Copyright (c) 2021 tecnovert
|
||
|
|
# Distributed under the MIT software license, see the accompanying
|
||
|
|
# file LICENSE or http://www.opensource.org/licenses/mit-license.php.
|
||
|
|
|
||
|
|
import json
|
||
|
|
import random
|
||
|
|
import logging
|
||
|
|
import unittest
|
||
|
|
from urllib.request import urlopen
|
||
|
|
|
||
|
|
from basicswap.basicswap import (
|
||
|
|
Coins,
|
||
|
|
SwapTypes,
|
||
|
|
BidStates,
|
||
|
|
DebugTypes,
|
||
|
|
)
|
||
|
|
from basicswap.basicswap_util import (
|
||
|
|
SEQUENCE_LOCK_BLOCKS,
|
||
|
|
)
|
||
|
|
from basicswap.util import (
|
||
|
|
make_int,
|
||
|
|
format_amount,
|
||
|
|
)
|
||
|
|
from tests.basicswap.common import (
|
||
|
|
wait_for_bid,
|
||
|
|
wait_for_offer,
|
||
|
|
wait_for_none_active,
|
||
|
|
wait_for_balance,
|
||
|
|
post_json_req,
|
||
|
|
)
|
||
|
|
|
||
|
|
from .test_xmr import BaseTest, test_delay_event
|
||
|
|
|
||
|
|
logger = logging.getLogger()
|
||
|
|
|
||
|
|
|
||
|
|
class Test(BaseTest):
|
||
|
|
__test__ = True
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def setUpClass(cls):
|
||
|
|
super(Test, cls).setUpClass()
|
||
|
|
|
||
|
|
js_0 = json.loads(urlopen('http://127.0.0.1:1800/json/wallets/part').read())
|
||
|
|
node0_blind_before = js_0['blind_balance'] + js_0['blind_unconfirmed']
|
||
|
|
|
||
|
|
post_json = {
|
||
|
|
'value': 100,
|
||
|
|
'address': js_0['stealth_address'],
|
||
|
|
'subfee': False,
|
||
|
|
'type_to': 'blind',
|
||
|
|
}
|
||
|
|
json_rv = json.loads(post_json_req('http://127.0.0.1:1800/json/wallets/part/withdraw', post_json))
|
||
|
|
assert(len(json_rv['txid']) == 64)
|
||
|
|
|
||
|
|
logging.info('Waiting for blind balance')
|
||
|
|
wait_for_balance(test_delay_event, 'http://127.0.0.1:1800/json/wallets/part', 'blind_balance', 100.0 + node0_blind_before)
|
||
|
|
js_0 = json.loads(urlopen('http://127.0.0.1:1800/json/wallets/part').read())
|
||
|
|
node0_blind_before = js_0['blind_balance'] + js_0['blind_unconfirmed']
|
||
|
|
|
||
|
|
def getBalance(self, js_wallets):
|
||
|
|
return float(js_wallets[str(int(Coins.PART))]['blind_balance']) + float(js_wallets[str(int(Coins.PART))]['blind_unconfirmed'])
|
||
|
|
|
||
|
|
def getXmrBalance(self, js_wallets):
|
||
|
|
return float(js_wallets[str(int(Coins.XMR))]['unconfirmed']) + float(js_wallets[str(int(Coins.XMR))]['balance'])
|
||
|
|
|
||
|
|
def test_01_part_xmr(self):
|
||
|
|
logging.info('---------- Test PARTct to XMR')
|
||
|
|
swap_clients = self.swap_clients
|
||
|
|
|
||
|
|
js_0 = json.loads(urlopen('http://127.0.0.1:1800/json/wallets/part').read())
|
||
|
|
assert(float(js_0['blind_balance']) > 10.0)
|
||
|
|
node0_blind_before = js_0['blind_balance'] + js_0['blind_unconfirmed']
|
||
|
|
|
||
|
|
js_1 = json.loads(urlopen('http://127.0.0.1:1801/json/wallets/part').read())
|
||
|
|
node1_blind_before = js_1['blind_balance'] + js_1['blind_unconfirmed']
|
||
|
|
|
||
|
|
js_0_xmr = json.loads(urlopen('http://127.0.0.1:1800/json/wallets/xmr').read())
|
||
|
|
js_1_xmr = json.loads(urlopen('http://127.0.0.1:1801/json/wallets/xmr').read())
|
||
|
|
|
||
|
|
amt_swap = make_int(random.uniform(0.1, 2.0), scale=8, r=1)
|
||
|
|
rate_swap = make_int(random.uniform(0.2, 20.0), scale=12, r=1)
|
||
|
|
offer_id = swap_clients[0].postOffer(Coins.PART_BLIND, Coins.XMR, amt_swap, rate_swap, amt_swap, SwapTypes.XMR_SWAP)
|
||
|
|
wait_for_offer(test_delay_event, swap_clients[1], offer_id)
|
||
|
|
offers = swap_clients[0].listOffers(filters={'offer_id': offer_id})
|
||
|
|
offer = offers[0]
|
||
|
|
|
||
|
|
bid_id = swap_clients[1].postXmrBid(offer_id, offer.amount_from)
|
||
|
|
|
||
|
|
wait_for_bid(test_delay_event, swap_clients[0], bid_id, BidStates.BID_RECEIVED)
|
||
|
|
|
||
|
|
swap_clients[0].acceptXmrBid(bid_id)
|
||
|
|
|
||
|
|
wait_for_bid(test_delay_event, swap_clients[0], bid_id, BidStates.SWAP_COMPLETED, wait_for=180)
|
||
|
|
wait_for_bid(test_delay_event, swap_clients[1], bid_id, BidStates.SWAP_COMPLETED, sent=True)
|
||
|
|
|
||
|
|
amount_from = float(format_amount(amt_swap, 8))
|
||
|
|
js_1 = json.loads(urlopen('http://127.0.0.1:1801/json/wallets/part').read())
|
||
|
|
node1_blind_after = js_1['blind_balance'] + js_1['blind_unconfirmed']
|
||
|
|
assert(node1_blind_after > node1_blind_before + (amount_from - 0.05))
|
||
|
|
|
||
|
|
js_0 = json.loads(urlopen('http://127.0.0.1:1800/json/wallets/part').read())
|
||
|
|
node0_blind_after = js_0['blind_balance'] + js_0['blind_unconfirmed']
|
||
|
|
assert(node0_blind_after < node0_blind_before - amount_from)
|
||
|
|
|
||
|
|
js_0_xmr_after = json.loads(urlopen('http://127.0.0.1:1800/json/wallets/xmr').read())
|
||
|
|
js_1_xmr_after = json.loads(urlopen('http://127.0.0.1:1801/json/wallets/xmr').read())
|
||
|
|
|
||
|
|
scale_from = 8
|
||
|
|
amount_to = int((amt_swap * rate_swap) // (10 ** scale_from))
|
||
|
|
amount_to_float = float(format_amount(amount_to, 12))
|
||
|
|
node1_xmr_after = float(js_1_xmr_after['unconfirmed']) + float(js_1_xmr_after['balance'])
|
||
|
|
node1_xmr_before = float(js_1_xmr['unconfirmed']) + float(js_1_xmr['balance'])
|
||
|
|
assert(node1_xmr_after > node1_xmr_before + (amount_to_float - 0.02))
|
||
|
|
|
||
|
|
def test_02_leader_recover_a_lock_tx(self):
|
||
|
|
logging.info('---------- Test PARTct to XMR leader recovers coin a lock tx')
|
||
|
|
swap_clients = self.swap_clients
|
||
|
|
|
||
|
|
js_w0_before = json.loads(urlopen('http://127.0.0.1:1800/json/wallets').read())
|
||
|
|
node0_blind_before = self.getBalance(js_w0_before)
|
||
|
|
|
||
|
|
amt_swap = make_int(random.uniform(0.1, 2.0), scale=8, r=1)
|
||
|
|
rate_swap = make_int(random.uniform(0.2, 20.0), scale=12, r=1)
|
||
|
|
offer_id = swap_clients[0].postOffer(
|
||
|
|
Coins.PART_BLIND, Coins.XMR, amt_swap, rate_swap, amt_swap, SwapTypes.XMR_SWAP,
|
||
|
|
lock_type=SEQUENCE_LOCK_BLOCKS, lock_value=12)
|
||
|
|
wait_for_offer(test_delay_event, swap_clients[1], offer_id)
|
||
|
|
offer = swap_clients[1].getOffer(offer_id)
|
||
|
|
|
||
|
|
bid_id = swap_clients[1].postXmrBid(offer_id, offer.amount_from)
|
||
|
|
|
||
|
|
wait_for_bid(test_delay_event, swap_clients[0], bid_id, BidStates.BID_RECEIVED)
|
||
|
|
|
||
|
|
bid, xmr_swap = swap_clients[0].getXmrBid(bid_id)
|
||
|
|
assert(xmr_swap)
|
||
|
|
|
||
|
|
swap_clients[1].setBidDebugInd(bid_id, DebugTypes.BID_STOP_AFTER_COIN_A_LOCK)
|
||
|
|
|
||
|
|
swap_clients[0].acceptXmrBid(bid_id)
|
||
|
|
|
||
|
|
wait_for_bid(test_delay_event, swap_clients[0], bid_id, BidStates.XMR_SWAP_FAILED_REFUNDED, wait_for=180)
|
||
|
|
wait_for_bid(test_delay_event, swap_clients[1], bid_id, BidStates.XMR_SWAP_FAILED_REFUNDED, sent=True)
|
||
|
|
|
||
|
|
js_w0_after = json.loads(urlopen('http://127.0.0.1:1800/json/wallets').read())
|
||
|
|
node0_blind_after = self.getBalance(js_w0_after)
|
||
|
|
assert(node0_blind_before - node0_blind_after < 0.02)
|
||
|
|
|
||
|
|
def test_03_follower_recover_a_lock_tx(self):
|
||
|
|
logging.info('---------- Test PARTct to XMR follower recovers coin a lock tx')
|
||
|
|
swap_clients = self.swap_clients
|
||
|
|
|
||
|
|
js_w0_before = json.loads(urlopen('http://127.0.0.1:1800/json/wallets').read())
|
||
|
|
js_w1_before = json.loads(urlopen('http://127.0.0.1:1801/json/wallets').read())
|
||
|
|
|
||
|
|
amt_swap = make_int(random.uniform(0.1, 2.0), scale=8, r=1)
|
||
|
|
rate_swap = make_int(random.uniform(0.2, 20.0), scale=12, r=1)
|
||
|
|
offer_id = swap_clients[0].postOffer(
|
||
|
|
Coins.PART_BLIND, Coins.XMR, amt_swap, rate_swap, amt_swap, SwapTypes.XMR_SWAP,
|
||
|
|
lock_type=SEQUENCE_LOCK_BLOCKS, lock_value=12)
|
||
|
|
wait_for_offer(test_delay_event, swap_clients[1], offer_id)
|
||
|
|
offer = swap_clients[1].getOffer(offer_id)
|
||
|
|
|
||
|
|
bid_id = swap_clients[1].postXmrBid(offer_id, offer.amount_from)
|
||
|
|
|
||
|
|
wait_for_bid(test_delay_event, swap_clients[0], bid_id, BidStates.BID_RECEIVED)
|
||
|
|
|
||
|
|
bid, xmr_swap = swap_clients[0].getXmrBid(bid_id)
|
||
|
|
assert(xmr_swap)
|
||
|
|
|
||
|
|
swap_clients[1].setBidDebugInd(bid_id, DebugTypes.BID_STOP_AFTER_COIN_A_LOCK)
|
||
|
|
swap_clients[0].setBidDebugInd(bid_id, DebugTypes.BID_DONT_SPEND_COIN_A_LOCK_REFUND)
|
||
|
|
|
||
|
|
swap_clients[0].acceptXmrBid(bid_id)
|
||
|
|
|
||
|
|
wait_for_bid(test_delay_event, swap_clients[0], bid_id, BidStates.BID_STALLED_FOR_TEST, wait_for=180)
|
||
|
|
wait_for_bid(test_delay_event, swap_clients[1], bid_id, BidStates.XMR_SWAP_FAILED_SWIPED, wait_for=80, sent=True)
|
||
|
|
|
||
|
|
js_w1_after = json.loads(urlopen('http://127.0.0.1:1801/json/wallets').read())
|
||
|
|
|
||
|
|
node1_blind_before = self.getBalance(js_w1_before)
|
||
|
|
node1_blind_after = self.getBalance(js_w1_after)
|
||
|
|
amount_from = float(format_amount(amt_swap, 8))
|
||
|
|
assert(node1_blind_after - node1_blind_before > (amount_from - 0.02))
|
||
|
|
|
||
|
|
wait_for_none_active(test_delay_event, 1800)
|
||
|
|
wait_for_none_active(test_delay_event, 1801)
|
||
|
|
|
||
|
|
def test_04_follower_recover_b_lock_tx(self):
|
||
|
|
logging.info('---------- Test PARTct to XMR follower recovers coin b lock tx')
|
||
|
|
|
||
|
|
swap_clients = self.swap_clients
|
||
|
|
|
||
|
|
js_w0_before = json.loads(urlopen('http://127.0.0.1:1800/json/wallets').read())
|
||
|
|
js_w1_before = json.loads(urlopen('http://127.0.0.1:1801/json/wallets').read())
|
||
|
|
|
||
|
|
amt_swap = make_int(random.uniform(0.1, 2.0), scale=8, r=1)
|
||
|
|
rate_swap = make_int(random.uniform(0.2, 20.0), scale=12, r=1)
|
||
|
|
offer_id = swap_clients[0].postOffer(
|
||
|
|
Coins.PART_BLIND, Coins.XMR, amt_swap, rate_swap, amt_swap, SwapTypes.XMR_SWAP,
|
||
|
|
lock_type=SEQUENCE_LOCK_BLOCKS, lock_value=18)
|
||
|
|
wait_for_offer(test_delay_event, swap_clients[1], offer_id)
|
||
|
|
offer = swap_clients[1].getOffer(offer_id)
|
||
|
|
|
||
|
|
bid_id = swap_clients[1].postXmrBid(offer_id, offer.amount_from)
|
||
|
|
|
||
|
|
wait_for_bid(test_delay_event, swap_clients[0], bid_id, BidStates.BID_RECEIVED)
|
||
|
|
|
||
|
|
bid, xmr_swap = swap_clients[0].getXmrBid(bid_id)
|
||
|
|
assert(xmr_swap)
|
||
|
|
|
||
|
|
swap_clients[1].setBidDebugInd(bid_id, DebugTypes.CREATE_INVALID_COIN_B_LOCK)
|
||
|
|
|
||
|
|
swap_clients[0].acceptXmrBid(bid_id)
|
||
|
|
|
||
|
|
wait_for_bid(test_delay_event, swap_clients[0], bid_id, BidStates.XMR_SWAP_FAILED_REFUNDED, wait_for=180)
|
||
|
|
wait_for_bid(test_delay_event, swap_clients[1], bid_id, BidStates.XMR_SWAP_FAILED_REFUNDED, sent=True)
|
||
|
|
|
||
|
|
js_w0_after = json.loads(urlopen('http://127.0.0.1:1800/json/wallets').read())
|
||
|
|
js_w1_after = json.loads(urlopen('http://127.0.0.1:1801/json/wallets').read())
|
||
|
|
|
||
|
|
node0_blind_before = self.getBalance(js_w0_before)
|
||
|
|
node0_blind_after = self.getBalance(js_w0_after)
|
||
|
|
assert(node0_blind_before - node0_blind_after < 0.02)
|
||
|
|
|
||
|
|
node1_xmr_before = self.getXmrBalance(js_w1_before)
|
||
|
|
node1_xmr_after = self.getXmrBalance(js_w1_after)
|
||
|
|
assert(node1_xmr_before - node1_xmr_after < 0.02)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|