From f9778043c27da7fee1310996ae354eb3c2171a63 Mon Sep 17 00:00:00 2001
From: The MMGen Project <mmgen@tuta.io>
Date: Thu, 24 Oct 2019 16:31:54 +0000
Subject: [PATCH] unit_tests.py: add test for base conversion routines

Run test:

    $ test/unit_tests.py -v baseconv
---
 test/unit_tests_d/ut_baseconv.py | 137 +++++++++++++++++++++++++++++++
 1 file changed, 137 insertions(+)
 create mode 100755 test/unit_tests_d/ut_baseconv.py

diff --git a/test/unit_tests_d/ut_baseconv.py b/test/unit_tests_d/ut_baseconv.py
new file mode 100755
index 00000000..b657369b
--- /dev/null
+++ b/test/unit_tests_d/ut_baseconv.py
@@ -0,0 +1,137 @@
+#!/usr/bin/env python3
+"""
+test/unit_tests_d/ut_baseconv.py: Base conversion unit test for the MMGen suite
+"""
+
+from mmgen.common import *
+from mmgen.exception import *
+
+class unit_test(object):
+
+	vectors = {
+		'b58': (
+			(('00',None),''),
+			(('00',1),'1'),
+			(('00',2),'11'),
+			(('01',None),'2'),
+			(('01',1),'2'),
+			(('01',2),'12'),
+			(('0f',None),'G'),
+			(('0f',1),'G'),
+			(('0f',2),'1G'),
+			(('deadbeef',None),'6h8cQN'),
+			(('deadbeef',20),'111111111111116h8cQN'),
+			(('00000000',None),''),
+			(('00000000',20),'11111111111111111111'),
+			(('ffffffff',None),'7YXq9G'),
+			(('ffffffff',20),'111111111111117YXq9G'),
+		),
+		# MMGen-flavored base32 using simple base conversion
+		'b32': (
+			(('00',None),''),
+			(('00',1),'A'),
+			(('00',2),'AA'),
+			(('01',None),'B'),
+			(('01',1),'B'),
+			(('01',2),'AB'),
+			(('0f',None),'P'),
+			(('0f',1),'P'),
+			(('0f',2),'AP'),
+			(('deadbeef',None),'DPK3PXP'),
+			(('deadbeef',20),'AAAAAAAAAAAAADPK3PXP'),
+			(('00000000',None),''),
+			(('00000000',20),'AAAAAAAAAAAAAAAAAAAA'),
+			(('ffffffff',None),'D777777'),
+			(('ffffffff',20),'AAAAAAAAAAAAAD777777'),
+		),
+		'b16': (
+			(('00',None),''),
+			(('00',1),'0'),
+			(('00',2),'00'),
+			(('01',None),'1'),
+			(('01',1),'1'),
+			(('01',2),'01'),
+			(('0f',None),'f'),
+			(('0f',1),'f'),
+			(('0f',2),'0f'),
+			(('deadbeef',None),'deadbeef'),
+			(('deadbeef',20),'000000000000deadbeef'),
+			(('00000000',None),''),
+			(('00000000',20),'00000000000000000000'),
+			(('ffffffff',None),'ffffffff'),
+			(('ffffffff',20),'000000000000ffffffff'),
+		),
+		'b10': (
+			(('00',None),''),
+			(('00',1),'0'),
+			(('00',2),'00'),
+			(('01',None),'1'),
+			(('01',1),'1'),
+			(('01',2),'01'),
+			(('0f',None),'15'),
+			(('0f',1),'15'),
+			(('0f',2),'15'),
+			(('deadbeef',None),'3735928559'),
+			(('deadbeef',20),'00000000003735928559'),
+			(('00000000',None),''),
+			(('00000000',20),'00000000000000000000'),
+			(('ffffffff',None),'4294967295'),
+			(('ffffffff',20),'00000000004294967295'),
+		),
+		'b8': (
+			(('00',None),''),
+			(('00',1),'0'),
+			(('00',2),'00'),
+			(('01',None),'1'),
+			(('01',1),'1'),
+			(('01',2),'01'),
+			(('0f',None),'17'),
+			(('0f',1),'17'),
+			(('0f',2),'17'),
+			(('deadbeef',None),'33653337357'),
+			(('deadbeef',20),'00000000033653337357'),
+			(('00000000',None),''),
+			(('00000000',20),'00000000000000000000'),
+			(('ffffffff',None),'37777777777'),
+			(('ffffffff',20),'00000000037777777777'),
+		),
+	}
+
+	def run_test(self,name):
+
+		msg_r('Testing base conversion routines...')
+
+		from mmgen.util import baseconv
+		perr = "length of {!r} less than pad length ({})"
+		rerr = "return value ({!r}) does not match reference value ({!r})"
+
+		qmsg_r('\nChecking hex-to-base conversion:')
+		fs = "  {h:10} {p:6} {r}"
+		for base,data in self.vectors.items():
+			if not opt.verbose: qmsg_r(' {}'.format(base))
+			vmsg('\nBase: {}'.format(base))
+			vmsg(fs.format(h='Input',p='Pad',r='Output'))
+			for (hexstr,pad),ret_chk in data:
+				ret = baseconv.fromhex(hexstr,wl_id=base,pad=pad,tostr=True)
+				assert len(ret) >= (pad or 0), perr.format(ret,pad)
+				assert ret == ret_chk, rerr.format(ret,ret_chk)
+				vmsg(fs.format(h=hexstr,r=ret,p=str(pad)))
+#				msg("(('{h}',{p}),'{r}'),".format(h=hexstr,r=ret,c=ret_chk,p=pad))
+#			msg('')
+#		return True
+		qmsg_r('\nChecking base-to-hex conversion:')
+		fs = "  {h:24} {p:<6} {r}"
+		for base,data in self.vectors.items():
+			if not opt.verbose: qmsg_r(' {}'.format(base))
+			vmsg('\nBase: {}'.format(base))
+			vmsg(fs.format(h='Input',p='Pad',r='Output'))
+			for (hexstr,pad),ret_chk in data:
+				ret = baseconv.tohex(ret_chk,wl_id=base,pad=len(hexstr))
+				assert ret == hexstr, rerr.format(ret,ret_chk)
+				vmsg(fs.format(h=ret_chk,r=ret,p=len(hexstr)))
+#				msg("(('{h}',{p}),'{r}'),".format(h=hexstr,r=ret_chk,c=ret_chk,p=pad))
+
+		qmsg('')
+		msg('OK')
+
+		return True