def test_get_all_by_version_ids(self, mock_get_all_by_version_ids):
# Arrange
version_x = ObjectId()
version_y = ObjectId()
mock_template1 = _create_mock_template_version_manager("mock1", [version_x])
mock_template2 = _create_mock_template_version_manager("mock2", [version_y])
mock_get_all_by_version_ids.return_value = [mock_template1, mock_template2]
result = version_manager_api.get_all_by_version_ids([version_x, version_y])
# Assert
self.assertTrue(all(isinstance(item, TemplateVersionManager) for item in result))
def get_all_by_template_ids(template_ids):
""" Get all OaiProviderSet used by a list of templates ids.
Args:
template_ids: List of templates ids.
Returns:
List of OaiProviderSet.
"""
# Be sure to have string and not ObjectId()
string_ids = map(str, template_ids)
# Get all templates managers thanks to templates ids.
templates_manager = template_version_manager_api.get_all_by_version_ids(
string_ids)
# Get all OaiProviderSet used by those templates manager
return get_all_by_templates_manager(templates_manager)
def test_get_all_by_version_ids_global_template_as_superuser_returns_template(
self):
mock_request = create_mock_request(user=self.superuser1)
template_vm_api.get_all_by_version_ids(
[str(self.fixture.global_template.id)], request=mock_request)
def test_get_all_by_version_ids_other_users_template_as_staff_raises_access_control_error(
self, ):
mock_request = create_mock_request(user=self.staff_user1)
with self.assertRaises(AccessControlError):
template_vm_api.get_all_by_version_ids(
[str(self.fixture.user2_template.id)], request=mock_request)
def test_get_all_by_version_ids_own_template_as_staff_returns_template(
self):
mock_request = create_mock_request(user=self.staff_user1)
template_vm_api.get_all_by_version_ids(
[str(self.fixture.user1_template.id)], request=mock_request)
def test_get_all_by_version_ids_global_template_as_anonymous_raises_access_control_error(
self, ):
mock_request = create_mock_request(user=self.anonymous_user)
with self.assertRaises(AccessControlError):
template_vm_api.get_all_by_version_ids(
[str(self.fixture.global_template.id)], request=mock_request)
