ut_tx.py 7.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253
  1. #!/usr/bin/env python3
  2. """
  3. test.modtest_d.ut_tx: TX unit tests for the MMGen suite
  4. """
  5. import os
  6. from mmgen.tx import CompletedTX, UnsignedTX
  7. from mmgen.tx.file import MMGenTxFile
  8. from mmgen.cfg import Config
  9. from ..include.common import cfg, qmsg, vmsg, gr_uc, make_burn_addr
  10. async def do_txfile_test(desc, fns, cfg=cfg, check=False):
  11. qmsg(f'\n Testing CompletedTX initializer ({desc})')
  12. for fn in fns:
  13. qmsg(f' parsing: {os.path.basename(fn)}')
  14. fpath = os.path.join('test', 'ref', fn)
  15. tx = await CompletedTX(cfg=cfg, filename=fpath, quiet_open=True)
  16. vmsg('\n' + tx.info.format())
  17. f = MMGenTxFile(tx)
  18. fn_gen = f.make_filename()
  19. if cfg.debug_utf8:
  20. fn_gen = fn_gen.replace('-α', '')
  21. assert fn_gen == os.path.basename(fn), f'{fn_gen} != {fn}'
  22. if check:
  23. import json
  24. from mmgen.tx.file import json_dumps
  25. from mmgen.util import make_chksum_6
  26. text = f.format()
  27. with open(fpath) as fh:
  28. text_chk = fh.read()
  29. data_chk = json.loads(text_chk)
  30. outputs = data_chk['MMGenTransaction']['outputs']
  31. for n, o in enumerate(outputs):
  32. outputs[n] = {k:v for k,v in o.items() if not (type(v) is bool and v is False)}
  33. data_chk['chksum'] = make_chksum_6(json_dumps(data_chk['MMGenTransaction']))
  34. text_chk_fixed = json_dumps(data_chk)
  35. assert text == text_chk_fixed, f'\nformatted text:\n{text}\n !=\noriginal file:\n{text_chk_fixed}'
  36. qmsg(' OK')
  37. return True
  38. class unit_tests:
  39. altcoin_deps = ('txfile_alt', 'txfile_alt_legacy')
  40. async def txfile(self, name, ut, desc='displaying transaction files (BTC)'):
  41. return await do_txfile_test(
  42. 'Bitcoin',
  43. (
  44. 'tx/7A8157[6.65227,34].rawtx',
  45. 'tx/BB3FD2[7.57134314,123].sigtx',
  46. 'tx/0A869F[1.23456,32].regtest.asubtx',
  47. ),
  48. check = True
  49. )
  50. async def txfile_alt(self, name, ut, desc='displaying transaction files (LTC, BCH, ETH)'):
  51. return await do_txfile_test(
  52. 'altcoins',
  53. (
  54. 'tx/C09D73-LTC[981.73747,2000].testnet.rawtx',
  55. 'tx/91060A-BCH[1.23456].regtest.arawtx',
  56. 'tx/D850C6-MM1[43.21,50000].subtx', # token tx
  57. ),
  58. # token resolved by tracking wallet under data_dir:
  59. cfg = Config({'data_dir': 'test/ref/data_dir'}),
  60. check = True
  61. )
  62. async def txfile_legacy(self, name, ut, desc='displaying transaction files (legacy format, BTC)'):
  63. return await do_txfile_test(
  64. 'Bitcoin - legacy file format',
  65. (
  66. '0B8D5A[15.31789,14,tl=1320969600].rawtx',
  67. '542169[5.68152,34].sigtx',
  68. '0C7115[15.86255,14,tl=1320969600].testnet.rawtx',
  69. '25EFA3[2.34].testnet.rawtx',
  70. )
  71. )
  72. async def txfile_alt_legacy(self, name, ut, desc='displaying transaction files (legacy format, LTC, BCH, ETH)'):
  73. return await do_txfile_test(
  74. 'altcoins - legacy file format',
  75. (
  76. '460D4D-BCH[10.19764,tl=1320969600].rawtx',
  77. 'ethereum/5881D2-MM1[1.23456,50000].rawtx',
  78. 'ethereum/6BDB25-MM1[1.23456,50000].testnet.rawtx',
  79. 'ethereum/88FEFD-ETH[23.45495,40000].rawtx',
  80. 'ethereum/B472BD-ETH[23.45495,40000].testnet.rawtx',
  81. 'ethereum/B472BD-ETH[23.45495,40000].testnet.sigtx',
  82. 'litecoin/A5A1E0-LTC[1454.64322,1453,tl=1320969600].testnet.rawtx',
  83. 'litecoin/AF3CDF-LTC[620.76194,1453,tl=1320969600].rawtx',
  84. )
  85. )
  86. def errors(self, name, ut, desc='reading transaction files (error handling)'):
  87. async def bad1():
  88. await CompletedTX(cfg, filename='foo')
  89. def bad2():
  90. UnsignedTX(cfg, filename='foo')
  91. bad_data = (
  92. ('forbidden positional args', 'TypeError', 'positional arguments', bad1),
  93. ('forbidden positional args', 'TypeError', 'positional arguments', bad2),
  94. )
  95. ut.process_bad_data(bad_data)
  96. return True
  97. def op_return_data(self, name, ut, desc='OpReturnData class'):
  98. from mmgen.proto.btc.tx.op_return_data import OpReturnData
  99. vecs = [
  100. 'data:=:ETH.ETH:0x86d526d6624AbC0178cF7296cD538Ecc080A95F1:0/1/0',
  101. 'hexdata:3d3a4554482e4554483a30783836643532366436363234416243303137'
  102. '38634637323936634435333845636330383041393546313a302f312f30',
  103. 'hexdata:00010203040506',
  104. 'data:a\n',
  105. 'data:a\tb',
  106. 'data:' + gr_uc[:24],
  107. ]
  108. assert OpReturnData(cfg._proto, vecs[0]) == OpReturnData(cfg._proto, vecs[1])
  109. for vec in vecs:
  110. d = OpReturnData(cfg._proto, vec)
  111. assert d == OpReturnData(cfg._proto, repr(d)) # repr() must return a valid initializer
  112. assert isinstance(d, bytes)
  113. assert isinstance(str(d), str)
  114. vmsg('-' * 80)
  115. vmsg(vec)
  116. vmsg(repr(d))
  117. vmsg(d.hl())
  118. vmsg(d.hl(add_label=True))
  119. bad_data = [
  120. 'data:',
  121. 'hexdata:',
  122. 'data:' + ('x' * 81),
  123. 'hexdata:' + ('deadbeef' * 20) + 'ee',
  124. 'hex:0abc',
  125. 'da:xyz',
  126. 'hexdata:xyz',
  127. 'hexdata:abcde',
  128. b'data:abc',
  129. ]
  130. def bad(n):
  131. return lambda: OpReturnData(cfg._proto, bad_data[n])
  132. vmsg('-' * 80)
  133. vmsg('Testing error handling:')
  134. ut.process_bad_data((
  135. ('bad1', 'AssertionError', 'not in range', bad(0)),
  136. ('bad2', 'AssertionError', 'not in range', bad(1)),
  137. ('bad3', 'AssertionError', 'not in range', bad(2)),
  138. ('bad4', 'AssertionError', 'not in range', bad(3)),
  139. ('bad5', 'ValueError', 'must start', bad(4)),
  140. ('bad6', 'ValueError', 'must start', bad(5)),
  141. ('bad7', 'AssertionError', 'not in hex', bad(6)),
  142. ('bad8', 'AssertionError', 'even', bad(7)),
  143. ('bad9', 'AssertionError', 'a string', bad(8)),
  144. ), pfx='')
  145. return True
  146. def memo(self, name, ut, desc='Swap transaction memo'):
  147. from mmgen.protocol import init_proto
  148. from mmgen.swap.proto.thorchain.memo import Memo
  149. for coin, addrtype in (
  150. ('ltc', 'bech32'),
  151. ('bch', 'compressed'),
  152. ):
  153. proto = init_proto(cfg, coin, need_amt=True)
  154. addr = make_burn_addr(proto, addrtype)
  155. for limit, limit_chk in (
  156. ('123.4567', 12340000000),
  157. ('1.234567', 123400000),
  158. ('0.01234567', 1234000),
  159. ('0.00012345', 12345),
  160. (None, 0),
  161. ):
  162. vmsg('\nTesting memo initialization:')
  163. m = Memo(proto, addr, trade_limit=proto.coin_amt(limit) if limit else None)
  164. vmsg(f'str(memo): {m}')
  165. vmsg(f'repr(memo): {m!r}')
  166. vmsg(f'limit: {limit}')
  167. p = Memo.parse(m)
  168. limit_dec = proto.coin_amt(p.trade_limit, from_unit='satoshi')
  169. vmsg(f'limit_dec: {limit_dec.hl()}')
  170. vmsg('\nTesting memo parsing:')
  171. from pprint import pformat
  172. vmsg(pformat(p._asdict()))
  173. assert p.proto == 'THORChain'
  174. assert p.function == 'SWAP'
  175. assert p.chain == coin.upper()
  176. assert p.asset == coin.upper()
  177. assert p.address == addr.views[addr.view_pref]
  178. assert p.trade_limit == limit_chk
  179. assert p.stream_interval == 1
  180. assert p.stream_quantity == 0 # auto
  181. vmsg('\nTesting is_partial_memo():')
  182. for vec in (
  183. str(m),
  184. 'SWAP:xyz',
  185. '=:xyz',
  186. 's:xyz',
  187. 'a:xz',
  188. '+:xz',
  189. 'WITHDRAW:xz',
  190. 'LOAN+:xz:x:x',
  191. 'TRADE-:xz:x:x',
  192. 'BOND:xz',
  193. ):
  194. vmsg(f' pass: {vec}')
  195. assert Memo.is_partial_memo(vec), vec
  196. for vec in (
  197. '=',
  198. 'swap',
  199. 'swap:',
  200. 'swap:abc',
  201. 'SWAP:a',
  202. ):
  203. vmsg(f' fail: {vec}')
  204. assert not Memo.is_partial_memo(vec), vec
  205. vmsg('\nTesting error handling:')
  206. def bad(s):
  207. return lambda: Memo.parse(s)
  208. ut.process_bad_data((
  209. ('bad1', 'SwapMemoParseError', 'must contain', bad('x')),
  210. ('bad2', 'SwapMemoParseError', 'must contain', bad('y:z:x')),
  211. ('bad3', 'SwapMemoParseError', 'function abbrev', bad('z:l:foobar:0/1/0')),
  212. ('bad4', 'SwapMemoParseError', 'asset abbrev', bad('=:x:foobar:0/1/0')),
  213. ('bad5', 'SwapMemoParseError', 'failed to parse', bad('=:l:foobar:n')),
  214. ('bad6', 'SwapMemoParseError', 'invalid specifier', bad('=:l:foobar:x/1/0')),
  215. ('bad7', 'SwapMemoParseError', 'extra', bad('=:l:foobar:0/1/0:x')),
  216. ), pfx='')
  217. return True