X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/10402f74ccdae453ae353c389a02c0415adcf1e7..59a972c831bc8f7cd4e896ed8e1c71277b97f04e:/services/nodemanager/tests/test_failure.py diff --git a/services/nodemanager/tests/test_failure.py b/services/nodemanager/tests/test_failure.py index ef4423dafa..8bf3ea8741 100644 --- a/services/nodemanager/tests/test_failure.py +++ b/services/nodemanager/tests/test_failure.py @@ -17,6 +17,7 @@ import pykka from . import testutil import arvnodeman.baseactor +import arvnodeman.status as status class BogusActor(arvnodeman.baseactor.BaseNodeManagerActor): def __init__(self, e, killfunc=None): @@ -45,11 +46,16 @@ class ActorUnhandledExceptionTest(testutil.ActorTestMixin, unittest.TestCase): self.assertTrue(kill_mock.called) def test_nonfatal_error(self): + status.tracker.update({'actor_exceptions': 0}) kill_mock = mock.Mock('os.kill') - act = BogusActor.start(OSError(errno.ENOENT, ""), killfunc=kill_mock).tell_proxy() + bgact = BogusActor.start(OSError(errno.ENOENT, ""), killfunc=kill_mock) + act_thread = bgact.proxy().get_thread().get() + act = bgact.tell_proxy() act.doStuff() act.actor_ref.stop(block=True) + act_thread.join() self.assertFalse(kill_mock.called) + self.assertEqual(1, status.tracker.get('actor_exceptions')) class WatchdogActorTest(testutil.ActorTestMixin, unittest.TestCase):