X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/5bcba288077488791daa43a15d5fd5fb0c6e653c..861cff5ffc2eb2739573b8991fca1cfdc388377f:/services/nodemanager/tests/test_clientactor.py diff --git a/services/nodemanager/tests/test_clientactor.py b/services/nodemanager/tests/test_clientactor.py index 57a0d32d06..cee9c85a22 100644 --- a/services/nodemanager/tests/test_clientactor.py +++ b/services/nodemanager/tests/test_clientactor.py @@ -47,7 +47,8 @@ class RemotePollLoopActorTestCase(testutil.RemotePollLoopActorTestMixin, def test_late_subscribers_get_responses(self): self.build_monitor(['pre_late_test', 'late_test']) - self.monitor.subscribe(lambda response: None).get(self.TIMEOUT) + mock_subscriber = mock.Mock(name='mock_subscriber') + self.monitor.subscribe(mock_subscriber).get(self.TIMEOUT) self.monitor.subscribe(self.subscriber) self.monitor.poll().get(self.TIMEOUT) self.stop_proxy(self.monitor) @@ -146,4 +147,3 @@ class RemotePollLoopActorWithKeysTestCase(testutil.RemotePollLoopActorTestMixin, if __name__ == '__main__': unittest.main() -