Use 'addCleanup' instead of 'tearDown' in engine(s) test

Change-Id: I1d9b13b6f5cd48b7ac98f5c34ef9cb837f9ca7d1

1 files changed, 6 insertions, 5 deletions

diff --git a/taskflow/tests/unit/test_engines.py b/taskflow/tests/unit/test_engines.py
index 0c3ac03..1a68aa7 100644
--- a/taskflow/tests/unit/test_engines.py
+++ b/taskflow/tests/unit/test_engines.py
@@ -1106,14 +1106,15 @@ class WorkerBasedEngineTest(EngineTaskTest,
         self.worker_thread = tu.daemon_thread(self.worker.run)
         self.worker_thread.start()
 
+        # Ensure worker and thread is stopped when test is done; these are
+        # called in reverse order, so make sure we signal the stop before
+        # performing the join (because the reverse won't work).
+        self.addCleanup(self.worker_thread.join)
+        self.addCleanup(self.worker.stop)
+
         # Make sure the worker is started before we can continue...
         self.worker.wait()
 
-    def tearDown(self):
-        self.worker.stop()
-        self.worker_thread.join()
-        super(WorkerBasedEngineTest, self).tearDown()
-
     def _make_engine(self, flow,
                      flow_detail=None, store=None):
         return taskflow.engines.load(flow, flow_detail=flow_detail,