Merge "Compare against 'RequestIdProxy.wrapped'"

This commit is contained in:
Zuul
2018-01-02 22:51:25 +00:00
committed by Gerrit Code Review
2 changed files with 5 additions and 6 deletions

View File

@@ -581,11 +581,10 @@ class ShellV2Test(testtools.TestCase):
args = self._make_args(
{'id': 'IMG-01', 'file': 'test', 'progress': True})
with mock.patch.object(self.gc.images, 'data') as mocked_data:
def _data():
for c in 'abcedf':
yield c
mocked_data.return_value = utils.IterableWithLength(_data(), 5)
with mock.patch.object(self.gc.images, 'data') as mocked_data, \
mock.patch.object(utils, '_extract_request_id'):
mocked_data.return_value = utils.RequestIdProxy(
[c for c in 'abcdef'])
test_shell.do_image_download(self.gc, args)
mocked_data.assert_called_once_with('IMG-01')

View File

@@ -353,7 +353,7 @@ def do_image_download(gc, args):
msg = "Unable to download image '%s'. (%s)" % (args.id, e)
utils.exit(msg)
if body is None:
if body.wrapped is None:
msg = ('Image %s has no data.' % args.id)
utils.exit(msg)