def test_not_a_page(self):
"""Assert non-paginated responses are immediately returned."""
kwargs = {key: mock.Mock() for key in _HANDLER_ARGS}
kwargs['client']._cfg.pulp_version = Version('3') # pylint:disable=protected-access
with mock.patch.object(api, 'json_handler') as json_handler:
json_handler.return_value = {}
return_value = api.page_handler(**kwargs)
self.assertIs(return_value, json_handler.return_value)
def test_not_a_page(self):
"""Assert non-paginated responses are immediately returned."""
kwargs = {key: mock.Mock() for key in _HANDLER_ARGS}
kwargs["client"]._cfg.pulp_version = Version("3") # pylint:disable=protected-access
with mock.patch.object(api, "json_handler") as json_handler:
json_handler.return_value = {}
return_value = api.page_handler(**kwargs)
self.assertIs(return_value, json_handler.return_value)
def test_204_check_run(self):
"""Assert HTTP 204 responses are immediately returned."""
kwargs = {key: mock.Mock() for key in _HANDLER_ARGS}
kwargs['client']._cfg.pulp_version = Version('3') # pylint:disable=protected-access
with mock.patch.object(api, 'json_handler') as json_handler:
json_handler.return_value = mock.Mock()
return_value = api.page_handler(**kwargs)
self.assertIs(return_value, json_handler.return_value)
def test_is_a_page(self):
"""Assert paginated responses are collected."""
kwargs = {key: mock.Mock() for key in _HANDLER_ARGS}
kwargs['client']._cfg.pulp_version = Version('3') # pylint:disable=protected-access
with mock.patch.object(api, 'json_handler') as json_handler:
json_handler.return_value = {'results': None}
with mock.patch.object(api, '_walk_pages') as walk_pages:
walk_pages.return_value = ((1, 2), (3, 4))
return_value = api.page_handler(**kwargs)
self.assertEqual(return_value, [1, 2, 3, 4])
def test_is_a_page(self):
"""Assert paginated responses are collected."""
kwargs = {key: mock.Mock() for key in _HANDLER_ARGS}
kwargs["client"]._cfg.pulp_version = Version("3") # pylint:disable=protected-access
with mock.patch.object(api, "json_handler") as json_handler:
json_handler.return_value = {"results": None}
with mock.patch.object(api, "_walk_pages") as walk_pages:
walk_pages.return_value = ((1, 2), (3, 4))
return_value = api.page_handler(**kwargs)
self.assertEqual(return_value, [1, 2, 3, 4])
def test_pulp_2_error(self):
"""Assert this handler can't be used with Pulp 2."""
kwargs = {key: mock.Mock() for key in _HANDLER_ARGS}
kwargs['client']._cfg.pulp_version = Version('2') # pylint:disable=protected-access
with self.assertRaises(ValueError):
api.page_handler(**kwargs)
def test_pulp_2_error(self):
"""Assert this handler can't be used with Pulp 2."""
kwargs = {key: mock.Mock() for key in _HANDLER_ARGS}
kwargs["client"]._cfg.pulp_version = Version("2") # pylint:disable=protected-access
with self.assertRaises(ValueError):
api.page_handler(**kwargs)
