def test_get_all_images(self):
scrub_queue = scrubber.ScrubDBQueue()
images = self._create_image_list(15)
image_pager = ImagePager(images)
def make_get_images_detailed(pager):
def mock_get_images_detailed(filters, marker=None):
return pager()
return mock_get_images_detailed
with patch.object(scrub_queue.registry, 'get_images_detailed') \
as _mock_get_images_detailed:
_mock_get_images_detailed.side_effect = \
make_get_images_detailed(image_pager)
actual = list(scrub_queue._get_all_images())
self.assertEqual(images, actual)
def test_get_all_images_paged(self):
scrub_queue = scrubber.ScrubDBQueue()
images = self._create_image_list(15)
image_pager = ImagePager(images, page_size=4)
def make_get_images_detailed(pager):
def mock_get_images_detailed(ctx, filters, marker=None,
limit=None):
return pager()
return mock_get_images_detailed
with patch.object(db_api, 'image_get_all') as (
_mock_get_images_detailed):
_mock_get_images_detailed.side_effect = (
make_get_images_detailed(image_pager))
actual = list(scrub_queue._get_all_images())
self.assertEqual(images, actual)
