Skip to content

Commit 2724eaa

Browse files
ting.wangsimenhhu
andcommitted
Refactor TestImageList with FakeImage class
Change-Id: Ia60f75f65bba4c25a0a87b570b081424f982efca Co-Authored-By: xiexs <xiexs@cn.fujitsu.com> Implements: blueprint improve-image-unittest-framework
1 parent 2b4c543 commit 2724eaa

1 file changed

Lines changed: 21 additions & 18 deletions

File tree

openstackclient/tests/image/v2/test_image.py

Lines changed: 21 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020

2121
from glanceclient.v2 import schemas
2222
from openstackclient.common import exceptions
23+
from openstackclient.common import utils as common_utils
2324
from openstackclient.image.v2 import image
2425
from openstackclient.tests import fakes
2526
from openstackclient.tests.identity.v3 import fakes as identity_fakes
@@ -472,25 +473,26 @@ def test_image_delete_multi_images(self):
472473

473474
class TestImageList(TestImage):
474475

476+
_image = image_fakes.FakeImage.create_one_image()
477+
475478
columns = (
476479
'ID',
477480
'Name',
478481
'Status',
479482
)
483+
480484
datalist = (
481-
(
482-
image_fakes.image_id,
483-
image_fakes.image_name,
484-
'',
485-
),
486-
)
485+
_image.id,
486+
_image.name,
487+
'',
488+
),
487489

488490
def setUp(self):
489491
super(TestImageList, self).setUp()
490492

491493
self.api_mock = mock.Mock()
492494
self.api_mock.image_list.side_effect = [
493-
[copy.deepcopy(image_fakes.IMAGE)], [],
495+
[image_fakes.FakeImage.get_image_info(self._image)], [],
494496
]
495497
self.app.client_manager.image.api = self.api_mock
496498

@@ -615,23 +617,24 @@ def test_image_list_long_option(self):
615617

616618
self.assertEqual(collist, columns)
617619
datalist = ((
618-
image_fakes.image_id,
619-
image_fakes.image_name,
620-
'',
620+
self._image.id,
621+
self._image.name,
621622
'',
622623
'',
623624
'',
624-
'public',
625-
False,
626-
image_fakes.image_owner,
627625
'',
626+
self._image.visibility,
627+
self._image.protected,
628+
self._image.owner,
629+
common_utils.format_list(self._image.tags),
628630
), )
629631
self.assertEqual(datalist, tuple(data))
630632

631633
@mock.patch('openstackclient.api.utils.simple_filter')
632634
def test_image_list_property_option(self, sf_mock):
633635
sf_mock.return_value = [
634-
copy.deepcopy(image_fakes.IMAGE),
636+
copy.deepcopy(
637+
image_fakes.FakeImage.get_image_info(self._image)),
635638
]
636639

637640
arglist = [
@@ -648,7 +651,7 @@ def test_image_list_property_option(self, sf_mock):
648651
columns, data = self.cmd.take_action(parsed_args)
649652
self.api_mock.image_list.assert_called_with()
650653
sf_mock.assert_called_with(
651-
[image_fakes.IMAGE],
654+
[image_fakes.FakeImage.get_image_info(self._image)],
652655
attr='a',
653656
value='1',
654657
property_field='properties',
@@ -660,7 +663,8 @@ def test_image_list_property_option(self, sf_mock):
660663
@mock.patch('openstackclient.common.utils.sort_items')
661664
def test_image_list_sort_option(self, si_mock):
662665
si_mock.return_value = [
663-
copy.deepcopy(image_fakes.IMAGE)
666+
copy.deepcopy(
667+
image_fakes.FakeImage.get_image_info(self._image))
664668
]
665669

666670
arglist = ['--sort', 'name:asc']
@@ -673,10 +677,9 @@ def test_image_list_sort_option(self, si_mock):
673677
columns, data = self.cmd.take_action(parsed_args)
674678
self.api_mock.image_list.assert_called_with()
675679
si_mock.assert_called_with(
676-
[image_fakes.IMAGE],
680+
[image_fakes.FakeImage.get_image_info(self._image)],
677681
'name:asc'
678682
)
679-
680683
self.assertEqual(self.columns, columns)
681684
self.assertEqual(self.datalist, tuple(data))
682685

0 commit comments

Comments
 (0)