|
|
#!/usr/bin/env python3 |
|
|
# -*- coding: utf-8 -*- |
|
|
|
|
|
# Copyright (c) 2019 tecnovert |
|
|
# Distributed under the MIT software license, see the accompanying |
|
|
# file LICENSE.txt or http://www.opensource.org/licenses/mit-license.php. |
|
|
|
|
|
""" |
|
|
|
|
|
export TEST_RELOAD_PATH=/tmp/test_basicswap |
|
|
mkdir -p ${TEST_RELOAD_PATH}/bin/{particl,bitcoin} |
|
|
cp ~/tmp/particl-0.18.1.2-x86_64-linux-gnu.tar.gz ${TEST_RELOAD_PATH}/bin/particl |
|
|
cp ~/tmp/bitcoin-0.18.0-x86_64-linux-gnu.tar.gz ${TEST_RELOAD_PATH}/bin/bitcoin |
|
|
|
|
|
|
|
|
""" |
|
|
|
|
|
import os |
|
|
import sys |
|
|
import time |
|
|
import unittest |
|
|
import logging |
|
|
import shutil |
|
|
import json |
|
|
import traceback |
|
|
import multiprocessing |
|
|
from unittest.mock import patch |
|
|
from urllib.request import urlopen |
|
|
from urllib import parse |
|
|
|
|
|
import bin.basicswap_prepare as prepareSystem |
|
|
import bin.basicswap_run as runSystem |
|
|
|
|
|
test_path = os.path.expanduser(os.getenv('TEST_RELOAD_PATH', '~/test_basicswap1')) |
|
|
PARTICL_PORT_BASE = int(os.getenv('PARTICL_PORT_BASE', '11938')) |
|
|
BITCOIN_PORT_BASE = int(os.getenv('BITCOIN_PORT_BASE', '10938')) |
|
|
|
|
|
logger = logging.getLogger() |
|
|
logger.level = logging.DEBUG |
|
|
if not len(logger.handlers): |
|
|
logger.addHandler(logging.StreamHandler(sys.stdout)) |
|
|
|
|
|
|
|
|
def waitForServer(): |
|
|
for i in range(20): |
|
|
try: |
|
|
time.sleep(1) |
|
|
summary = json.loads(urlopen('http://localhost:12700/json').read()) |
|
|
break |
|
|
except Exception: |
|
|
traceback.print_exc() |
|
|
|
|
|
|
|
|
class Test(unittest.TestCase): |
|
|
@classmethod |
|
|
def setUpClass(cls): |
|
|
super(Test, cls).setUpClass() |
|
|
|
|
|
mnemonics = [ |
|
|
'abandon baby cabbage dad eager fabric gadget habit ice kangaroo lab absorb', |
|
|
'actuel comédie poésie noble facile éprouver brave cellule rotule académie hilarant chambre', |
|
|
'ちしき いてざ きおち あしあと ぽちぶくろ こえる さつえい むえき あける ほんき むさぼる ねいろ', |
|
|
] |
|
|
|
|
|
for i in range(3): |
|
|
client_path = os.path.join(test_path, 'client{}'.format(i)) |
|
|
config_path = os.path.join(client_path, 'basicswap.json') |
|
|
try: |
|
|
shutil.rmtree(client_path) |
|
|
except Exception as ex: |
|
|
logger.warning('setUpClass %s', str(ex)) |
|
|
testargs = [ |
|
|
'basicswap-prepare', |
|
|
'-datadir="{}"'.format(client_path), |
|
|
'-bindir="{}"'.format(test_path + '/bin'), |
|
|
'-portoffset={}'.format(i), |
|
|
'-particl_mnemonic="{}"'.format(mnemonics[i]), |
|
|
'-regtest', '-withoutcoin=litecoin', '-withcoin=bitcoin'] |
|
|
with patch.object(sys, 'argv', testargs): |
|
|
prepareSystem.main() |
|
|
|
|
|
with open(os.path.join(client_path, 'particl', 'particl.conf'), 'a') as fp: |
|
|
fp.write('port={}\n'.format(PARTICL_PORT_BASE + i)) |
|
|
for ip in range(3): |
|
|
fp.write('addnode=localhost:{}\n'.format(PARTICL_PORT_BASE + ip)) |
|
|
with open(os.path.join(client_path, 'bitcoin', 'bitcoin.conf'), 'a') as fp: |
|
|
fp.write('port={}\n'.format(BITCOIN_PORT_BASE + i)) |
|
|
|
|
|
assert(os.path.exists(config_path)) |
|
|
|
|
|
def run_thread(self, client_id): |
|
|
client_path = os.path.join(test_path, 'client{}'.format(client_id)) |
|
|
testargs = ['basicswap-run', '-datadir=' + client_path, '-regtest'] |
|
|
with patch.object(sys, 'argv', testargs): |
|
|
runSystem.main() |
|
|
|
|
|
def test_reload(self): |
|
|
processes = [] |
|
|
|
|
|
for i in range(3): |
|
|
processes.append(multiprocessing.Process(target=self.run_thread, args=(i,))) |
|
|
processes[-1].start() |
|
|
|
|
|
try: |
|
|
waitForServer() |
|
|
|
|
|
data = parse.urlencode({ |
|
|
'addr_from': '-1', |
|
|
'coin_from': '1', |
|
|
'coin_to': '2', |
|
|
'amt_from': '1', |
|
|
'amt_to': '1', |
|
|
'lockhrs': '24'}).encode() |
|
|
|
|
|
offer_id = json.loads(urlopen('http://localhost:12700/json/offers/new', data=data).read()) |
|
|
summary = json.loads(urlopen('http://localhost:12700/json').read()) |
|
|
assert(summary['num_sent_offers'] == 1) |
|
|
except Exception: |
|
|
traceback.print_exc() |
|
|
|
|
|
logger.warning('TODO') |
|
|
time.sleep(5) |
|
|
|
|
|
for p in processes: |
|
|
p.terminate() |
|
|
for p in processes: |
|
|
p.join() |
|
|
|
|
|
|
|
|
if __name__ == '__main__': |
|
|
unittest.main()
|
|
|
|
