From db4c0158fb76b976a4ecd218effb250fc51d240d Mon Sep 17 00:00:00 2001
From: The MMGen Project <mmgen@tuta.io>
Date: Sun, 26 Sep 2021 21:16:54 +0000
Subject: [PATCH] test.py: clean tmpdirs before initializing test groups

---
 test/test.py              | 12 ++++++------
 test/test_py_d/ts_base.py |  2 ++
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/test/test.py b/test/test.py
index 3eaccc18..b047abcd 100755
--- a/test/test.py
+++ b/test/test.py
@@ -715,9 +715,13 @@ class TestSuiteRunner(object):
 		m = '{} test{} performed.  Elapsed time: {:02d}:{:02d}\n'
 		sys.stderr.write(green(m.format(self.cmd_total,suf(self.cmd_total),t//60,t%60)))
 
-	def init_group(self,gname,cmd=None,quiet=False):
+	def init_group(self,gname,cmd=None,quiet=False,do_clean=True):
+
 		ts_cls = CmdGroupMgr().load_mod(gname)
 
+		if do_clean:
+			clean(ts_cls.tmpdir_nums)
+
 		for k in ('segwit','segwit_random','bech32'):
 			if getattr(opt,k):
 				segwit_opt = k
@@ -774,7 +778,6 @@ class TestSuiteRunner(object):
 				if arg in self.gm.cmd_groups:
 					if not self.init_group(arg):
 						continue
-					clean(self.ts.tmpdir_nums)
 					for cmd in self.gm.cmd_list:
 						self.check_needs_rerun(cmd,build=True)
 						do_between()
@@ -789,10 +792,8 @@ class TestSuiteRunner(object):
 						gname = self.gm.find_cmd_in_groups(arg)
 					if gname:
 						same_grp = gname == gname_save # same group as previous cmd: don't clean, suppress blue msg
-						if not self.init_group(gname,arg,quiet=same_grp):
+						if not self.init_group(gname,arg,quiet=same_grp,do_clean=not same_grp):
 							continue
-						if not same_grp:
-							clean(self.ts.tmpdir_nums)
 						try:
 							self.check_needs_rerun(arg,build=True)
 						except Exception as e: # allow calling of functions not in cmd_group
@@ -817,7 +818,6 @@ class TestSuiteRunner(object):
 					continue
 				if not self.init_group(gname):
 					continue
-				clean(self.ts.tmpdir_nums)
 				for cmd in self.gm.cmd_list:
 					self.check_needs_rerun(cmd,build=True)
 					do_between()
diff --git a/test/test_py_d/ts_base.py b/test/test_py_d/ts_base.py
index e263eb51..a14549f9 100755
--- a/test/test_py_d/ts_base.py
+++ b/test/test_py_d/ts_base.py
@@ -48,6 +48,8 @@ class TestSuiteBase(object):
 		self.tn_ext = ('','.testnet')[self.proto.testnet]
 		d = {'bch':'btc','btc':'btc','ltc':'ltc'}
 		self.fork = d[self.proto.coin.lower()] if self.proto.coin.lower() in d else None
+		if len(self.tmpdir_nums) == 1:
+			self.tmpdir_num = self.tmpdir_nums[0]
 
 	@property
 	def tmpdir(self):