Merge "Mock sleep in UT"

This commit is contained in:
Zuul 2018-01-30 00:32:19 +00:00 committed by Gerrit Code Review
commit ac79cd00b5
1 changed files with 5 additions and 3 deletions

View File

@ -102,10 +102,12 @@ class TestLocalDisk(test.NoDBTestCase):
mock_get_image.reset_mock()
exception = Exception
mock_get_image.side_effect = exception
self.assertRaises(exception,
self.get_ls(self.apt).create_disk_from_image,
None, inst, powervm.TEST_IMAGE1)
with mock.patch('time.sleep', autospec=True) as mock_sleep:
self.assertRaises(exception,
self.get_ls(self.apt).create_disk_from_image,
None, inst, powervm.TEST_IMAGE1)
self.assertEqual(mock_get_image.call_count, 4)
self.assertEqual(3, mock_sleep.call_count)
@mock.patch('pypowervm.tasks.storage.upload_new_vdisk', autospec=True)
@mock.patch('nova.image.api.API.download')