test.py ref ref_altcoin: minor cleanups

This commit is contained in:
The MMGen Project 2023-04-18 18:35:52 +00:00
commit af97327541
Signed by: mmgen
GPG key ID: 3F8B1861E32B7DA2
2 changed files with 7 additions and 7 deletions

View file

@ -210,7 +210,6 @@ class TestSuiteRef(TestSuiteBase,TestSuiteShared):
subdir = None,
pfx = None,
mmtype = None,
add_args = [],
id_key = None,
pat = None ):
@ -220,7 +219,7 @@ class TestSuiteRef(TestSuiteBase,TestSuiteShared):
af = joinpath(ref_dir,(subdir or self.ref_subdir,'')[ftype=='passwd'],af_fn)
coin_arg = [] if coin == None else ['--coin='+coin]
tool_cmd = ftype.replace('segwit','').replace('bech32','')+'file_chksum'
t = self.spawn('mmgen-tool',coin_arg+['--verbose','-p1',tool_cmd,af]+add_args)
t = self.spawn( 'mmgen-tool', coin_arg + ['--verbose','-p1',tool_cmd,af] )
if ftype == 'keyaddr':
t.do_decrypt_ka_data(hp=ref_kafile_hash_preset,pw=ref_kafile_pass,have_yes_opt=True)
chksum_key = '_'.join([af_key,'chksum'] + ([coin.lower()] if coin else []) + ([mmtype] if mmtype else []))

View file

@ -114,19 +114,20 @@ class TestSuiteRefAltcoin(TestSuiteRef,TestSuiteBase):
stop_test_daemons(proto.network_id)
return 'ok'
def ref_altcoin_addrgen(self,coin,mmtype,gen_what='addr',coin_suf='',add_args=[]):
def ref_altcoin_addrgen(self,coin,mmtype,gen_what='addr',coin_suf='',add_args=[],addr_idx_list=None):
wf = dfl_words_file
t = self.spawn(f'mmgen-{gen_what}gen',
t = self.spawn(
'mmgen-keygen' if 'key' in gen_what else 'mmgen-addrgen',
['-Sq','--coin='+coin] +
(['--type='+mmtype] if mmtype else []) +
add_args +
[wf,dfl_addr_idx_list])
if gen_what == 'key':
[wf,addr_idx_list or dfl_addr_idx_list])
if 'key' in gen_what:
t.expect('Encrypt key list? (y/N): ','N')
chk = t.expect_getend(r'.* data checksum for \S*: ',regex=True)
chk_ref = self.chk_data[
'ref_{}addrfile_chksum_{}{}'.format(
('key' if gen_what == 'key' else ''),
(gen_what if 'key' in gen_what else ''),
coin.lower(),
coin_suf )
]