diff options
| -rw-r--r-- | glanceclient/tests/unit/v2/test_shell_v2.py | 9 | ||||
| -rw-r--r-- | glanceclient/v2/shell.py | 2 |
2 files changed, 5 insertions, 6 deletions
diff --git a/glanceclient/tests/unit/v2/test_shell_v2.py b/glanceclient/tests/unit/v2/test_shell_v2.py index 4ac8c17..d75613f 100644 --- a/glanceclient/tests/unit/v2/test_shell_v2.py +++ b/glanceclient/tests/unit/v2/test_shell_v2.py @@ -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') diff --git a/glanceclient/v2/shell.py b/glanceclient/v2/shell.py index 131e826..c9f1fe1 100644 --- a/glanceclient/v2/shell.py +++ b/glanceclient/v2/shell.py @@ -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) |