From be5867dd2cff5949df0d94d263791ce8676a9da4 Mon Sep 17 00:00:00 2001
From: The MMGen Project <mmgen@tuta.io>
Date: Thu, 3 Feb 2022 20:40:41 +0000
Subject: [PATCH] test.py: minor overlay fix

---
 test/overlay/__init__.py |  5 ++++-
 test/test.py             | 12 ++++++------
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/test/overlay/__init__.py b/test/overlay/__init__.py
index 3333824b..f174ae73 100644
--- a/test/overlay/__init__.py
+++ b/test/overlay/__init__.py
@@ -1,5 +1,8 @@
 import sys,os,shutil
 
+def get_overlay_dir(repo_root):
+	return os.path.join(repo_root,'test','overlay','tree')
+
 def overlay_setup(repo_root):
 
 	def process_srcdir(d):
@@ -24,7 +27,7 @@ def overlay_setup(repo_root):
 					os.path.join(srcdir,fn),
 					os.path.join(destdir,link_fn) )
 
-	overlay_dir = os.path.join(repo_root,'test','overlay','tree')
+	overlay_dir = get_overlay_dir(repo_root)
 	fakemod_dir = os.path.join(repo_root,'test','overlay','fakemods')
 	fakemods  = os.listdir(fakemod_dir)
 	make_link = os.symlink if sys.platform == 'linux' else shutil.copy2
diff --git a/test/test.py b/test/test.py
index 104f3e3e..6c5e08f3 100755
--- a/test/test.py
+++ b/test/test.py
@@ -73,8 +73,8 @@ def create_shm_dir(data_dir,trash_dir):
 import sys,os,time
 
 from include.tests_header import repo_root
-from test.overlay import overlay_setup
-overlay_dir = overlay_setup(repo_root)
+from test.overlay import get_overlay_dir,overlay_setup
+overlay_dir = get_overlay_dir(repo_root)
 sys.path.insert(0,overlay_dir)
 
 try: os.unlink(os.path.join(repo_root,'my.err'))
@@ -796,12 +796,9 @@ class TestSuiteRunner(object):
 		self.start_time = time.time()
 		self.daemons_started = False
 		gname_save = None
+		overlay_setup(repo_root)
 		if usr_args:
 			for arg in usr_args:
-				if arg in utils:
-					params = usr_args[usr_args.index(arg)+1:]
-					globals()[arg](*params)
-					sys.exit(0)
 				if arg in self.gm.cmd_groups:
 					if not self.init_group(arg):
 						continue
@@ -1027,6 +1024,9 @@ if opt.list_cmd_groups:
 	CmdGroupMgr().list_cmd_groups()
 elif opt.list_cmds:
 	list_cmds()
+elif usr_args and usr_args[0] in utils:
+	globals()[usr_args[0]](*usr_args[1:])
+	sys.exit(0)
 
 if opt.pause:
 	set_restore_term_at_exit()