def get_global_version_managers():
"""Return all Version Managers with user set to None.
Returns:
"""
return VersionManager.get_global_version_managers()
def get_all_type_version_manager():
"""Return all Version Managers of all users.
Args:
Returns:
"""
return VersionManager.objects(_cls=TypeVersionManager.class_name).all()
def _create_version_manager(title="", versions=[]):
"""
Returns a templates version manager
:param title:
:return:
"""
return VersionManager(id=ObjectId(),
title=title,
versions=versions,
disabled_versions=[])
def get(version_manager_id):
"""Get a version manager by its id.
Args:
version_manager_id:
Returns:
"""
return VersionManager.get_by_id(version_manager_id)
def get_by_id_list(list_id):
"""Get a version managers with the given id list.
Args:
list_id:
Returns:
"""
return VersionManager.get_by_id_list(list_id)
def get_active_global_version_manager_by_title(version_manager_title):
"""Get active global version manager by title
Args:
version_manager_title:
Returns:
"""
return VersionManager.get_active_global_version_manager_by_title(
version_manager_title)
def get_all_type_version_manager_except_user_id(user_id):
""" Return all Version Managers of all users except user with given user id.
Args:
user_id:
Returns:
"""
return VersionManager.objects(_cls=TypeVersionManager.class_name,
user__nin=str(user_id)).all()
def get_active_global_version_manager_by_title(version_manager_title):
"""Return active Version Manager by its title with user set to None.
Args:
version_manager_title: Version Manager title
Returns:
Version Manager instance
"""
return VersionManager.get_active_global_version_manager_by_title(
version_manager_title)
def get_from_version(version):
"""Return a version manager from a version.
Args:
version:
Returns:
"""
version_managers = VersionManager.get_all()
for version_manager in version_managers:
if str(version.id) in version_manager.versions:
return version_manager
raise exceptions.ApiError(
"No version manager could be found for this version.")
def test_staff_returns_http_200(self, template_version_manager_data,
version_manager_get_by_id):
template_version_manager_data.return_value = True
version_manager_get_by_id.return_value = VersionManager(user="******")
mock_user = create_mock_user("1", is_staff=True)
response = RequestMock.do_request_get(
template_version_manager_views.TemplateVersionManagerDetail.
as_view(),
mock_user,
param={"pk": self.fake_id},
)
self.assertEqual(response.status_code, status.HTTP_200_OK)
def test_staff_returns_http_200(
self,
template_get_by_id,
version_manager_get_from_version,
version_manager_disable_version,
):
version_manager_disable_version.return_value = VersionManager(user="******")
version_manager_get_from_version.return_value = {}
template_get_by_id.return_value = Template(user="******")
mock_user = create_mock_user("1", is_staff=True)
response = RequestMock.do_request_patch(
template_version_manager_views.DisableTemplateVersion.as_view(),
mock_user,
param={"pk": self.fake_id},
)
self.assertEqual(response.status_code, status.HTTP_200_OK)
def test_staff_returns_http_201(
self,
template_version_manager_serializer_data,
template_version_manager_serializer_save,
template_version_manager_serializer_valid,
version_manager_get_by_id,
):
version_manager_get_by_id.return_value = VersionManager(user=None)
template_version_manager_serializer_data.return_value = True
template_version_manager_serializer_save.return_value = None
template_version_manager_serializer_valid.return_value = {}
mock_user = create_mock_user("1", is_staff=True)
response = RequestMock.do_request_post(
template_version_manager_views.TemplateVersion.as_view(),
mock_user,
data={},
param={"pk": self.fake_id},
)
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
def test_staff_returns_http_200(
self,
bucket_serializer_data,
bucket_serializer_save,
bucket_serializer_valid,
version_manager_get_by_id,
bucket_get_by_id,
):
bucket_get_by_id.return_value = {}
version_manager_get_by_id.return_value = VersionManager(user=None)
bucket_serializer_data.return_value = True
bucket_serializer_save.return_value = None
bucket_serializer_valid.return_value = {}
mock_user = create_mock_user("1", is_staff=True)
response = RequestMock.do_request_patch(
bucket_views.TypeVersionManagerBuckets.as_view(),
mock_user,
param={"pk": self.fake_id},
data=[],
)
self.assertEqual(response.status_code, status.HTTP_200_OK)
