ut_addrparse.py 2.8 KB

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