ut_addrparse.py 2.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697
  1. #!/usr/bin/env python3
  2. """
  3. test/unit_tests_d/ut_addrparse: address parsing tests for the MMGen suite
  4. """
  5. from mmgen.common import *
  6. vectors = {
  7. 'btc_mainnet': [
  8. {'std': '1C5VPtgq9xQ6AcTgMAR3J6GDrs72HC4pS1'},
  9. {'std': '3AhjTiWHhVJAi1s5CfKMcLzYps12x3gZhg'},
  10. {'std': 'bc1q6pqnfwwakuuejpm9w52ds342f9d5u36v0qnz7c'}
  11. ],
  12. 'ltc_mainnet': [
  13. {'std': 'LUbHQNYoy23RByq4dKQotLA4ugk9FhpAMT'},
  14. {'std': 'MCoZrHYPqYKqvpiwyzzqf3EPxF5no6puEf'},
  15. {'std': 'ltc1qvmqas4maw7lg9clqu6kqu9zq9cluvllnst5pxs'}
  16. ],
  17. 'xmr_mainnet': [
  18. { # ut_xmrseed.vectors[0]:
  19. 'std': '42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm',
  20. # https://github.com/monero-project/monero/tests/functional_tests/integrated_address.py
  21. 'int': '4CMe2PUhs4J4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfSbLRB61BQVATzerHGj',
  22. 'id': '0123456789abcdef'
  23. },{
  24. 'std': '46r4nYSevkfBUMhuykdK3gQ98XDqDTYW1hNLaXNvjpsJaSbNtdXh1sKMsdVgqkaihChAzEy29zEDPMR3NHQvGoZCLGwTerK',
  25. 'int': '4GYjoMG9Y2BBUMhuykdK3gQ98XDqDTYW1hNLaXNvjpsJaSbNtdXh1sKMsdVgqkaihChAzEy29zEDPMR3NHQvGoZCVSs1ZojwrDCGS5rUuo',
  26. 'id': '1122334455667788'
  27. }
  28. ],
  29. 'zec_mainnet': [
  30. {'std': 't1KQYLBvjpmcQuATommo6gx2QTQDLPikB8Q'},
  31. {'std': 'zceQDpyNwek7dKqF5ZuFGj7YrNVxh7X1aPkrVxDLVxWSiZAFDEuy5C7XNV8VhyZ3ghTPQ61xjCGiyLT3wqpiN1Yi6mdmaCq'},
  32. ],
  33. 'eth_mainnet': [
  34. {'std': '7e5f4552091a69125d5dfcb7b8c2659029395bdf'},
  35. ],
  36. }
  37. def test_network(proto,addrs):
  38. def check_equal(a,b):
  39. assert a == b, f'{a.hex()} != {b.hex()}'
  40. def check_bytes(addr):
  41. if addr.parsed.ver_bytes is not None:
  42. check_equal(
  43. addr.parsed.ver_bytes,
  44. proto.addr_fmt_to_ver_bytes.get(addr.addr_fmt) )
  45. check_equal(
  46. addr.parsed.data + ((addr.parsed.payment_id or b'') if proto.coin == 'XMR' else b''),
  47. addr.bytes )
  48. def fmt_addr_data(addr):
  49. return pp_fmt({k:(v.hex() if isinstance(v,bytes) else v) for k,v in addr.parsed._asdict().items()})
  50. def print_info(addr):
  51. vmsg('\n{}\n{}\n{}'.format(
  52. yellow(addr.addr_fmt),
  53. cyan(addr),
  54. fmt_addr_data(addr)))
  55. msg_r(f'Testing {proto.coin} address parsing...')
  56. vmsg('')
  57. from mmgen.addr import CoinAddr
  58. for addr in addrs:
  59. a1 = CoinAddr(proto,addr['std'])
  60. print_info(a1)
  61. check_bytes(a1)
  62. assert not hasattr(a1.parsed,'payment_id') or a1.parsed.payment_id == None
  63. if 'int' in addr:
  64. a2 = CoinAddr(proto,addr['int'])
  65. print_info(a2)
  66. check_bytes(a2)
  67. check_equal( a1.parsed.data, a2.parsed.data )
  68. check_equal( a2.parsed.payment_id, bytes.fromhex(addr['id']) )
  69. msg('OK')
  70. vmsg('')
  71. class unit_test(object):
  72. def run_test(self,name,ut):
  73. from mmgen.protocol import init_proto
  74. for net_id,addrs in vectors.items():
  75. coin,network = net_id.split('_')
  76. test_network(
  77. init_proto(coin,network=network),
  78. addrs )
  79. return True