X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/c193d814c22e2a4227c7f49e76b0d9b589cff4be..2ab1b540a710fd454f33e9ba22db601e67f3d217:/services/nodemanager/tests/test_failure.py?ds=inline

diff --git a/services/nodemanager/tests/test_failure.py b/services/nodemanager/tests/test_failure.py
index f543f645f6..285aa03c7d 100644
--- a/services/nodemanager/tests/test_failure.py
+++ b/services/nodemanager/tests/test_failure.py
@@ -29,27 +29,27 @@ class BogusActor(arvnodeman.baseactor.BaseNodeManagerActor):
         time.sleep(2)
         return True
 
-class ActorUnhandledExceptionTest(unittest.TestCase):
+class ActorUnhandledExceptionTest(testutil.ActorTestMixin, unittest.TestCase):
     def test_fatal_error(self):
         for e in (MemoryError(), threading.ThreadError(), OSError(errno.ENOMEM, "")):
-            with mock.patch('os.killpg') as killpg_mock:
+            with mock.patch('os.kill') as kill_mock:
                 act = BogusActor.start(e).tell_proxy()
                 act.doStuff()
                 act.actor_ref.stop(block=True)
-                self.assertTrue(killpg_mock.called)
+                self.assertTrue(kill_mock.called)
 
-    @mock.patch('os.killpg')
-    def test_nonfatal_error(self, killpg_mock):
+    @mock.patch('os.kill')
+    def test_nonfatal_error(self, kill_mock):
         act = BogusActor.start(OSError(errno.ENOENT, "")).tell_proxy()
         act.doStuff()
         act.actor_ref.stop(block=True)
-        self.assertFalse(killpg_mock.called)
+        self.assertFalse(kill_mock.called)
 
-class WatchdogActorTest(unittest.TestCase):
-    @mock.patch('os.killpg')
-    def test_time_timout(self, killpg_mock):
+class WatchdogActorTest(testutil.ActorTestMixin, unittest.TestCase):
+    @mock.patch('os.kill')
+    def test_time_timout(self, kill_mock):
         act = BogusActor.start(OSError(errno.ENOENT, ""))
         watch = arvnodeman.baseactor.WatchdogActor.start(1, act)
         watch.stop(block=True)
         act.stop(block=True)
-        self.assertTrue(killpg_mock.called)
+        self.assertTrue(kill_mock.called)