mmgen-wallet/test/modtest_d/ut_mn_entry.py

#!/usr/bin/env python3

"""
test.modtest_d.ut_mn_entry: Mnemonic user entry unit test for the MMGen suite
"""

from mmgen.mn_entry import mn_entry
from ..include.common import cfg, vmsg

class unit_tests:

	vectors = {
		'mmgen':   {
			'usl': 10, 'sw': 3, 'lw': 12,
			'idx_minimal': ( # None: non-unique match. False: no match
				('a',          None),
				('aa',         False),
				('as',         None),
				('ask',        70),
				('afte',       None),
				('after',      None),
				('aftern',     20),
				('afternoon',  20),
				('afternoons', False),
				('g',          None),
				('gg',         False),
				('z',          False),
				('abi',        False),
				('abo',        None),
				('abl',        0),
				('able',       0),
				('abler',      False),
				('you',        None),
				('yout',       1625),
				('youth',      1625),
				('youths',     False),
			),
		},
		'xmrseed': {'usl': 3, 'sw': 4, 'lw': 12},
		'bip39':   {'usl': 4, 'sw': 3, 'lw': 8},
	}

	def wl(self, name, ut, desc='MnemonicEntry - computed wordlist constants'):
		for wl_id in self.vectors:
			for j, k in (
					('uniq_ss_len',   'usl'),
					('shortest_word', 'sw'),
					('longest_word',  'lw')):
				a = getattr(mn_entry(cfg, wl_id), j)
				b = self.vectors[wl_id][k]
				assert a == b, f'{wl_id}:{j} {a} != {b}'
		return True

	def idx(self, name, ut, desc='MnemonicEntry - idx()'):
		junk = 'a g z aa gg zz aaa ggg zzz aaaa gggg zzzz aaaaaaaaaaaaaa gggggggggggggg zzzzzzzzzzzzzz'
		for wl_id in self.vectors:
			m = mn_entry(cfg, wl_id)
			vmsg('Wordlist: '+wl_id)
			for entry_mode in ('full', 'short'):
				for a, word in enumerate(m.wl):
					b = m.idx(word, entry_mode)
					assert a == b, f'{a} != {b} ({word!r} - entry mode: {entry_mode!r})'
				a = None
				for word in junk.split():
					b = m.idx(word, entry_mode)
					assert a == b, f'{a} != {b} ({word!r} - entry mode: {entry_mode!r})'
			if 'idx_minimal' in self.vectors[wl_id]:
				for vec in self.vectors[wl_id]['idx_minimal']:
					chk = vec[1]
					b = m.idx(vec[0], 'minimal')
					if chk is False:
						assert b is None, (b, None)
					elif chk is None:
						assert type(b) is tuple, (type(b), tuple)
					elif type(chk) is int:
						assert b == chk, (b, chk)
		return True