X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/85c71a36173550f14fb1d5f4092f2050ec8dc033..fbd54468b13466839c24d880a3d041d0a49371af:/services/nodemanager/tests/testutil.py diff --git a/services/nodemanager/tests/testutil.py b/services/nodemanager/tests/testutil.py index b266ed62da..6cde766fa3 100644 --- a/services/nodemanager/tests/testutil.py +++ b/services/nodemanager/tests/testutil.py @@ -131,12 +131,13 @@ class DriverTestMixin(object): return getattr(self.driver_mock(), method_name).call_args def test_driver_create_retry(self): - driver_mock2 = mock.MagicMock(name='driver_mock2') - self.driver_mock.side_effect = (Exception("oops"), driver_mock2) - kwargs = {'user_id': 'foo'} - driver = self.new_driver(auth_kwargs=kwargs) - self.assertTrue(self.driver_mock.called) - self.assertIs(driver.real, driver_mock2) + with mock.patch('time.sleep'): + driver_mock2 = mock.MagicMock(name='driver_mock2') + self.driver_mock.side_effect = (Exception("oops"), driver_mock2) + kwargs = {'user_id': 'foo'} + driver = self.new_driver(auth_kwargs=kwargs) + self.assertTrue(self.driver_mock.called) + self.assertIs(driver.real, driver_mock2) class RemotePollLoopActorTestMixin(ActorTestMixin): def build_monitor(self, *args, **kwargs):