mmgen-wallet/test/unit_tests_d/ut_addrlist.py

#!/usr/bin/env python3

"""
test.unit_tests_d.ut_addrlist: address list unit tests for the MMGen suite
"""

from mmgen.common import *
from mmgen.seed import Seed
from mmgen.addr import MMGenAddrType
from mmgen.addrlist import AddrIdxList,AddrList,KeyList,KeyAddrList
from mmgen.passwdlist import PasswordList
from mmgen.protocol import init_proto

def do_test(list_type,chksum,idx_spec=None,pw_id_str=None,add_kwargs=None):
	qmsg(blue(f'Testing {list_type.__name__}'))
	proto = init_proto('btc')
	seed = Seed(seed_bin=bytes.fromhex('feedbead'*8))
	mmtype = MMGenAddrType(proto,'C')
	idxs = AddrIdxList(idx_spec or '1-3')

	if opt.verbose:
		debug_addrlist_save = g.debug_addrlist
		g.debug_addrlist = True

	kwargs = {
		'seed': seed,
		'pw_idxs': idxs,
		'pw_id_str': pw_id_str,
		'pw_fmt': 'b58',
	} if pw_id_str else {
		'seed': seed,
		'addr_idxs': idxs,
		'mmtype': mmtype,
	}

	if add_kwargs:
		kwargs.update(add_kwargs)

	al = list_type( proto, **kwargs )

	af = al.get_file()
	af.format()

	qmsg(f'Filename: {af.filename}\n')
#	af.write('-')
	vmsg(f'------------\n{af.fmt_data}\n------------')

	if chksum:
		assert al.chksum == chksum, f'{al.chksum} != {chksum}'

	if opt.verbose:
		g.debug_addrlist = debug_addrlist_save

	return True

class unit_tests:

	def idxlist(self,name,ut):
		for i,o in (
				('99,88-102,1-3,4,9,818,444-445,816',        '1-4,9,88-102,444-445,816,818'),
				('99,88-99,100,102,4-7,9,818,444-445,816,1', '1,4-7,9,88-100,102,444-445,816,818'),
				('8',             '8'),
				('2-4',           '2-4'),
				('1,2-4',         '1-4'),
				('2-4,1-9,9,1,8', '1-9'),
				('2-4,1',         '1-4'),
				('2-2',           '2'),
				('2,2',           '2'),
				('2-3',           '2-3'),
				('2,3',           '2-3'),
				('3,2',           '2-3'),
				('2,4',           '2,4'),
				('',              ''),
			):
			l = AddrIdxList(i)
			if opt.verbose:
				msg('list: {}\nin:   {}\nout:  {}\n'.format(list(l),i,o))
			assert l.id_str == o, f'{l.id_str} != {o}'

		return True

	def addr(self,name,ut):
		return (
			do_test(AddrList,'BCE8 082C 0973 A525','1-3') and
			do_test(AddrList,'88FA B04B A380 C1CB','199999,99-101,77-78,7,3,2-9')
		)

	def key(self,name,ut):
		return do_test(KeyList,None)

	def keyaddr(self,name,ut):
		return do_test(KeyAddrList,'4A36 AA65 8C2B 7C35')

	def passwd(self,name,ut):
		return do_test(PasswordList,'FF4A B716 4513 8F8F',pw_id_str='foo')

	def passwd_bip39(self,name,ut):
		return do_test(PasswordList,'C3A8 B2B2 1AA1 FB40',pw_id_str='foo',add_kwargs={'pw_fmt':'bip39'})