ct_automount.py 9.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263
  1. #!/usr/bin/env python3
  2. #
  3. # MMGen Wallet, a terminal-based cryptocurrency wallet
  4. # Copyright (C)2013-2025 The MMGen Project <mmgen@tuta.io>
  5. # Licensed under the GNU General Public License, Version 3:
  6. # https://www.gnu.org/licenses
  7. # Public project repositories:
  8. # https://github.com/mmgen/mmgen-wallet
  9. # https://gitlab.com/mmgen/mmgen-wallet
  10. """
  11. test.cmdtest_d.ct_automount: autosigning with automount tests for the cmdtest.py test suite
  12. """
  13. import time
  14. from .ct_autosign import CmdTestAutosignThreaded
  15. from .ct_regtest import CmdTestRegtest, rt_pw
  16. from ..include.common import cfg, gr_uc
  17. class CmdTestAutosignAutomount(CmdTestAutosignThreaded, CmdTestRegtest):
  18. 'automounted transacting operations via regtest mode'
  19. networks = ('btc', 'bch', 'ltc')
  20. tmpdir_nums = [49]
  21. bdb_wallet = True
  22. rt_data = {
  23. 'rtFundAmt': {'btc':'500', 'bch':'500', 'ltc':'5500'},
  24. }
  25. cmd_group = (
  26. ('setup', 'regtest mode setup'),
  27. ('walletgen_alice', 'wallet generation (Alice)'),
  28. ('addrgen_alice', 'address generation (Alice)'),
  29. ('addrimport_alice', 'importing Alice’s addresses'),
  30. ('fund_alice', 'funding Alice’s wallet'),
  31. ('generate', 'mining a block'),
  32. ('alice_bal1', 'checking Alice’s balance'),
  33. ('alice_txcreate1', 'creating a transaction'),
  34. ('alice_txcreate_bad_have_unsigned', 'creating the transaction again (error)'),
  35. ('alice_run_autosign_setup', 'running ‘autosign setup’ (with default wallet)'),
  36. ('wait_loop_start', 'starting autosign wait loop'),
  37. ('alice_txstatus1', 'getting transaction status (unsigned)'),
  38. ('alice_txstatus2', 'getting transaction status (unsent)'),
  39. ('alice_txcreate_bad_have_unsent', 'creating the transaction again (error)'),
  40. ('alice_txsend1', 'sending a transaction, editing comment'),
  41. ('alice_txstatus3', 'getting transaction status (in mempool)'),
  42. ('alice_txsend_bad_no_unsent', 'sending the transaction again (error)'),
  43. ('generate', 'mining a block'),
  44. ('alice_txstatus4', 'getting transaction status (one confirmation)'),
  45. ('alice_txcreate2', 'creating a transaction'),
  46. ('alice_txsend_abort1', 'aborting the transaction (raw only)'),
  47. ('alice_txsend_abort2', 'aborting the transaction again (error)'),
  48. ('alice_txcreate3', 'creating a transaction'),
  49. ('alice_txsend_abort3', 'aborting the transaction (user exit)'),
  50. ('alice_txsend_abort4', 'aborting the transaction (raw + signed)'),
  51. ('alice_txsend_abort5', 'aborting the transaction again (error)'),
  52. ('generate', 'mining a block'),
  53. ('alice_txcreate4', 'creating a transaction'),
  54. ('alice_txbump1', 'bumping the unsigned transaction (error)'),
  55. ('alice_txbump2', 'bumping the unsent transaction (error)'),
  56. ('alice_txsend2', 'sending the transaction'),
  57. ('alice_txbump3', 'bumping the transaction'),
  58. ('alice_txsend3', 'sending the bumped transaction'),
  59. ('alice_txbump4', 'bumping the transaction (new outputs, fee too low)'),
  60. ('alice_txbump_abort1', 'aborting the transaction'),
  61. ('alice_txbump5', 'bumping the transaction (new outputs)'),
  62. ('alice_txsend5', 'sending the bumped transaction'),
  63. ('alice_txstatus5', 'getting transaction status (in mempool)'),
  64. ('generate', 'mining a block'),
  65. ('alice_bal2', 'checking Alice’s balance'),
  66. ('wait_loop_kill', 'stopping autosign wait loop'),
  67. ('stop', 'stopping regtest daemon'),
  68. ('txview', 'viewing transactions'),
  69. )
  70. def __init__(self, trunner, cfgs, spawn):
  71. self.coins = [cfg.coin.lower()]
  72. CmdTestAutosignThreaded.__init__(self, trunner, cfgs, spawn)
  73. CmdTestRegtest.__init__(self, trunner, cfgs, spawn)
  74. if trunner is None:
  75. return
  76. self.opts.append('--alice')
  77. def alice_txcreate1(self):
  78. return self._user_txcreate(
  79. 'alice',
  80. chg_addr = 'C:5',
  81. data_arg = 'data:'+gr_uc[:24])
  82. def alice_txcreate2(self):
  83. return self._user_txcreate('alice', chg_addr='L:5')
  84. alice_txcreate3 = alice_txcreate2
  85. def alice_txcreate4(self):
  86. return self._user_txcreate('alice', chg_addr='L:4', need_rbf=True)
  87. def _alice_txsend_abort(self, err=False, send_resp='y', expect=None, shred_expect=[]):
  88. self.insert_device_online()
  89. t = self.spawn(
  90. 'mmgen-txsend',
  91. ['--quiet', '--abort'],
  92. exit_val = 2 if err else 1 if send_resp == 'n' else None)
  93. if err:
  94. t.expect(expect)
  95. else:
  96. t.expect('(y/N): ', send_resp)
  97. if expect:
  98. t.expect(expect)
  99. for pat in shred_expect:
  100. t.expect(pat, regex=True)
  101. t.read()
  102. self.remove_device_online()
  103. return t
  104. def alice_txsend_abort1(self):
  105. return self._alice_txsend_abort(shred_expect=['Shredding .*arawtx'])
  106. def alice_txsend_abort2(self):
  107. return self._alice_txsend_abort(err=True, expect='No unsent transactions')
  108. def alice_txsend_abort3(self):
  109. return self._alice_txsend_abort(send_resp='n', expect='Exiting at user request')
  110. def alice_txsend_abort4(self):
  111. self._wait_signed('transaction')
  112. return self._alice_txsend_abort(shred_expect=[r'Shredding .*arawtx', r'Shredding .*asigtx'])
  113. alice_txsend_abort5 = alice_txsend_abort2
  114. def alice_txcreate_bad_have_unsigned(self):
  115. return self._user_txcreate('alice', chg_addr='C:5', exit_val=2, expect_str='already present')
  116. def alice_txcreate_bad_have_unsent(self):
  117. return self._user_txcreate('alice', chg_addr='C:5', exit_val=2, expect_str='unsent transaction')
  118. def alice_run_autosign_setup(self):
  119. return self.run_setup(mn_type='default', use_dfl_wallet=True, passwd=rt_pw)
  120. def alice_txsend1(self):
  121. return self._user_txsend('alice', 'This one’s worth a comment', no_wait=True)
  122. def alice_txsend2(self):
  123. return self._user_txsend('alice', need_rbf=True)
  124. def alice_txsend3(self):
  125. return self._user_txsend('alice', need_rbf=True)
  126. def alice_txsend5(self):
  127. return self._user_txsend('alice', need_rbf=True)
  128. def _alice_txstatus(self, expect, exit_val=None, need_rbf=False):
  129. if need_rbf and not self.proto.cap('rbf'):
  130. return 'skip'
  131. self.insert_device_online()
  132. t = self.spawn(
  133. 'mmgen-txsend',
  134. ['--alice', '--autosign', '--status', '--verbose'],
  135. exit_val = exit_val)
  136. t.expect(expect)
  137. if not exit_val:
  138. t.expect('view: ', 'n')
  139. t.read()
  140. self.remove_device_online()
  141. return t
  142. def alice_txstatus1(self):
  143. return self._alice_txstatus('unsigned', 1)
  144. def alice_txstatus2(self):
  145. self._wait_signed('transaction')
  146. return self._alice_txstatus('unsent', 1)
  147. def alice_txstatus3(self):
  148. return self._alice_txstatus('in mempool', 0)
  149. def alice_txstatus4(self):
  150. return self._alice_txstatus('1 confirmation', 0)
  151. def alice_txstatus5(self):
  152. return self._alice_txstatus('in mempool', need_rbf=True)
  153. def alice_txsend_bad_no_unsent(self):
  154. self.insert_device_online()
  155. t = self.spawn('mmgen-txsend', ['--quiet', '--autosign'], exit_val=2)
  156. t.expect('No unsent transactions')
  157. t.read()
  158. self.remove_device_online()
  159. return t
  160. def _alice_txbump(self, fee_opt=None, output_args=[], bad_tx_expect=None, low_fee_fix=None):
  161. if not self.proto.cap('rbf'):
  162. return 'skip'
  163. self.insert_device_online()
  164. t = self.spawn(
  165. 'mmgen-txbump',
  166. ['--alice', '--autosign']
  167. + ([fee_opt] if fee_opt else [])
  168. + output_args,
  169. exit_val = 1 if bad_tx_expect else None)
  170. if bad_tx_expect:
  171. time.sleep(0.5)
  172. t.expect('Only sent transactions')
  173. t.expect(bad_tx_expect)
  174. else:
  175. if not output_args:
  176. t.expect(r'to deduct the fee from .* change output\): ', '\n', regex=True)
  177. t.expect(r'(Y/n): ', 'y') # output OK?
  178. if low_fee_fix or not fee_opt:
  179. if low_fee_fix:
  180. t.expect('Please choose a higher fee')
  181. t.expect('transaction fee: ', (low_fee_fix or '200s') + '\n')
  182. if output_args:
  183. t.expect(r'(Y/n): ', 'y')
  184. t.expect(r'(Y/n): ', 'y') # fee OK?
  185. t.expect(r'(y/N): ', '\n') # add comment?
  186. t.expect(r'(y/N): ', 'y') # save?
  187. t.read()
  188. self.remove_device_online()
  189. return t
  190. def alice_txbump1(self):
  191. return self._alice_txbump(bad_tx_expect='unsigned transaction')
  192. def alice_txbump2(self):
  193. self._wait_signed('transaction')
  194. return self._alice_txbump(bad_tx_expect='unsent transaction')
  195. def alice_txbump3(self):
  196. return self._alice_txbump()
  197. def alice_txbump4(self):
  198. sid = self._user_sid('alice')
  199. return self._alice_txbump(
  200. fee_opt = '--fee=3s',
  201. output_args = [f'{self.burn_addr},7.654321', f'{sid}:C:1'],
  202. low_fee_fix = '300s')
  203. def alice_txbump_abort1(self):
  204. if not self.proto.cap('rbf'):
  205. return 'skip'
  206. return self._alice_txsend_abort(shred_expect=['Shredding .*arawtx'])
  207. def alice_txbump5(self):
  208. sid = self._user_sid('alice')
  209. return self._alice_txbump(
  210. fee_opt = '--fee=400s',
  211. output_args = ['data:message for posterity', f'{self.burn_addr},7.654321', f'{sid}:C:1'])
  212. def alice_bal2(self):
  213. bals = {
  214. 'btc': '491.11002204',
  215. 'ltc': '5491.11002204',
  216. 'bch': '498.7653392',
  217. }
  218. return self.user_bal('alice', bals.get(self.coin, None))