ut_rpc.py 6.3 KB

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