ut_rpc.py 6.2 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','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. async def run():
  141. networks = init_proto( cfg, 'xmr' ).networks
  142. daemons = [(
  143. CoinDaemon( cfg, proto=proto, test_suite=True ),
  144. MoneroWalletDaemon(
  145. cfg = cfg,
  146. proto = proto,
  147. test_suite = True,
  148. wallet_dir = os.path.join('test','trash2'),
  149. datadir = os.path.join('test','trash2','wallet_rpc'),
  150. passwd = 'ut_rpc_passw0rd' )
  151. ) for proto in (init_proto( cfg, 'xmr', network=network ) for network in networks) ]
  152. for md,wd in daemons:
  153. if not cfg.no_daemon_autostart:
  154. md.start()
  155. wd.start()
  156. test_monerod_rpc(md)
  157. c = MoneroWalletRPCClient( cfg=cfg, daemon=wd )
  158. fn = f'monero-{wd.network}-junk-wallet'
  159. qmsg(f'Creating {wd.network} wallet')
  160. c.call(
  161. 'restore_deterministic_wallet',
  162. filename = fn,
  163. password = 'foo',
  164. seed = xmrseed().fromhex('beadface'*8,tostr=True) )
  165. if gc.platform == 'win':
  166. wd.stop()
  167. wd.start()
  168. qmsg(f'Opening {wd.network} wallet')
  169. c.call( 'open_wallet', filename=fn, password='foo' )
  170. await c.stop_daemon()
  171. if not cfg.no_daemon_stop:
  172. await md.rpc.stop_daemon()
  173. gmsg('OK')
  174. from mmgen.xmrseed import xmrseed
  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