mmgen-wallet/test/cmdtest_d/ct_opts.py

#!/usr/bin/env python3
#
# MMGen Wallet, a terminal-based cryptocurrency wallet
# Copyright (C)2013-2024 The MMGen Project <mmgen@tuta.io>
#
# Project source code repository: https://github.com/mmgen/mmgen-wallet
# Licensed according to the terms of GPL Version 3.  See LICENSE for details.

"""
test.cmdtest_d.ct_opts: options processing tests for the MMGen cmdtest.py test suite
"""

import os, time

from ..include.common import cfg
from .ct_base import CmdTestBase

class CmdTestOpts(CmdTestBase):
	'command-line options parsing and processing'
	networks = ('btc',)
	tmpdir_nums = [41]
	cmd_group = (
		('opt_helpscreen',       (41, 'helpscreen output', [])),
		('opt_noargs',           (41, 'invocation with no user options or arguments', [])),
		('opt_good1',            (41, 'good opts (long opts only)', [])),
		('opt_good2',            (41, 'good opts (mixed short and long opts)', [])),
		('opt_good3',            (41, 'good opts (max arg count)', [])),
		('opt_good4',            (41, 'good opts (maxlen arg)', [])),
		('opt_good5',            (41, 'good opts (long opt substring)', [])),
		('opt_good6',            (41, 'good global opt (--coin=xmr)', [])),
		('opt_good7',            (41, 'good global opt (--coin xmr)', [])),
		('opt_good8',            (41, 'good global opt (--pager)', [])),
		('opt_good9',            (41, 'good cmdline arg ‘-’', [])),
		('opt_good10',           (41, 'good cmdline arg ‘-’ with arg', [])),
		('opt_good11',           (41, 'good cmdline arg ‘-’ with option', [])),
		('opt_good12',           (41, 'good cmdline opt (short opt with option)', [])),
		('opt_good13',           (41, 'good cmdline opt (short opt with option)', [])),
		('opt_good14',           (41, 'good cmdline opt (combined short opt with option)', [])),
		('opt_good15',           (41, 'good cmdline opt (combined short opt with option)', [])),
		('opt_good16',           (41, 'good cmdline opt (param with equals signs)', [])),
		('opt_good17',           (41, 'good cmdline opt (param with equals signs)', [])),
		('opt_good18',           (41, 'good cmdline opt (param with equals signs)', [])),
		('opt_good19',           (41, 'good cmdline opt (param with equals signs)', [])),
		('opt_good20',           (41, 'good cmdline opt (opt + negated opt)', [])),
		('opt_good21',           (41, 'good cmdline opt (negated negative opt)', [])),
		('opt_good22',           (41, 'good cmdline opt (opt + negated opt [substring])', [])),
		('opt_good23',           (41, 'good cmdline opt (negated negative opt [substring])', [])),
		('opt_good24',           (41, 'good cmdline opt (negated opt + opt [substring])', [])),
		('opt_bad_param',        (41, 'bad global opt (--pager=1)', [])),
		('opt_bad_infile',       (41, 'bad infile parameter', [])),
		('opt_bad_outdir',       (41, 'bad outdir parameter', [])),
		('opt_bad_incompatible', (41, 'incompatible opts', [])),
		('opt_bad_autoset',      (41, 'invalid autoset value', [])),
		('opt_invalid_1',        (41, 'invalid cmdline opt ‘--x’', [])),
		('opt_invalid_2',        (41, 'invalid cmdline opt ‘---’', [])),
		('opt_invalid_5',        (41, 'invalid cmdline opt (missing parameter)', [])),
		('opt_invalid_6',        (41, 'invalid cmdline opt (missing parameter)', [])),
		('opt_invalid_7',        (41, 'invalid cmdline opt (parameter not required)', [])),
		('opt_invalid_8',        (41, 'invalid cmdline opt (non-existent option)', [])),
		('opt_invalid_9',        (41, 'invalid cmdline opt (non-existent option)', [])),
		('opt_invalid_10',       (41, 'invalid cmdline opt (missing parameter)', [])),
		('opt_invalid_11',       (41, 'invalid cmdline opt (missing parameter)', [])),
		('opt_invalid_12',       (41, 'invalid cmdline opt (non-existent option)', [])),
		('opt_invalid_13',       (41, 'invalid cmdline opt (ambiguous long opt substring)', [])),
		('opt_invalid_14',       (41, 'invalid cmdline opt (long opt substring too short)', [])),
		('opt_invalid_15',       (41, 'invalid cmdline (too many args)', [])),
		('opt_invalid_16',       (41, 'invalid cmdline (overlong arg)', [])),
	)

	def spawn_prog(self, args, exit_val=None):
		return self.spawn('test/misc/opts.py', args, cmd_dir='.', exit_val=exit_val)

	def check_vals(self, args, vals):
		t = self.spawn_prog(args)
		for k, v in vals:
			t.expect(rf'{k}:\s+{v}', regex=True)
		return t

	def do_run(self, args, expect, exit_val, regex=False):
		t = self.spawn_prog(args, exit_val=exit_val or None)
		t.expect(expect, regex=regex)
		return t

	def opt_helpscreen(self):
		expect = r'OPTS.PY: Opts test.*USAGE:\s+opts.py'
		if not cfg.pexpect_spawn:
			expect += r'.*--minconf.*NOTES FOR THIS.*a note'
		t = self.do_run(['--help'], expect, 0, regex=True)
		if t.pexpect_spawn:
			time.sleep(0.4)
			t.send('q')
		return t

	def opt_noargs(self):
		return self.check_vals(
			[],
			(
				('cfg.foo',                 'None'),         # added opt
				('cfg.print_checksum',      'None'),         # sets 'quiet'
				('cfg.quiet',               'False'),        # _incompatible_opts
				('cfg.verbose',             'False'),        # _incompatible_opts
				('cfg.passwd_file',         ''),             # _infile_opts - check_infile()
				('cfg.outdir',              ''),             # check_outdir()
				('cfg.cached_balances',     'False'),
				('cfg.minconf',             '1'),
				('cfg.coin',                'BTC'),
				('cfg.pager',               'False'),
				('cfg.fee_estimate_mode',   'conservative'), # _autoset_opts
			))

	def opt_good1(self):
		pf_base = 'testfile'
		pf = os.path.join(self.tmpdir, pf_base)
		self.write_to_tmpfile(pf_base, '')
		return self.check_vals(
			[
				'--print-checksum',
				'--fee-estimate-mode=E',
				'--passwd-file='+pf,
				'--outdir='+self.tmpdir,
				'--cached-balances',
				f'--hidden-incog-input-params={pf},123',
			], (
				('cfg.print_checksum',           'True'),
				('cfg.quiet',                    'True'), # set by print_checksum
				('cfg.passwd_file',              pf),
				('cfg.outdir',                   self.tmpdir),
				('cfg.cached_balances',          'True'),
				('cfg.hidden_incog_input_params', pf+',123'),
				('cfg.fee_estimate_mode',         'economical'),
			))

	def opt_good2(self):
		return self.check_vals(
			[
				'--print-checksum',
				'-qX',
				f'--outdir={self.tmpdir}',
				'-p5',
				'-m', '0',
				'--seed-len=256',
				'-L--my-label',
				'--seed-len', '128',
				'--min-temp=-30',
				'-T-10',
				'--',
				'x', 'y', '12345'
			], (
				('cfg.print_checksum',  'True'),
				('cfg.quiet',           'True'),
				('cfg.outdir',          self.tmpdir),
				('cfg.cached_balances', 'True'),
				('cfg.minconf',         '0'),
				('cfg.keep_label',      'None'),
				('cfg.seed_len',        '128'),
				('cfg.hash_preset',     '5'),
				('cfg.label',           '--my-label'),
				('cfg.min_temp',        '-30'),
				('cfg.max_temp',        '-10'),
				('arg1',                'x'),
				('arg2',                'y'),
				('arg3',                '12345'),
			))

	def opt_good3(self):
		return self.check_vals(['m'] * 256, (('arg256', 'm'),))

	def opt_good4(self):
		return self.check_vals(['e' * 4096], (('arg1', 'e' * 4096),))

	def opt_good5(self):
		return self.check_vals(['--minc=7'], (('cfg.minconf', '7'),))

	def opt_good6(self):
		return self.check_vals(['--coin=xmr'], (('cfg.coin', 'XMR'),))

	def opt_good7(self):
		return self.check_vals(['--coin', 'xmr'], (('cfg.coin', 'XMR'),))

	def opt_good8(self):
		return self.check_vals(['--pager'], (('cfg.pager', 'True'),))

	def opt_good9(self):
		return self.check_vals(['-'], (('arg1', '-'),))

	def opt_good10(self):
		return self.check_vals(['-', '-x'], (('arg1', '-'), ('arg2', '-x')))

	def opt_good11(self):
		return self.check_vals(['-q', '-', '-x'], (('arg1', '-'), ('arg2', '-x')))

	def opt_good12(self):
		return self.check_vals(['-l128'], (('cfg.seed_len', '128'),))

	def opt_good13(self):
		return self.check_vals(['-l', '128'], (('cfg.seed_len', '128'),))

	def opt_good14(self):
		return self.check_vals(['-kl128'], (('cfg.keep_label', 'True'), ('cfg.seed_len', '128')))

	def opt_good15(self):
		return self.check_vals(['-kl', '128'], (('cfg.keep_label', 'True'), ('cfg.seed_len', '128')))

	def opt_good16(self):
		return self.check_vals(['--point=x=1,y=2,z=3'], (('cfg.point', 'x=1,y=2,z=3'),))

	def opt_good17(self):
		return self.check_vals(['--point', 'x=1,y=2,z=3'], (('cfg.point', 'x=1,y=2,z=3'),))

	def opt_good18(self):
		return self.check_vals(['-xx=1,y=2,z=3'], (('cfg.point', 'x=1,y=2,z=3'),))

	def opt_good19(self):
		return self.check_vals(['-x', 'x=1,y=2,z=3'], (('cfg.point', 'x=1,y=2,z=3'),))

	def opt_good20(self):
		return self.check_vals(['--pager', '--no-pager'], (('cfg.pager', 'False'),))

	def opt_good21(self):
		return self.check_vals(['--foobleize'], (('cfg.no_foobleize', 'False'),))

	def opt_good22(self):
		return self.check_vals(['--quiet', '--no-q'], (('cfg.quiet', 'False'),))

	def opt_good23(self):
		return self.check_vals(['--foobl'], (('cfg.no_foobleize', 'False'),))

	def opt_good24(self):
		return self.check_vals(['--no-pag', '--pag'], (('cfg.pager', 'True'),))

	def opt_bad_param(self):
		return self.do_run(['--pager=1'], 'no parameter', 1)

	def opt_bad_infile(self):
		pf = os.path.join(self.tmpdir, 'fubar')
		return self.do_run(['--passwd-file='+pf], 'not found', 1)

	def opt_bad_outdir(self):
		bo = self.tmpdir+'_fubar'
		return self.do_run(['--outdir='+bo], 'not found', 1)

	def opt_bad_incompatible(self):
		return self.do_run(['--label=Label', '--keep-label'], 'Conflicting options', 1)

	def opt_bad_autoset(self):
		return self.do_run(['--fee-estimate-mode=Fubar'], 'not unique substring', 1)

	def opt_invalid(self, args, expect, exit_val):
		t = self.spawn_prog(args, exit_val=exit_val)
		t.expect(expect)
		return t

	def opt_invalid_1(self):
		return self.opt_invalid(['--x'], 'must be at least', 1)

	def opt_invalid_2(self):
		return self.opt_invalid(['---'], 'must be at least', 1)

	def opt_invalid_5(self):
		return self.opt_invalid(['-l'], 'missing parameter', 1)

	def opt_invalid_6(self):
		return self.opt_invalid(['-l', '-k'], 'missing parameter', 1)

	def opt_invalid_7(self):
		return self.opt_invalid(['--quiet=1'], 'requires no parameter', 1)

	def opt_invalid_8(self):
		return self.opt_invalid(['-w'], 'unrecognized option', 1)

	def opt_invalid_9(self):
		return self.opt_invalid(['--frobnicate'], 'unrecognized option', 1)

	def opt_invalid_10(self):
		return self.opt_invalid(['--label', '-q'], 'missing parameter', 1)

	def opt_invalid_11(self):
		return self.opt_invalid(['-T', '-10'], 'missing parameter', 1)

	def opt_invalid_12(self):
		return self.opt_invalid(['-q', '-10'], 'unrecognized option', 1)

	def opt_invalid_13(self):
		return self.opt_invalid(['--mi=3'], 'ambiguous option', 1)

	def opt_invalid_14(self):
		return self.opt_invalid(['--m=3'], 'must be at least', 1)

	def opt_invalid_15(self):
		return self.opt_invalid(['m'] * 257, 'too many', 1)

	def opt_invalid_16(self):
		return self.opt_invalid(['e' * 4097], 'too long', 1)