ut_rpc.py 6.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245
  1. #!/usr/bin/env python3
  2. """
  3. test.unit_tests_d.ut_rpc: RPC unit test for the MMGen suite
  4. """
  5. import sys,os,time
  6. from mmgen.cfg import Config
  7. from mmgen.color import yellow,cyan
  8. from mmgen.util import msg,gmsg,make_timestr,pp_fmt,die
  9. from mmgen.protocol import init_proto
  10. from mmgen.rpc import rpc_init
  11. from mmgen.daemon import CoinDaemon
  12. from mmgen.proto.xmr.rpc import MoneroRPCClient,MoneroWalletRPCClient
  13. from mmgen.proto.xmr.daemon import MoneroWalletDaemon
  14. from ..include.common import cfg,qmsg,vmsg
  15. async def cfg_file_auth_test(cfg, d, bad_auth=False):
  16. m = 'missing credentials' if bad_auth else f'credentials from {d.cfg_file}'
  17. qmsg(cyan(f'\n Testing authentication with {m}:'))
  18. time.sleep(0.1) # race condition
  19. d.remove_datadir() # removes cookie file to force authentication from cfg file
  20. os.makedirs(d.network_datadir)
  21. if not bad_auth:
  22. cf = os.path.join(d.datadir,d.cfg_file)
  23. with open(cf,'a') as fp:
  24. fp.write('\nrpcuser = ut_rpc\nrpcpassword = ut_rpc_passw0rd\n')
  25. d.flag.keep_cfg_file = True
  26. d.start()
  27. if bad_auth:
  28. os.rename(d.auth_cookie_fn,d.auth_cookie_fn+'.bak')
  29. try:
  30. await rpc_init(cfg, d.proto)
  31. except Exception as e:
  32. vmsg(yellow(str(e)))
  33. else:
  34. die(3,'No error on missing credentials!')
  35. os.rename(d.auth_cookie_fn+'.bak',d.auth_cookie_fn)
  36. else:
  37. rpc = await rpc_init(cfg, d.proto)
  38. assert rpc.auth.user == 'ut_rpc', f'{rpc.auth.user}: user is not ut_rpc!'
  39. d.stop()
  40. async def print_daemon_info(rpc):
  41. if rpc.proto.base_proto == 'Monero':
  42. msg(f"""
  43. DAEMON VERSION: {rpc.daemon_version} [{rpc.daemon_version_str}]
  44. NETWORK: {rpc.proto.coin} {rpc.proto.network.upper()}
  45. """.rstrip())
  46. else:
  47. msg(f"""
  48. DAEMON VERSION: {rpc.daemon_version} [{rpc.daemon_version_str}]
  49. CAPS: {rpc.caps}
  50. NETWORK: {rpc.proto.coin} {rpc.proto.network.upper()}
  51. CHAIN: {rpc.chain}
  52. BLOCKCOUNT: {rpc.blockcount}
  53. CUR_DATE: {rpc.cur_date} [{make_timestr(rpc.cur_date)}]
  54. """.rstrip())
  55. if rpc.proto.base_proto == 'Bitcoin':
  56. def fmt_dict(d):
  57. return '\n ' + '\n '.join( pp_fmt(d).split('\n') ) + '\n'
  58. msg(f"""
  59. NETWORKINFO: {fmt_dict(rpc.cached["networkinfo"])}
  60. BLOCKCHAININFO: {fmt_dict(rpc.cached["blockchaininfo"])}
  61. DEPLOYMENTINFO: {fmt_dict(rpc.cached["deploymentinfo"])}
  62. WALLETINFO: {fmt_dict(await rpc.walletinfo)}
  63. """.rstrip())
  64. msg('')
  65. def do_msg(rpc,backend):
  66. bname = type(rpc.backend).__name__
  67. qmsg(' Testing backend {!r}{}'.format( bname, '' if backend == bname else f' [{backend}]' ))
  68. class init_test:
  69. @staticmethod
  70. async def btc(cfg, daemon, backend):
  71. rpc = await rpc_init(cfg, daemon.proto, backend, daemon)
  72. do_msg(rpc,backend)
  73. if cfg.tw_name:
  74. wi = await rpc.walletinfo
  75. assert wi['walletname'] == rpc.cfg.tw_name, f'{wi["walletname"]!r} != {rpc.cfg.tw_name!r}'
  76. bh = (await rpc.call('getblockchaininfo',timeout=300))['bestblockhash']
  77. await rpc.gathered_call('getblock',((bh,),(bh,1)),timeout=300)
  78. await rpc.gathered_call(None,(('getblock',(bh,)),('getblock',(bh,1))),timeout=300)
  79. return rpc
  80. @staticmethod
  81. async def bch(cfg, daemon, backend):
  82. rpc = await rpc_init(cfg, daemon.proto, backend, daemon)
  83. do_msg(rpc,backend)
  84. return rpc
  85. ltc = bch
  86. @staticmethod
  87. async def eth(cfg, daemon, backend):
  88. rpc = await rpc_init(cfg, daemon.proto, backend, daemon)
  89. do_msg(rpc,backend)
  90. await rpc.call('eth_blockNumber',timeout=300)
  91. return rpc
  92. etc = eth
  93. async def run_test(network_ids, test_cf_auth=False, daemon_ids=None, cfg_in=None):
  94. async def do_test(d, cfg):
  95. if not cfg.no_daemon_stop:
  96. d.stop()
  97. if not cfg.no_daemon_autostart:
  98. d.remove_datadir()
  99. d.start()
  100. for n, backend in enumerate(cfg._autoset_opts['rpc_backend'].choices):
  101. test = getattr(init_test, d.proto.coin.lower())
  102. rpc = await test(cfg, d, backend)
  103. if not n and cfg.verbose:
  104. await print_daemon_info(rpc)
  105. if not cfg.no_daemon_stop:
  106. d.stop()
  107. if test_cf_auth and sys.platform != 'win32':
  108. await cfg_file_auth_test(cfg, d)
  109. await cfg_file_auth_test(cfg, d, bad_auth=True)
  110. qmsg('')
  111. cfg_arg = cfg_in or cfg
  112. for network_id in network_ids:
  113. proto = init_proto(cfg_arg, network_id=network_id)
  114. all_ids = CoinDaemon.get_daemon_ids(cfg_arg, proto.coin)
  115. ids = set(daemon_ids) & set(all_ids) if daemon_ids else all_ids
  116. for daemon_id in ids:
  117. await do_test(CoinDaemon(cfg_arg, proto=proto,test_suite=True,daemon_id=daemon_id), cfg_arg)
  118. return True
  119. class unit_tests:
  120. altcoin_deps = ('ltc','bch','geth','erigon','parity','xmrwallet')
  121. arm_skip = ('parity',) # no prebuilt binaries for ARM
  122. async def btc(self, name, ut):
  123. return await run_test(
  124. ['btc','btc_tn'],
  125. test_cf_auth = True,
  126. cfg_in = Config({'_clone': cfg, 'tw_name': 'alternate-tracking-wallet'}))
  127. async def ltc(self, name, ut):
  128. return await run_test(['ltc','ltc_tn'], test_cf_auth=True)
  129. async def bch(self, name, ut):
  130. return await run_test(['bch','bch_tn'], test_cf_auth=True)
  131. async def geth(self, name, ut):
  132. return await run_test(['eth_tn','eth_rt'], daemon_ids=['geth']) # mainnet returns EIP-155 error on empty blockchain
  133. async def erigon(self, name, ut):
  134. return await run_test(['eth','eth_tn','eth_rt'], daemon_ids=['erigon'])
  135. async def parity(self, name, ut):
  136. return await run_test(['etc'])
  137. async def xmrwallet(self, name, ut):
  138. async def test_monerod_rpc(md):
  139. rpc = MoneroRPCClient(
  140. cfg = cfg,
  141. proto = md.proto,
  142. host = 'localhost',
  143. port = md.rpc_port,
  144. user = None,
  145. passwd = None,
  146. daemon = md,
  147. )
  148. if cfg.verbose:
  149. await print_daemon_info(rpc)
  150. rpc.call_raw('get_height')
  151. rpc.call('get_last_block_header')
  152. from mmgen.xmrseed import xmrseed
  153. async def run():
  154. networks = init_proto( cfg, 'xmr' ).networks
  155. daemons = [(
  156. CoinDaemon( cfg, proto=proto, test_suite=True ),
  157. MoneroWalletDaemon(
  158. cfg = cfg,
  159. proto = proto,
  160. test_suite = True,
  161. wallet_dir = os.path.join('test','trash2'),
  162. datadir = os.path.join('test','trash2','wallet_rpc'),
  163. passwd = 'ut_rpc_passw0rd' )
  164. ) for proto in (init_proto( cfg, 'xmr', network=network ) for network in networks) ]
  165. for md,wd in daemons:
  166. if not cfg.no_daemon_autostart:
  167. md.start()
  168. wd.start()
  169. await test_monerod_rpc(md)
  170. c = MoneroWalletRPCClient( cfg=cfg, daemon=wd )
  171. fn = f'monero-{wd.network}-junk-wallet'
  172. qmsg(f'Creating {wd.network} wallet')
  173. c.call(
  174. 'restore_deterministic_wallet',
  175. filename = fn,
  176. password = 'foo',
  177. seed = xmrseed().fromhex('beadface'*8,tostr=True) )
  178. if sys.platform == 'win32':
  179. wd.stop()
  180. wd.start()
  181. qmsg(f'Opening {wd.network} wallet')
  182. c.call( 'open_wallet', filename=fn, password='foo' )
  183. await c.stop_daemon()
  184. if not cfg.no_daemon_stop:
  185. await md.rpc.stop_daemon()
  186. gmsg('OK')
  187. import shutil
  188. shutil.rmtree('test/trash2',ignore_errors=True)
  189. os.makedirs('test/trash2/wallet_rpc')
  190. await run()
  191. return True