Merge "Fix intermittent v2 shell unit test failures"

This commit is contained in:
Zuul 2018-04-03 14:41:45 +00:00 committed by Gerrit Code Review
commit 6bab2404ec
1 changed files with 7 additions and 2 deletions

View File

@ -758,10 +758,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:
@ -773,10 +776,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: