cmdtest.py: make test methods async where applicable

This commit is contained in:
The MMGen Project 2025-10-01 15:30:56 +00:00
commit dc3df437ab
Signed by: mmgen
GPG key ID: 3F8B1861E32B7DA2
7 changed files with 49 additions and 53 deletions

View file

@ -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+')

View file

@ -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])

View 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)

View file

@ -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)

View file

@ -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:

View file

@ -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

View file

@ -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))