tecnovert
7 months ago
parent
dc4f0ac2d3
commit
fd7977b35a
7 changed files with 237 additions and 74 deletions
@ -0,0 +1,191 @@ |
||||
#!/usr/bin/env python3 |
||||
# -*- coding: utf-8 -*- |
||||
|
||||
# Copyright (c) 2024 tecnovert |
||||
# Distributed under the MIT software license, see the accompanying |
||||
# file LICENSE or http://www.opensource.org/licenses/mit-license.php. |
||||
|
||||
import time |
||||
import logging |
||||
import os |
||||
|
||||
from basicswap.basicswap import ( |
||||
Coins, |
||||
) |
||||
import basicswap.config as cfg |
||||
from basicswap.rpc_xmr import ( |
||||
callrpc_xmr, |
||||
) |
||||
from tests.basicswap.common import ( |
||||
stopDaemons, |
||||
) |
||||
from tests.basicswap.test_xmr import BaseTest |
||||
from bin.basicswap_run import startXmrDaemon, startXmrWalletDaemon |
||||
|
||||
from tests.basicswap.extended.test_dcr import ( |
||||
run_test_ads_success_path, |
||||
run_test_ads_both_refund, |
||||
run_test_ads_swipe_refund, |
||||
) |
||||
|
||||
NUM_NODES = 3 |
||||
|
||||
WOW_BINDIR = os.path.expanduser(os.getenv('WOW_BINDIR', os.path.join(cfg.DEFAULT_TEST_BINDIR, 'wownero'))) |
||||
WOWD = os.getenv('WOWD', 'wownerod' + cfg.bin_suffix) |
||||
WOW_WALLET_RPC = os.getenv('WOW_WALLET', 'wownero-wallet-rpc' + cfg.bin_suffix) |
||||
|
||||
WOW_BASE_PORT = 54932 |
||||
WOW_BASE_RPC_PORT = 55932 |
||||
WOW_BASE_WALLET_RPC_PORT = 55952 |
||||
WOW_BASE_ZMQ_PORT = 55972 |
||||
|
||||
|
||||
def prepareWOWDataDir(datadir, node_id, conf_file): |
||||
node_dir = os.path.join(datadir, 'wow_' + str(node_id)) |
||||
if not os.path.exists(node_dir): |
||||
os.makedirs(node_dir) |
||||
cfg_file_path = os.path.join(node_dir, conf_file) |
||||
if os.path.exists(cfg_file_path): |
||||
return |
||||
with open(cfg_file_path, 'w+') as fp: |
||||
fp.write('regtest=1\n') |
||||
fp.write('log-level=4\n') |
||||
fp.write('keep-fakechain=1\n') |
||||
fp.write('data-dir={}\n'.format(node_dir)) |
||||
fp.write('fixed-difficulty=1\n') |
||||
fp.write('p2p-bind-port={}\n'.format(WOW_BASE_PORT + node_id)) |
||||
fp.write('rpc-bind-port={}\n'.format(WOW_BASE_RPC_PORT + node_id)) |
||||
fp.write('p2p-bind-ip=127.0.0.1\n') |
||||
fp.write('rpc-bind-ip=127.0.0.1\n') |
||||
fp.write('prune-blockchain=1\n') |
||||
fp.write('zmq-rpc-bind-port={}\n'.format(WOW_BASE_ZMQ_PORT + node_id)) |
||||
fp.write('zmq-rpc-bind-ip=127.0.0.1\n') |
||||
|
||||
for i in range(0, NUM_NODES): |
||||
if node_id == i: |
||||
continue |
||||
fp.write('add-exclusive-node=127.0.0.1:{}\n'.format(WOW_BASE_PORT + i)) |
||||
|
||||
|
||||
def waitForWOWNode(rpc_offset, max_tries=7, auth=None): |
||||
for i in range(max_tries + 1): |
||||
try: |
||||
if auth is None: |
||||
callrpc_xmr(WOW_BASE_RPC_PORT + rpc_offset, 'get_block_count') |
||||
else: |
||||
callrpc_xmr(WOW_BASE_WALLET_RPC_PORT + rpc_offset, 'get_languages', auth=auth) |
||||
return |
||||
except Exception as ex: |
||||
if i < max_tries: |
||||
logging.warning('Can\'t connect to WOW%s RPC: %s. Retrying in %d second/s.', '' if auth is None else ' wallet', str(ex), (i + 1)) |
||||
time.sleep(i + 1) |
||||
raise ValueError('waitForWOWNode failed') |
||||
|
||||
|
||||
class Test(BaseTest): |
||||
__test__ = True |
||||
test_coin = Coins.WOW |
||||
wow_daemons = [] |
||||
wow_wallet_auth = [] |
||||
start_ltc_nodes = False |
||||
start_xmr_nodes = False |
||||
wow_addr = None |
||||
extra_wait_time = 0 |
||||
|
||||
@classmethod |
||||
def prepareExtraCoins(cls): |
||||
pass |
||||
num_blocks = 300 |
||||
cls.wow_addr = cls.callwownodewallet(cls, 1, 'get_address')['address'] |
||||
if callrpc_xmr(WOW_BASE_RPC_PORT + 1, 'get_block_count')['count'] < num_blocks: |
||||
logging.info('Mining %d Wownero blocks to %s.', num_blocks, cls.wow_addr) |
||||
callrpc_xmr(WOW_BASE_RPC_PORT + 1, 'generateblocks', {'wallet_address': cls.wow_addr, 'amount_of_blocks': num_blocks}) |
||||
logging.info('WOW blocks: %d', callrpc_xmr(WOW_BASE_RPC_PORT + 1, 'get_block_count')['count']) |
||||
|
||||
@classmethod |
||||
def tearDownClass(cls): |
||||
logging.info('Finalising Wownero Test') |
||||
super(Test, cls).tearDownClass() |
||||
|
||||
stopDaemons(cls.wow_daemons) |
||||
cls.wow_daemons.clear() |
||||
|
||||
@classmethod |
||||
def coins_loop(cls): |
||||
super(Test, cls).coins_loop() |
||||
|
||||
if cls.wow_addr is not None: |
||||
callrpc_xmr(WOW_BASE_RPC_PORT + 0, 'generateblocks', {'wallet_address': cls.wow_addr, 'amount_of_blocks': 1}) |
||||
|
||||
@classmethod |
||||
def prepareExtraDataDir(cls, i): |
||||
if not cls.restore_instance: |
||||
prepareWOWDataDir(cfg.TEST_DATADIRS, i, 'wownerod.conf') |
||||
|
||||
node_dir = os.path.join(cfg.TEST_DATADIRS, 'wow_' + str(i)) |
||||
cls.wow_daemons.append(startXmrDaemon(node_dir, WOW_BINDIR, WOWD)) |
||||
logging.info('Started %s %d', WOWD, cls.wow_daemons[-1].handle.pid) |
||||
waitForWOWNode(i) |
||||
|
||||
opts = [ |
||||
'--daemon-address=127.0.0.1:{}'.format(WOW_BASE_RPC_PORT + i), |
||||
'--no-dns', |
||||
'--rpc-bind-port={}'.format(WOW_BASE_WALLET_RPC_PORT + i), |
||||
'--wallet-dir={}'.format(os.path.join(node_dir, 'wallets')), |
||||
'--log-file={}'.format(os.path.join(node_dir, 'wallet.log')), |
||||
'--rpc-login=test{0}:test_pass{0}'.format(i), |
||||
'--wow-shared-ringdb-dir={}'.format(os.path.join(node_dir, 'shared-ringdb')), |
||||
'--allow-mismatched-daemon-version', |
||||
] |
||||
cls.wow_daemons.append(startXmrWalletDaemon(node_dir, WOW_BINDIR, WOW_WALLET_RPC, opts=opts)) |
||||
|
||||
cls.wow_wallet_auth.append(('test{0}'.format(i), 'test_pass{0}'.format(i))) |
||||
|
||||
waitForWOWNode(i, auth=cls.wow_wallet_auth[i]) |
||||
|
||||
if not cls.restore_instance: |
||||
logging.info('Creating WOW wallet %i', i) |
||||
cls.callwownodewallet(cls, i, 'create_wallet', {'filename': 'testwallet', 'language': 'English'}) |
||||
cls.callwownodewallet(cls, i, 'open_wallet', {'filename': 'testwallet'}) |
||||
|
||||
@classmethod |
||||
def addPIDInfo(cls, sc, i): |
||||
sc.setDaemonPID(Coins.WOW, cls.wow_daemons[i].handle.pid) |
||||
|
||||
@classmethod |
||||
def addCoinSettings(cls, settings, datadir, node_id): |
||||
settings['chainclients']['wownero'] = { |
||||
'connection_type': 'rpc', |
||||
'manage_daemon': False, |
||||
'rpcport': WOW_BASE_RPC_PORT + node_id, |
||||
'walletrpcport': WOW_BASE_WALLET_RPC_PORT + node_id, |
||||
'walletrpcuser': 'test' + str(node_id), |
||||
'walletrpcpassword': 'test_pass' + str(node_id), |
||||
'walletfile': 'testwallet', |
||||
'datadir': os.path.join(datadir, 'xmr_' + str(node_id)), |
||||
'bindir': WOW_BINDIR, |
||||
} |
||||
|
||||
def callwownodewallet(self, node_id, method, params=None): |
||||
return callrpc_xmr(WOW_BASE_WALLET_RPC_PORT + node_id, method, params, auth=self.wow_wallet_auth[node_id]) |
||||
|
||||
def test_01_ads_part_coin(self): |
||||
run_test_ads_success_path(self, Coins.PART, self.test_coin) |
||||
|
||||
def test_02_ads_coin_part(self): |
||||
# Reverse bid |
||||
run_test_ads_success_path(self, self.test_coin, Coins.PART) |
||||
|
||||
def test_03_ads_part_coin_both_refund(self): |
||||
run_test_ads_both_refund(self, Coins.PART, self.test_coin, lock_value=20) |
||||
|
||||
def test_04_ads_coin_part_both_refund(self): |
||||
# Reverse bid |
||||
run_test_ads_both_refund(self, self.test_coin, Coins.PART, lock_value=20) |
||||
|
||||
def test_05_ads_part_coin_swipe_refund(self): |
||||
run_test_ads_swipe_refund(self, Coins.PART, self.test_coin, lock_value=20) |
||||
|
||||
def test_06_ads_coin_part_swipe_refund(self): |
||||
# Reverse bid |
||||
run_test_ads_swipe_refund(self, self.test_coin, Coins.PART, lock_value=20) |
||||
Loading…
Reference in new issue