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)