Browse Source

minor test suite fixes

The MMGen Project 3 years ago
parent
commit
fe756f7303
4 changed files with 13 additions and 5 deletions
  1. 1 1
      mmgen/tool.py
  2. 4 2
      test/test-release.sh
  3. 7 1
      test/test_py_d/input.py
  4. 1 1
      test/tooltest2.py

+ 1 - 1
mmgen/tool.py

@@ -208,7 +208,7 @@ def _process_result(ret,pager=False,print_result=False):
 	if ret == True:
 	if ret == True:
 		return True
 		return True
 	elif ret in (False,None):
 	elif ret in (False,None):
-		ydie(1,"tool command returned '{}'".format(ret))
+		ydie(1,f'tool command returned {ret!r}')
 	elif isinstance(ret,str):
 	elif isinstance(ret,str):
 		return triage_result(ret)
 		return triage_result(ret)
 	elif isinstance(ret,int):
 	elif isinstance(ret,int):

+ 4 - 2
test/test-release.sh

@@ -10,8 +10,10 @@ elif uname -a | grep -q 'MSYS'; then
 	SUDO='' MSYS2=1;
 	SUDO='' MSYS2=1;
 fi
 fi
 
 
-RED="\e[31;1m" GREEN="\e[32;1m" YELLOW="\e[33;1m" BLUE="\e[34;1m" MAGENTA="\e[35;1m" CYAN="\e[36;1m"
-RESET="\e[0m"
+[ "$MMGEN_DISABLE_COLOR" ] || {
+	RED="\e[31;1m" GREEN="\e[32;1m" YELLOW="\e[33;1m" BLUE="\e[34;1m" MAGENTA="\e[35;1m" CYAN="\e[36;1m"
+	RESET="\e[0m"
+}
 
 
 trap 'echo -e "${GREEN}Exiting at user request$RESET"; exit' INT
 trap 'echo -e "${GREEN}Exiting at user request$RESET"; exit' INT
 
 

+ 7 - 1
test/test_py_d/input.py

@@ -49,7 +49,13 @@ def stealth_mnemonic_entry(t,mne,mn,entry_mode,pad_entry=False):
 	if entry_mode == 'fixed':
 	if entry_mode == 'fixed':
 		mn = ['bkr'] + mn[:5] + ['nfb'] + mn[5:]
 		mn = ['bkr'] + mn[:5] + ['nfb'] + mn[5:]
 		ssl = mne.uniq_ss_len
 		ssl = mne.uniq_ss_len
-		mn = [w[:ssl] if len(w) >= ssl else (w[0] + 'z\b{}'.format('#'*(ssl-len(w))) + w[1:]) for w in mn]
+		def gen_mn():
+			for w in mn:
+				if len(w) >= ssl:
+					yield w[:ssl]
+				else:
+					yield w[0] + 'z\b' + '#' * (ssl-len(w)) + w[1:]
+		mn = list(gen_mn())
 	elif entry_mode in ('full','short'):
 	elif entry_mode in ('full','short'):
 		mn = ['fzr'] + mn[:5] + ['grd','grdbxm'] + mn[5:]
 		mn = ['fzr'] + mn[:5] + ['grd','grdbxm'] + mn[5:]
 		mn = pad_mnemonic(mn,mne.em.ss_len)
 		mn = pad_mnemonic(mn,mne.em.ss_len)

+ 1 - 1
test/tooltest2.py

@@ -824,7 +824,7 @@ async def run_test(gid,cmd_name):
 			vmsg(cp.stderr.strip().decode())
 			vmsg(cp.stderr.strip().decode())
 		if cp.returncode != 0:
 		if cp.returncode != 0:
 			import re
 			import re
-			m = re.match(b"tool command returned '(None|False)'"+NL.encode(),cp.stderr)
+			m = re.match(b'tool command returned (None|False)'+NL.encode(),cp.stderr)
 			if m:
 			if m:
 				return { b'None': None, b'False': False }[m.group(1)]
 				return { b'None': None, b'False': False }[m.group(1)]
 			else:
 			else: