From 4d577609eead56a323c497609c4330a939795b23 Mon Sep 17 00:00:00 2001
From: The MMGen Project <mmgen@tuta.io>
Date: Fri, 8 Oct 2021 16:44:56 +0000
Subject: [PATCH] test.py --segwit: execute only relevant tests

---
 test/test.py              |  7 +++++++
 test/test_py_d/ts_main.py | 39 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 46 insertions(+)

diff --git a/test/test.py b/test/test.py
index a0e574f4..b461ead5 100755
--- a/test/test.py
+++ b/test/test.py
@@ -186,6 +186,8 @@ if not (opt.resume or opt.skip_deps):
 
 check_segwit_opts()
 
+testing_segwit = opt.segwit or opt.segwit_random or opt.bech32
+
 if g.test_suite_deterministic:
 	opt.no_timings = True
 	init_color(num_colors=0)
@@ -790,6 +792,7 @@ class TestSuiteRunner(object):
 
 		os.environ['MMGEN_BOGUS_WALLET_DATA'] = '' # zero this here, so test group doesn't have to
 		self.ts = self.gm.gm_init_group(self,gname,self.spawn_wrapper)
+		self.ts_clsname = type(self.ts).__name__
 
 		if opt.resume_after:
 			global resume
@@ -863,6 +866,10 @@ class TestSuiteRunner(object):
 			dpy          = False ):
 
 		self.ts.test_name = cmd
+
+		if self.ts_clsname == 'TestSuiteMain' and testing_segwit and cmd not in self.ts.segwit_do:
+			return False
+
 		rerun = root # force_delete is not passed to recursive call
 
 		fns = []
diff --git a/test/test_py_d/ts_main.py b/test/test_py_d/ts_main.py
index 044482e1..93bdf554 100755
--- a/test/test_py_d/ts_main.py
+++ b/test/test_py_d/ts_main.py
@@ -146,6 +146,45 @@ class TestSuiteMain(TestSuiteBase,TestSuiteShared):
 		('txcreate6', (21,'transaction creation with corrected vsize (6)', [[['addrs'],21]])),
 		('txsign6',   (21,'transaction signing with corrected vsize',      [[['mmdat','rawtx'],21]])),
 	)
+	segwit_do = (
+		'walletgen',
+		'addrgen',
+		'txcreate',
+		'txbump',
+		'txsign',
+		'txsend',
+		'txdo',
+		'export_incog',
+		'keyaddrgen',
+		'txsign_keyaddr',
+		'txcreate_ni',
+		'walletgen2',
+		'addrgen2',
+		'txcreate2',
+		'txsign2',
+		'export_mnemonic2',
+		'walletgen3',
+		'addrgen3',
+		'txcreate3',
+		'txsign3',
+		'walletgen14',
+		'addrgen14',
+		'keyaddrgen14',
+		'walletgen4',
+		'addrgen4',
+		'txcreate4',
+		'txsign4',
+		'txdo4',
+		'txbump4',
+		'walletgen5',
+		'addrgen5',
+		'txcreate5',
+		'txsign5',
+		'walletgen6',
+		'addrgen6',
+		'txcreate6',
+		'txsign6',
+	)
 
 	def __init__(self,trunner,cfgs,spawn):
 		TestSuiteBase.__init__(self,trunner,cfgs,spawn)