ut_rpc.py 6.3 KB

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