start/stop wb_get in TestRunnerBase, otherwise it never ends

diff --git a/src/openpower/test/runner.py b/src/openpower/test/runner.py
index fd6540253b85c1cddd92af317487193ba59ec076..cc91b75ddeda2604335343e9054e38b882474844 100644 (file)
--- a/src/openpower/test/runner.py
+++ b/src/openpower/test/runner.py
@@ -32,6 +32,7 @@ from openpower.decoder.power_decoder2 import PowerDecode2
 from soc.config.test.test_loadstore import TestMemPspec
 from nmutil.util import wrap
 from openpower.test.wb_get import wb_get
+import openpower.test.wb_get as wbget
 from openpower.test.state import TestState, StateRunner, ExpectedState
 
 
@@ -161,6 +162,10 @@ class TestRunnerBase(FHDLTestCase):
         # The methods contained in the respective Runner classes are
         # called using this list when possible
 
+        # allow wb_get to run
+        if self.rom is not None:
+            wbget.stop = False
+
         state_list = []
 
         if self.run_hdl:
@@ -314,6 +319,10 @@ class TestRunnerBase(FHDLTestCase):
             for runner in state_list:
                 yield from runner.cleanup() # TODO, some arguments?
 
+            # finally stop wb_get from going
+            if self.rom is not None:
+                wbget.stop = True
+
         styles = {
             'dec': {'base': 'dec'},
             'bin': {'base': 'bin'},