ts_shared.py 7.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237
  1. #!/usr/bin/env python3
  2. #
  3. # mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
  4. # Copyright (C)2013-2019 The MMGen Project <mmgen@tuta.io>
  5. #
  6. # This program is free software: you can redistribute it and/or modify
  7. # it under the terms of the GNU General Public License as published by
  8. # the Free Software Foundation, either version 3 of the License, or
  9. # (at your option) any later version.
  10. #
  11. # This program is distributed in the hope that it will be useful,
  12. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14. # GNU General Public License for more details.
  15. #
  16. # You should have received a copy of the GNU General Public License
  17. # along with this program. If not, see <http://www.gnu.org/licenses/>.
  18. """
  19. ts_shared.py: Shared methods for the test.py test suite
  20. """
  21. import os
  22. from mmgen.globalvars import g
  23. from mmgen.opts import opt
  24. from mmgen.util import ymsg
  25. from test.test_py_d.common import *
  26. from test.common import *
  27. class TestSuiteShared(object):
  28. 'shared methods for the test.py test suite'
  29. def txcreate_ui_common( self,t,
  30. caller = None,
  31. menu = [],
  32. inputs = '1',
  33. file_desc = 'Transaction',
  34. input_sels_prompt = 'to spend',
  35. bad_input_sels = False,
  36. non_mmgen_inputs = 0,
  37. interactive_fee = '',
  38. fee_desc = 'transaction fee',
  39. fee_res = None,
  40. eth_fee_res = None,
  41. add_comment = '',
  42. view = 't',
  43. save = True ):
  44. txdo = (caller or self.test_name)[:4] == 'txdo'
  45. for choice in menu + ['q']:
  46. t.expect(r'\[q\]uit view, .*?:.',choice,regex=True)
  47. if bad_input_sels:
  48. for r in ('x','3-1','9999'):
  49. t.expect(input_sels_prompt+': ',r+'\n')
  50. t.expect(input_sels_prompt+': ',inputs+'\n')
  51. if not txdo:
  52. for i in range(non_mmgen_inputs):
  53. t.expect('Accept? (y/N): ','y')
  54. have_est_fee = t.expect([fee_desc+': ','OK? (Y/n): ']) == 1
  55. if have_est_fee and not interactive_fee:
  56. t.send('y')
  57. else:
  58. if have_est_fee: t.send('n')
  59. if eth_fee_res:
  60. t.expect('or gas price: ',interactive_fee+'\n')
  61. else:
  62. t.send(interactive_fee+'\n')
  63. if fee_res: t.expect(fee_res)
  64. t.expect('OK? (Y/n): ','y')
  65. t.expect('(Y/n): ','\n') # chg amt OK?
  66. t.do_comment(add_comment)
  67. t.view_tx(view)
  68. if not txdo:
  69. t.expect('(y/N): ',('n','y')[save])
  70. t.written_to_file(file_desc)
  71. return t
  72. def txsign_ui_common( self,t,
  73. caller = None,
  74. view = 't',
  75. add_comment = '',
  76. file_desc = 'Signed transaction',
  77. ni = False,
  78. save = True,
  79. do_passwd = False,
  80. has_label = False ):
  81. txdo = (caller or self.test_name)[:4] == 'txdo'
  82. if do_passwd:
  83. t.passphrase('MMGen wallet',self.wpasswd)
  84. if not ni and not txdo:
  85. t.view_tx(view)
  86. t.do_comment(add_comment,has_label=has_label)
  87. t.expect('(Y/n): ',('n','y')[save])
  88. t.written_to_file(file_desc)
  89. return t
  90. def txsend_ui_common( self,t,
  91. caller = None,
  92. view = 'n',
  93. add_comment = '',
  94. file_desc = 'Sent transaction',
  95. confirm_send = True,
  96. bogus_send = True,
  97. quiet = False,
  98. has_label = False ):
  99. txdo = (caller or self.test_name)[:4] == 'txdo'
  100. if not txdo:
  101. t.license() # MMGEN_NO_LICENSE is set, so does nothing
  102. t.view_tx(view)
  103. t.do_comment(add_comment,has_label=has_label)
  104. self._do_confirm_send(t,quiet=quiet,confirm_send=confirm_send)
  105. if bogus_send:
  106. txid = ''
  107. t.expect('BOGUS transaction NOT sent')
  108. else:
  109. txid = t.expect_getend('Transaction sent: ')
  110. assert len(txid) == 64,"'{}': Incorrect txid length!".format(txid)
  111. t.written_to_file(file_desc)
  112. return txid
  113. def txsign_end(self,t,tnum=None,has_label=False):
  114. t.expect('Signing transaction')
  115. t.do_comment(False,has_label=has_label)
  116. t.expect('Save signed transaction.*?\? \(Y/n\): ','y',regex=True)
  117. t.written_to_file('Signed transaction' + (' #' + tnum if tnum else ''), oo=True)
  118. return t
  119. def txsign( self, txfile, wf,
  120. pf = '',
  121. bumpf = '',
  122. save = True,
  123. has_label = False,
  124. do_passwd = True,
  125. extra_opts = [],
  126. extra_desc = '' ):
  127. opts = extra_opts + ['-d',self.tmpdir,txfile] + ([wf] if wf else [])
  128. t = self.spawn('mmgen-txsign', opts, extra_desc)
  129. t.license()
  130. t.view_tx('n')
  131. if do_passwd: t.passphrase('MMGen wallet',self.wpasswd)
  132. if save:
  133. self.txsign_end(t,has_label=has_label)
  134. else:
  135. t.do_comment(False,has_label=has_label)
  136. t.expect('Save signed transaction? (Y/n): ','n')
  137. t.req_exit_val = 1
  138. return t
  139. def ref_brain_chk(self,bw_file=ref_bw_file):
  140. wf = joinpath(ref_dir,bw_file)
  141. add_args = ['-l{}'.format(self.seed_len), '-p'+ref_bw_hash_preset]
  142. return self.walletchk(wf,pf=None,add_args=add_args,
  143. desc='brainwallet',sid=self.ref_bw_seed_id)
  144. def walletchk(self,wf,pf,desc='MMGen wallet',add_args=[],sid=None,pw=False,extra_desc=''):
  145. args = []
  146. hp = self.hash_preset if hasattr(self,'hash_preset') else '1'
  147. wf_arg = [wf] if wf else []
  148. t = self.spawn('mmgen-walletchk',
  149. add_args+args+['-p',hp]+wf_arg,
  150. extra_desc=extra_desc)
  151. if desc != 'hidden incognito data':
  152. t.expect("Getting {} from file '".format(desc))
  153. if pw:
  154. t.passphrase(desc,self.wpasswd)
  155. t.expect(['Passphrase is OK', 'Passphrase.* are correct'],regex=True)
  156. chk = t.expect_getend('Valid {} for Seed ID '.format(desc))[:8]
  157. if sid: cmp_or_die(chk,sid)
  158. return t
  159. def addrgen(self,wf,pf=None,check_ref=False,ftype='addr',id_str=None,extra_args=[],mmtype=None):
  160. if not mmtype and ftype[:4] != 'pass':
  161. mmtype = self.segwit_mmtype
  162. cmd_pfx = (ftype,'pass')[ftype[:4]=='pass']
  163. t = self.spawn('mmgen-{}gen'.format(cmd_pfx),
  164. ['-d',self.tmpdir] + extra_args +
  165. ([],['--type='+str(mmtype)])[bool(mmtype)] +
  166. ([],[wf])[bool(wf)] +
  167. ([],[id_str])[bool(id_str)] +
  168. [getattr(self,'{}_idx_list'.format(cmd_pfx))],
  169. extra_desc='({})'.format(mmtype) if mmtype in ('segwit','bech32') else '')
  170. t.license()
  171. t.passphrase('MMGen wallet',self.wpasswd)
  172. t.expect('Passphrase is OK')
  173. desc = ('address','password')[ftype[:4]=='pass']
  174. chk = t.expect_getend(r'Checksum for {} data .*?: '.format(desc),regex=True)
  175. if ftype[:4] == 'pass':
  176. t.expect('Encrypt password list? (y/N): ','\n')
  177. t.written_to_file('Password list',oo=True)
  178. else:
  179. t.written_to_file('Addresses',oo=True)
  180. if check_ref:
  181. chk_ref = (self.chk_data[self.test_name] if ftype[:4] == 'pass' else
  182. self.chk_data[self.test_name][self.fork][g.testnet])
  183. cmp_or_die(chk,chk_ref,desc='{}list data checksum'.format(ftype))
  184. return t
  185. def keyaddrgen(self,wf,pf=None,check_ref=False,mmtype=None):
  186. if not mmtype:
  187. mmtype = self.segwit_mmtype
  188. args = ['-d',self.tmpdir,self.usr_rand_arg,wf,self.addr_idx_list]
  189. t = self.spawn('mmgen-keygen',
  190. ([],['--type='+str(mmtype)])[bool(mmtype)] + args,
  191. extra_desc='({})'.format(mmtype) if mmtype in ('segwit','bech32') else '')
  192. t.license()
  193. t.passphrase('MMGen wallet',self.wpasswd)
  194. chk = t.expect_getend(r'Checksum for key-address data .*?: ',regex=True)
  195. if check_ref:
  196. chk_ref = self.chk_data[self.test_name][self.fork][g.testnet]
  197. cmp_or_die(chk,chk_ref,desc='key-address list data checksum')
  198. t.expect('Encrypt key list? (y/N): ','y')
  199. t.usr_rand(self.usr_rand_chars)
  200. t.hash_preset('new key list','1')
  201. t.passphrase_new('new key list',self.kapasswd)
  202. t.written_to_file('Encrypted secret keys',oo=True)
  203. return t
  204. def _do_confirm_send(self,t,quiet=False,confirm_send=True):
  205. t.expect('Are you sure you want to broadcast this')
  206. m = ('YES, I REALLY WANT TO DO THIS','YES')[quiet]
  207. t.expect("'{}' to confirm: ".format(m),('',m)[confirm_send]+'\n')