diff --git a/teeth_agent/tests/standby_agent.py b/teeth_agent/tests/standby_agent.py index 37214fa25..f1fe97b86 100644 --- a/teeth_agent/tests/standby_agent.py +++ b/teeth_agent/tests/standby_agent.py @@ -67,8 +67,9 @@ class TestBaseTeethAgent(unittest.TestCase): invalid_info) def test_cache_images_success(self): - self.agent.cache_images('cache_images', - [self._build_fake_image_info()]) + result = self.agent.cache_images('cache_images', + [self._build_fake_image_info()]) + result.join() def test_cache_images_invalid_image_list(self): self.assertRaises(errors.InvalidCommandParamsError,