tecnovert
4 years ago
parent
9a182646f0
commit
269d10aeda
10 changed files with 792 additions and 436 deletions
@ -0,0 +1,306 @@ |
||||
#!/usr/bin/env python3 |
||||
# -*- coding: utf-8 -*- |
||||
|
||||
# Copyright (c) 2020 tecnovert |
||||
# Distributed under the MIT software license, see the accompanying |
||||
# file LICENSE or http://www.opensource.org/licenses/mit-license.php. |
||||
|
||||
import os |
||||
import json |
||||
import time |
||||
import shutil |
||||
import signal |
||||
import logging |
||||
import unittest |
||||
import traceback |
||||
import threading |
||||
from urllib.request import urlopen |
||||
|
||||
import basicswap.config as cfg |
||||
from basicswap.basicswap import ( |
||||
BasicSwap, |
||||
Coins, |
||||
SwapTypes, |
||||
) |
||||
from basicswap.util import ( |
||||
COIN, |
||||
toWIF, |
||||
) |
||||
from basicswap.rpc import ( |
||||
callrpc, |
||||
callrpc_cli, |
||||
waitForRPC, |
||||
) |
||||
from basicswap.contrib.key import ( |
||||
ECKey, |
||||
) |
||||
from basicswap.http_server import ( |
||||
HttpThread, |
||||
) |
||||
from tests.basicswap.common import ( |
||||
prepareDataDir, |
||||
make_rpc_func, |
||||
checkForks, |
||||
stopDaemons, |
||||
wait_for_offer, |
||||
TEST_HTTP_HOST, |
||||
TEST_HTTP_PORT, |
||||
BASE_P2P_PORT, |
||||
BASE_RPC_PORT, |
||||
BASE_ZMQ_PORT, |
||||
BTC_BASE_PORT, |
||||
BTC_BASE_RPC_PORT, |
||||
PREFIX_SECRET_KEY_REGTEST, |
||||
) |
||||
|
||||
from bin.basicswap_run import startDaemon |
||||
|
||||
|
||||
logger = logging.getLogger() |
||||
|
||||
NUM_NODES = 3 |
||||
NUM_BTC_NODES = 3 |
||||
TEST_DIR = cfg.TEST_DATADIRS |
||||
|
||||
delay_event = threading.Event() |
||||
stop_test = False |
||||
|
||||
|
||||
def prepare_swapclient_dir(datadir, node_id, network_key, network_pubkey): |
||||
basicswap_dir = os.path.join(datadir, 'basicswap_' + str(node_id)) |
||||
if not os.path.exists(basicswap_dir): |
||||
os.makedirs(basicswap_dir) |
||||
|
||||
settings_path = os.path.join(basicswap_dir, cfg.CONFIG_FILENAME) |
||||
settings = { |
||||
'debug': True, |
||||
'p2p_host': '127.0.0.1', |
||||
'p2p_port': BASE_P2P_PORT + node_id, |
||||
'zmqhost': 'tcp://127.0.0.1', |
||||
'zmqport': BASE_ZMQ_PORT + node_id, |
||||
'htmlhost': 'localhost', |
||||
'htmlport': TEST_HTTP_PORT + node_id, |
||||
'network_key': network_key, |
||||
'network_pubkey': network_pubkey, |
||||
'chainclients': { |
||||
'particl': { |
||||
'connection_type': 'rpc', |
||||
'manage_daemon': False, |
||||
'rpcport': BASE_RPC_PORT + node_id, |
||||
'rpcuser': 'test' + str(node_id), |
||||
'rpcpassword': 'test_pass' + str(node_id), |
||||
'datadir': os.path.join(datadir, 'part_' + str(node_id)), |
||||
'bindir': cfg.PARTICL_BINDIR, |
||||
'blocks_confirmed': 2, # Faster testing |
||||
}, |
||||
'bitcoin': { |
||||
'connection_type': 'rpc', |
||||
'manage_daemon': False, |
||||
'rpcport': BTC_BASE_RPC_PORT + node_id, |
||||
'rpcuser': 'test' + str(node_id), |
||||
'rpcpassword': 'test_pass' + str(node_id), |
||||
'datadir': os.path.join(datadir, 'btc_' + str(node_id)), |
||||
'bindir': cfg.BITCOIN_BINDIR, |
||||
'use_segwit': True, |
||||
} |
||||
|
||||
}, |
||||
'check_progress_seconds': 2, |
||||
'check_watched_seconds': 4, |
||||
'check_expired_seconds': 60, |
||||
'check_events_seconds': 1, |
||||
'check_xmr_swaps_seconds': 1, |
||||
'min_delay_event': 1, |
||||
'max_delay_event': 5, |
||||
'min_delay_retry': 2, |
||||
'max_delay_retry': 10 |
||||
} |
||||
|
||||
with open(settings_path, 'w') as fp: |
||||
json.dump(settings, fp, indent=4) |
||||
|
||||
|
||||
def partRpc(cmd, node_id=0): |
||||
return callrpc_cli(cfg.PARTICL_BINDIR, os.path.join(TEST_DIR, 'part_' + str(node_id)), 'regtest', cmd, cfg.PARTICL_CLI) |
||||
|
||||
|
||||
def btcRpc(cmd, node_id=0): |
||||
return callrpc_cli(cfg.BITCOIN_BINDIR, os.path.join(TEST_DIR, 'btc_' + str(node_id)), 'regtest', cmd, cfg.BITCOIN_CLI) |
||||
|
||||
|
||||
def signal_handler(sig, frame): |
||||
global stop_test |
||||
logging.info('signal {} detected.'.format(sig)) |
||||
stop_test = True |
||||
delay_event.set() |
||||
|
||||
|
||||
def callnoderpc(node_id, method, params=[], wallet=None, base_rpc_port=BASE_RPC_PORT): |
||||
auth = 'test{0}:test_pass{0}'.format(node_id) |
||||
return callrpc(base_rpc_port + node_id, auth, method, params, wallet) |
||||
|
||||
|
||||
def run_coins_loop(cls): |
||||
global stop_test |
||||
while not stop_test: |
||||
if cls.btc_addr is not None: |
||||
btcRpc('generatetoaddress 1 {}'.format(cls.btc_addr)) |
||||
time.sleep(1.0) |
||||
|
||||
|
||||
def run_loop(cls): |
||||
global stop_test |
||||
while not stop_test: |
||||
for c in cls.swap_clients: |
||||
c.update() |
||||
time.sleep(1.0) |
||||
|
||||
|
||||
class Test(unittest.TestCase): |
||||
|
||||
@classmethod |
||||
def setUpClass(cls): |
||||
super(Test, cls).setUpClass() |
||||
|
||||
cls.update_thread = None |
||||
cls.coins_update_thread = None |
||||
cls.http_threads = [] |
||||
cls.swap_clients = [] |
||||
cls.part_daemons = [] |
||||
cls.btc_daemons = [] |
||||
|
||||
cls.part_stakelimit = 0 |
||||
cls.btc_addr = None |
||||
|
||||
logger.propagate = False |
||||
logger.handlers = [] |
||||
logger.setLevel(logging.INFO) # DEBUG shows many messages from requests.post |
||||
formatter = logging.Formatter('%(asctime)s %(levelname)s : %(message)s') |
||||
stream_stdout = logging.StreamHandler() |
||||
stream_stdout.setFormatter(formatter) |
||||
logger.addHandler(stream_stdout) |
||||
|
||||
if os.path.isdir(TEST_DIR): |
||||
logging.info('Removing ' + TEST_DIR) |
||||
shutil.rmtree(TEST_DIR) |
||||
if not os.path.exists(TEST_DIR): |
||||
os.makedirs(TEST_DIR) |
||||
|
||||
cls.stream_fp = logging.FileHandler(os.path.join(TEST_DIR, 'test.log')) |
||||
cls.stream_fp.setFormatter(formatter) |
||||
logger.addHandler(cls.stream_fp) |
||||
|
||||
try: |
||||
logging.info('Preparing coin nodes.') |
||||
for i in range(NUM_NODES): |
||||
prepareDataDir(TEST_DIR, i, 'particl.conf', 'part_') |
||||
|
||||
cls.part_daemons.append(startDaemon(os.path.join(TEST_DIR, 'part_' + str(i)), cfg.PARTICL_BINDIR, cfg.PARTICLD)) |
||||
logging.info('Started %s %d', cfg.PARTICLD, cls.part_daemons[-1].pid) |
||||
|
||||
for i in range(NUM_NODES): |
||||
# Load mnemonics after all nodes have started to avoid staking getting stuck in TryToSync |
||||
rpc = make_rpc_func(i) |
||||
waitForRPC(rpc) |
||||
if i == 0: |
||||
rpc('extkeyimportmaster', ['abandon baby cabbage dad eager fabric gadget habit ice kangaroo lab absorb']) |
||||
elif i == 1: |
||||
rpc('extkeyimportmaster', ['pact mammal barrel matrix local final lecture chunk wasp survey bid various book strong spread fall ozone daring like topple door fatigue limb olympic', '', 'true']) |
||||
rpc('getnewextaddress', ['lblExtTest']) |
||||
rpc('rescanblockchain') |
||||
else: |
||||
rpc('extkeyimportmaster', [rpc('mnemonic', ['new'])['master']]) |
||||
# Lower output split threshold for more stakeable outputs |
||||
rpc('walletsettings', ['stakingoptions', {'stakecombinethreshold': 100, 'stakesplitthreshold': 200}]) |
||||
|
||||
for i in range(NUM_BTC_NODES): |
||||
prepareDataDir(TEST_DIR, i, 'bitcoin.conf', 'btc_', base_p2p_port=BTC_BASE_PORT, base_rpc_port=BTC_BASE_RPC_PORT) |
||||
|
||||
cls.btc_daemons.append(startDaemon(os.path.join(TEST_DIR, 'btc_' + str(i)), cfg.BITCOIN_BINDIR, cfg.BITCOIND)) |
||||
logging.info('Started %s %d', cfg.BITCOIND, cls.part_daemons[-1].pid) |
||||
|
||||
waitForRPC(make_rpc_func(i, base_rpc_port=BTC_BASE_RPC_PORT)) |
||||
|
||||
logging.info('Preparing swap clients.') |
||||
eckey = ECKey() |
||||
eckey.generate() |
||||
cls.network_key = toWIF(PREFIX_SECRET_KEY_REGTEST, eckey.get_bytes()) |
||||
cls.network_pubkey = eckey.get_pubkey().get_bytes().hex() |
||||
|
||||
for i in range(NUM_NODES): |
||||
prepare_swapclient_dir(TEST_DIR, i, cls.network_key, cls.network_pubkey) |
||||
basicswap_dir = os.path.join(os.path.join(TEST_DIR, 'basicswap_' + str(i))) |
||||
settings_path = os.path.join(basicswap_dir, cfg.CONFIG_FILENAME) |
||||
with open(settings_path) as fs: |
||||
settings = json.load(fs) |
||||
fp = open(os.path.join(basicswap_dir, 'basicswap.log'), 'w') |
||||
sc = BasicSwap(fp, basicswap_dir, settings, 'regtest', log_name='BasicSwap{}'.format(i)) |
||||
sc.setDaemonPID(Coins.BTC, cls.btc_daemons[i].pid) |
||||
sc.setDaemonPID(Coins.PART, cls.part_daemons[i].pid) |
||||
sc.start() |
||||
cls.swap_clients.append(sc) |
||||
|
||||
t = HttpThread(cls.swap_clients[i].fp, TEST_HTTP_HOST, TEST_HTTP_PORT + i, False, cls.swap_clients[i]) |
||||
cls.http_threads.append(t) |
||||
t.start() |
||||
|
||||
cls.btc_addr = callnoderpc(0, 'getnewaddress', ['mining_addr', 'bech32'], base_rpc_port=BTC_BASE_RPC_PORT) |
||||
|
||||
num_blocks = 500 |
||||
logging.info('Mining %d Bitcoin blocks to %s', num_blocks, cls.btc_addr) |
||||
callnoderpc(0, 'generatetoaddress', [num_blocks, cls.btc_addr], base_rpc_port=BTC_BASE_RPC_PORT) |
||||
|
||||
checkForks(callnoderpc(0, 'getblockchaininfo', base_rpc_port=BTC_BASE_RPC_PORT)) |
||||
|
||||
logging.info('Starting update thread.') |
||||
signal.signal(signal.SIGINT, signal_handler) |
||||
cls.update_thread = threading.Thread(target=run_loop, args=(cls,)) |
||||
cls.update_thread.start() |
||||
|
||||
cls.coins_update_thread = threading.Thread(target=run_coins_loop, args=(cls,)) |
||||
cls.coins_update_thread.start() |
||||
except Exception: |
||||
traceback.print_exc() |
||||
Test.tearDownClass() |
||||
raise ValueError('setUpClass() failed.') |
||||
|
||||
@classmethod |
||||
def tearDownClass(cls): |
||||
global stop_test |
||||
logging.info('Finalising') |
||||
stop_test = True |
||||
if cls.update_thread is not None: |
||||
try: |
||||
cls.update_thread.join() |
||||
except Exception: |
||||
logging.info('Failed to join update_thread') |
||||
if cls.coins_update_thread is not None: |
||||
try: |
||||
cls.coins_update_thread.join() |
||||
except Exception: |
||||
logging.info('Failed to join coins_update_thread') |
||||
|
||||
for t in cls.http_threads: |
||||
t.stop() |
||||
t.join() |
||||
for c in cls.swap_clients: |
||||
c.finalise() |
||||
c.fp.close() |
||||
|
||||
stopDaemons(cls.part_daemons) |
||||
stopDaemons(cls.btc_daemons) |
||||
|
||||
super(Test, cls).tearDownClass() |
||||
|
||||
def test_01_part_btc(self): |
||||
logging.info('---------- Test PART to BTC') |
||||
swap_clients = self.swap_clients |
||||
|
||||
js_1 = json.loads(urlopen('http://localhost:1801/json/wallets').read()) |
||||
|
||||
offer_id = swap_clients[0].postOffer(Coins.PART, Coins.BTC, 100 * COIN, 0.1 * COIN, 100 * COIN, SwapTypes.SELLER_FIRST) |
||||
wait_for_offer(delay_event, swap_clients[1], offer_id) |
||||
|
||||
|
||||
if __name__ == '__main__': |
||||
unittest.main() |
||||
Loading…
Reference in new issue