cmdtest.py: make test methods async where applicable
This commit is contained in:
parent
6ccf29fb21
commit
dc3df437ab
7 changed files with 49 additions and 53 deletions
|
|
@ -91,8 +91,8 @@ class CmdTestAutosignETH(CmdTestAutosignThreaded, CmdTestEthdev, CmdTestEthdevMe
|
|||
def token_addrimport(self):
|
||||
return self._token_addrimport('token_addr1', '11-13', expect='3/3')
|
||||
|
||||
def token_fund_user(self):
|
||||
return self._token_transfer_ops(op='fund_user', mm_idxs=[11])
|
||||
async def token_fund_user(self):
|
||||
return await self._token_transfer_ops(op='fund_user', mm_idxs=[11])
|
||||
|
||||
def token_bal1(self):
|
||||
return self._token_bal_check(pat=r':E:11\s+1000\s+54\.321\s+')
|
||||
|
|
|
|||
|
|
@ -70,8 +70,8 @@ class CmdTestEthBumpMethods:
|
|||
fee_desc = 'or gas price',
|
||||
bad_fee = '0.9G')
|
||||
|
||||
def _token_fund_user(self, *, mm_idxs):
|
||||
return self._token_transfer_ops(
|
||||
async def _token_fund_user(self, *, mm_idxs):
|
||||
return await self._token_transfer_ops(
|
||||
op = 'fund_user',
|
||||
mm_idxs = mm_idxs,
|
||||
token_addr = 'token_addr1',
|
||||
|
|
@ -330,11 +330,11 @@ class CmdTestEthBump(CmdTestEthBumpMethods, CmdTestEthSwapMethods, CmdTestSwapMe
|
|||
def bal4(self):
|
||||
return self._bal_check(pat=rf'{dfl_sid}:E:12\s+4444\.3333\s')
|
||||
|
||||
def token_fund_user1(self):
|
||||
return self._token_fund_user(mm_idxs=[1])
|
||||
async def token_fund_user1(self):
|
||||
return await self._token_fund_user(mm_idxs=[1])
|
||||
|
||||
def token_fund_user11(self):
|
||||
return self._token_fund_user(mm_idxs=[11])
|
||||
async def token_fund_user11(self):
|
||||
return await self._token_fund_user(mm_idxs=[11])
|
||||
|
||||
def token_txdo1(self):
|
||||
return self._token_txcreate(cmd='txdo', args=[f'{dfl_sid}:E:2,1.23456', dfl_words_file])
|
||||
|
|
|
|||
|
|
@ -1390,11 +1390,11 @@ class CmdTestEthdev(CmdTestEthdevMethods, CmdTestBase, CmdTestShared):
|
|||
def bal6(self):
|
||||
return self.bal5()
|
||||
|
||||
def token_fund_users(self):
|
||||
return self._token_transfer_ops(op='fund_user', mm_idxs=[11, 21])
|
||||
async def token_fund_users(self):
|
||||
return await self._token_transfer_ops(op='fund_user', mm_idxs=[11, 21])
|
||||
|
||||
def token_user_bals(self):
|
||||
return self._token_transfer_ops(op='show_bals', mm_idxs=[11, 21])
|
||||
async def token_user_bals(self):
|
||||
return await self._token_transfer_ops(op='show_bals', mm_idxs=[11, 21])
|
||||
|
||||
def token_addrgen(self):
|
||||
return self._token_addrgen(mm_idxs=[11, 21], naddrs=3)
|
||||
|
|
|
|||
|
|
@ -71,8 +71,8 @@ class CmdTestEthSwapMethods:
|
|||
gas = 1_000_000,
|
||||
fn = f'{self.tmpdir}/THORChain_Router.bin')
|
||||
|
||||
def token_fund_user(self):
|
||||
return self._token_transfer_ops(
|
||||
async def token_fund_user(self):
|
||||
return await self._token_transfer_ops(
|
||||
op = 'fund_user',
|
||||
mm_idxs = [1, 2, 12],
|
||||
token_addr = 'token_addr1',
|
||||
|
|
@ -445,10 +445,10 @@ class CmdTestEthSwapEth(CmdTestEthSwapMethods, CmdTestSwapMethods, CmdTestEthdev
|
|||
def swaptxstatus1(self):
|
||||
return self._swaptxsend(add_opts=['--verbose', '--status'], status=True)
|
||||
|
||||
def swaptxmemo4(self):
|
||||
async def swaptxmemo4(self):
|
||||
import time
|
||||
time.sleep(1)
|
||||
return self._check_token_swaptx_memo('=:b:mkQsXA7mqDtnUpkaXMbDtAL1KMeof4GPw3:0/3/0')
|
||||
return await self._check_token_swaptx_memo('=:b:mkQsXA7mqDtnUpkaXMbDtAL1KMeof4GPw3:0/3/0')
|
||||
|
||||
def swaptxreceipt4(self):
|
||||
return self._swaptxsend(add_opts=['--receipt'], spawn_only=True)
|
||||
|
|
|
|||
|
|
@ -719,14 +719,14 @@ class CmdTestRegtest(CmdTestBase, CmdTestShared):
|
|||
tweaks = ['confirm_chg_non_mmgen'],
|
||||
skip_passphrase = skip_passphrase)
|
||||
|
||||
def fund_bob_deterministic(self):
|
||||
return self.fund_wallet_deterministic(f'{self._user_sid("bob")}:C:1', '1-11')
|
||||
async def fund_bob_deterministic(self):
|
||||
return await self.fund_wallet_deterministic(f'{self._user_sid("bob")}:C:1', '1-11')
|
||||
|
||||
def fund_alice_deterministic(self):
|
||||
async def fund_alice_deterministic(self):
|
||||
sid = self._user_sid('alice')
|
||||
mmtype = ('L', 'S')[self.proto.cap('segwit')]
|
||||
addr = self.get_addr_from_addrlist('alice', sid, mmtype, 0, addr_range='1-5')
|
||||
return self.fund_wallet_deterministic(addr, '1-11', skip_passphrase=True)
|
||||
return await self.fund_wallet_deterministic(addr, '1-11', skip_passphrase=True)
|
||||
|
||||
def generate_extra_deterministic(self):
|
||||
if not self.deterministic:
|
||||
|
|
|
|||
|
|
@ -214,17 +214,17 @@ class CmdTestXMRAutosign(CmdTestXMRWallet, CmdTestAutosignThreaded):
|
|||
def delete_dump_files(self):
|
||||
return self._delete_files('.dump')
|
||||
|
||||
def fund_alice1(self):
|
||||
return self.fund_alice(wallet=1)
|
||||
async def fund_alice1(self):
|
||||
return await self.fund_alice(wallet=1)
|
||||
|
||||
def check_bal_alice1(self):
|
||||
return self.check_bal_alice(wallet=1)
|
||||
async def check_bal_alice1(self):
|
||||
return await self.check_bal_alice(wallet=1)
|
||||
|
||||
def fund_alice2(self):
|
||||
return self.fund_alice(wallet=2)
|
||||
async def fund_alice2(self):
|
||||
return await self.fund_alice(wallet=2)
|
||||
|
||||
def check_bal_alice2(self):
|
||||
return self.check_bal_alice(wallet=2)
|
||||
async def check_bal_alice2(self):
|
||||
return await self.check_bal_alice(wallet=2)
|
||||
|
||||
def autosign_setup(self):
|
||||
return self.run_setup(
|
||||
|
|
@ -340,26 +340,19 @@ class CmdTestXMRAutosign(CmdTestXMRWallet, CmdTestAutosignThreaded):
|
|||
'1-2',
|
||||
lambda n, b, ub: b == ub and ((n == 1 and 0.8 < b < 0.86) or (n == 2 and b > 1.23)))
|
||||
|
||||
def _mine_chk(self, desc):
|
||||
bal_type = {'locked':'b', 'unlocked':'ub'}[desc]
|
||||
return self.mine_chk(
|
||||
'alice', 1, 0,
|
||||
lambda x: 0 < getattr(x, bal_type) < 1.234567891234,
|
||||
f'{desc} balance 0 < 1.234567891234')
|
||||
async def submit_transfer_tx1(self):
|
||||
return await self._submit_transfer_tx()
|
||||
|
||||
def submit_transfer_tx1(self):
|
||||
return self._submit_transfer_tx()
|
||||
|
||||
def resubmit_transfer_tx1(self):
|
||||
return self._submit_transfer_tx(
|
||||
async def resubmit_transfer_tx1(self):
|
||||
return await self._submit_transfer_tx(
|
||||
relay_parm = self.tx_relay_daemon_proxy_parm,
|
||||
op = 'resubmit',
|
||||
check_bal = False)
|
||||
|
||||
def submit_transfer_tx2(self):
|
||||
return self._submit_transfer_tx(relay_parm=self.tx_relay_daemon_parm)
|
||||
async def submit_transfer_tx2(self):
|
||||
return await self._submit_transfer_tx(relay_parm=self.tx_relay_daemon_parm)
|
||||
|
||||
def _submit_transfer_tx(self, relay_parm=None, ext=None, op='submit', check_bal=True):
|
||||
async def _submit_transfer_tx(self, relay_parm=None, ext=None, op='submit', check_bal=True):
|
||||
t = self._xmr_autosign_op(
|
||||
op = op,
|
||||
add_opts = [f'--tx-relay-daemon={relay_parm}'] if relay_parm else [],
|
||||
|
|
@ -372,7 +365,10 @@ class CmdTestXMRAutosign(CmdTestXMRWallet, CmdTestAutosignThreaded):
|
|||
self.remove_device_online() # device was inserted by _xmr_autosign_op()
|
||||
if check_bal:
|
||||
t.ok()
|
||||
return self._mine_chk('unlocked')
|
||||
return await self.mine_chk(
|
||||
'alice', 1, 0,
|
||||
lambda x: 0 < x.ub < 1.234567891234,
|
||||
'unlocked balance 0 < 1.234567891234')
|
||||
else:
|
||||
return t
|
||||
|
||||
|
|
|
|||
|
|
@ -505,25 +505,25 @@ class CmdTestXMRWallet(CmdTestBase):
|
|||
|
||||
return t if do_ret else t.ok()
|
||||
|
||||
def sweep_to_wallet(self):
|
||||
async def sweep_to_wallet(self):
|
||||
self.do_op('sweep', 'alice', '1:0,2')
|
||||
return self.mine_chk('alice', 2, 1, lambda x: x.ub > 1, 'unlocked balance > 1')
|
||||
return await self.mine_chk('alice', 2, 1, lambda x: x.ub > 1, 'unlocked balance > 1')
|
||||
|
||||
def sweep_to_account(self):
|
||||
async def sweep_to_account(self):
|
||||
self.do_op('sweep', 'alice', '2:1,2:0', use_existing=True)
|
||||
return self.mine_chk('alice', 2, 0, lambda x: x.ub > 1, 'unlocked balance > 1')
|
||||
return await self.mine_chk('alice', 2, 0, lambda x: x.ub > 1, 'unlocked balance > 1')
|
||||
|
||||
def sweep_to_wallet_account(self):
|
||||
async def sweep_to_wallet_account(self):
|
||||
self.do_op('sweep', 'alice', '2:0,3:0', use_existing=True, add_opts=['-Ee', '--full-address'])
|
||||
return self.mine_chk('alice', 3, 0, lambda x: x.ub > 1, 'unlocked balance > 1')
|
||||
return await self.mine_chk('alice', 3, 0, lambda x: x.ub > 1, 'unlocked balance > 1')
|
||||
|
||||
def sweep_to_wallet_account_proxy(self):
|
||||
async def sweep_to_wallet_account_proxy(self):
|
||||
self.do_op('sweep', 'alice', '3:0,2:1', self.tx_relay_daemon_proxy_parm, add_opts=['--priority=3', '-Ee'])
|
||||
return self.mine_chk('alice', 2, 1, lambda x: x.ub > 1, 'unlocked balance > 1')
|
||||
return await self.mine_chk('alice', 2, 1, lambda x: x.ub > 1, 'unlocked balance > 1')
|
||||
|
||||
def sweep_to_same_account_noproxy(self):
|
||||
async def sweep_to_same_account_noproxy(self):
|
||||
self.do_op('sweep', 'alice', '2:1', self.tx_relay_daemon_parm)
|
||||
return self.mine_chk('alice', 2, 1, lambda x: x.ub > 0.9, 'unlocked balance > 0.9')
|
||||
return await self.mine_chk('alice', 2, 1, lambda x: x.ub > 0.9, 'unlocked balance > 0.9')
|
||||
|
||||
async def transfer_to_miner_proxy(self):
|
||||
addr = read_from_file(self.users['miner'].addrfile_fs.format(2))
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue