ut_msg.py 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147
  1. #!/usr/bin/env python3
  2. """
  3. test.unit_tests_d.ut_msg: message signing unit tests for the MMGen suite
  4. """
  5. import os
  6. from test.include.common import silence,end_silence,restart_test_daemons,stop_test_daemons
  7. from mmgen.opts import opt
  8. from mmgen.util import msg,bmsg,pumsg
  9. from mmgen.protocol import CoinProtocol
  10. from mmgen.msg import NewMsg,UnsignedMsg,SignedMsg,SignedOnlineMsg,ExportedMsgSigs
  11. from mmgen.addr import MMGenID
  12. def get_obj(coin,network,msghash_type):
  13. if coin == 'bch':
  14. addrlists = 'DEADBEEF:C:1-20 98831F3A:C:8,2 A091ABAA:L:111 A091ABAA:C:1'
  15. elif coin == 'eth':
  16. addrlists = 'DEADBEEF:E:1-20 98831F3A:E:8,2 A091ABAA:E:111'
  17. else:
  18. # A091ABAA = 98831F3A:5S
  19. addrlists = 'DEADBEEF:C:1-20 98831F3A:B:8,2 A091ABAA:S:10-11 A091ABAA:111 A091ABAA:C:1'
  20. return NewMsg(
  21. coin = coin,
  22. network = network,
  23. message = '08/Jun/2021 Bitcoin Law Enacted by El Salvador Legislative Assembly',
  24. addrlists = addrlists,
  25. msghash_type = msghash_type )
  26. async def run_test(network_id,chksum,msghash_type='raw'):
  27. coin,network = CoinProtocol.Base.parse_network_id(network_id)
  28. if not opt.verbose:
  29. silence()
  30. bmsg(f'\nTesting {coin.upper()} {network.upper()}:\n')
  31. m = get_obj(coin,network,msghash_type)
  32. if m.proto.sign_mode == 'daemon':
  33. restart_test_daemons(network_id)
  34. pumsg('\nTesting data creation:\n')
  35. tmpdir = os.path.join('test','trash2')
  36. os.makedirs(tmpdir,exist_ok=True)
  37. assert m.chksum.upper() == chksum, f'{m.chksum.upper()} != {chksum}'
  38. m.write_to_file(
  39. outdir = tmpdir,
  40. ask_overwrite = False )
  41. pumsg('\nTesting signing:\n')
  42. m = UnsignedMsg( infile = os.path.join(tmpdir,get_obj(coin,network,msghash_type).filename) )
  43. await m.sign(wallet_files=['test/ref/98831F3A.mmwords'])
  44. m = SignedMsg( data=m.__dict__ )
  45. m.write_to_file(
  46. outdir = tmpdir,
  47. ask_overwrite = False )
  48. pumsg('\nTesting display:\n')
  49. m = SignedOnlineMsg( infile = os.path.join(tmpdir,get_obj(coin,network,msghash_type).signed_filename) )
  50. msg(m.format())
  51. single_addr = 'A091ABAA:E:111' if m.proto.base_proto == 'Ethereum' else 'A091ABAA:111'
  52. single_addr_coin = m.sigs[MMGenID(m.proto,single_addr)]['addr']
  53. pumsg('\nTesting single address display:\n')
  54. msg(m.format(single_addr))
  55. pumsg('\nTesting verification:\n')
  56. await m.verify(summary=opt.verbose)
  57. pumsg('\nTesting single address verification:\n')
  58. await m.verify(single_addr,summary=opt.verbose)
  59. pumsg('\nTesting JSON dump for export:\n')
  60. msg( m.get_json_for_export() )
  61. pumsg('\nTesting single address JSON dump for export:\n')
  62. msg( m.get_json_for_export(single_addr) )
  63. from mmgen.fileutil import write_data_to_file
  64. exported_sigs = os.path.join(tmpdir,'signatures.json')
  65. write_data_to_file(
  66. outfile = exported_sigs,
  67. data = m.get_json_for_export(),
  68. desc = 'signature data',
  69. ask_overwrite = False )
  70. m = ExportedMsgSigs( infile=exported_sigs )
  71. pumsg('\nTesting verification (exported data):\n')
  72. await m.verify(summary=opt.verbose)
  73. pumsg('\nTesting single address verification (exported data):\n')
  74. await m.verify(single_addr_coin,summary=opt.verbose)
  75. pumsg('\nTesting display (exported data):\n')
  76. msg(m.format())
  77. pumsg('\nTesting single address display (exported data):\n')
  78. msg(m.format(single_addr_coin))
  79. if m.proto.sign_mode == 'daemon':
  80. stop_test_daemons(network_id)
  81. msg('\n')
  82. if not opt.verbose:
  83. end_silence()
  84. return True
  85. class unit_tests:
  86. altcoin_deps = ('ltc','bch','eth','eth_raw')
  87. def btc(self,name,ut):
  88. return run_test('btc','AA0DB5')
  89. def btc_tn(self,name,ut):
  90. return run_test('btc_tn','A88E1D')
  91. def btc_rt(self,name,ut):
  92. return run_test('btc_rt','578018')
  93. def ltc(self,name,ut):
  94. return run_test('ltc','BA7549')
  95. def bch(self,name,ut):
  96. return run_test('bch','1B8065')
  97. def eth(self,name,ut):
  98. return run_test('eth','35BAD9',msghash_type='eth_sign')
  99. def eth_raw(self,name,ut):
  100. return run_test('eth','9D900C')