diff options
-rw-r--r-- | glanceclient/tests/unit/v2/test_shell_v2.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/glanceclient/tests/unit/v2/test_shell_v2.py b/glanceclient/tests/unit/v2/test_shell_v2.py index 48afaf7..f2f30e0 100644 --- a/glanceclient/tests/unit/v2/test_shell_v2.py +++ b/glanceclient/tests/unit/v2/test_shell_v2.py @@ -1405,10 +1405,13 @@ class ShellV2Test(testtools.TestCase): self.assert_exits_with_msg(func=test_shell.do_image_delete, func_args=args) + @mock.patch('sys.stdout', autospec=True) @mock.patch.object(utils, 'print_err') - def test_do_image_download_with_forbidden_id(self, mocked_print_err): + def test_do_image_download_with_forbidden_id(self, mocked_print_err, + mocked_stdout): args = self._make_args({'id': 'IMG-01', 'file': None, 'progress': False}) + mocked_stdout.isatty = lambda: False with mock.patch.object(self.gc.images, 'data') as mocked_data: mocked_data.side_effect = exc.HTTPForbidden try: @@ -1420,10 +1423,12 @@ class ShellV2Test(testtools.TestCase): self.assertEqual(1, mocked_data.call_count) self.assertEqual(1, mocked_print_err.call_count) + @mock.patch('sys.stdout', autospec=True) @mock.patch.object(utils, 'print_err') - def test_do_image_download_with_500(self, mocked_print_err): + def test_do_image_download_with_500(self, mocked_print_err, mocked_stdout): args = self._make_args({'id': 'IMG-01', 'file': None, 'progress': False}) + mocked_stdout.isatty = lambda: False with mock.patch.object(self.gc.images, 'data') as mocked_data: mocked_data.side_effect = exc.HTTPInternalServerError try: |