def test_type_upsert_invalid_core_type_raises_core_error(self, mock_save):
mock_user = create_mock_user("1", is_superuser=True)
mock_request = create_mock_request(user=mock_user)
type_object = _create_type(content="<schema></schema>")
mock_save.return_value = None
with self.assertRaises(exceptions.CoreError):
type_api.upsert(type_object, request=mock_request)
def insert(type_version_manager, type_object, request, list_bucket_ids=None):
"""Add a version to a type version manager.
Args:
type_version_manager:
type_object:
request:
list_bucket_ids:
Returns:
"""
# save the type in database
type_api.upsert(type_object, request=request)
try:
# insert the initial template in the version manager
version_manager_api.insert_version(
type_version_manager, type_object, request=request
)
# insert the version manager in database
version_manager_api.upsert(type_version_manager, request=request)
# Add to the selected buckets
bucket_api.add_type_to_buckets(type_version_manager, list_bucket_ids)
# get type display name
display_name = get_latest_version_name(type_version_manager)
# update saved template
template_api.set_display_name(type_object, display_name, request=request)
# update saved type
return type_version_manager
except Exception as e:
type_api.delete(type_object, request=request)
raise e
def test_type_upsert_invalid_filename_raises_validation_error(
self, mock_save):
mock_user = create_mock_user("1", is_superuser=True)
mock_request = create_mock_request(user=mock_user)
type_object = _create_type(filename=1)
mock_save.side_effect = django_exceptions.ValidationError("")
with self.assertRaises(django_exceptions.ValidationError):
type_api.upsert(type_object, request=mock_request)
def test_type_upsert_valid_returns_type(self, mock_save):
mock_user = create_mock_user("1", is_superuser=True)
mock_request = create_mock_request(user=mock_user)
type_object = _create_type()
mock_save.return_value = type_object
result = type_api.upsert(type_object, request=mock_request)
self.assertIsInstance(result, Type)
def test_type_upsert_invalid_core_type_raises_core_error(self, mock_save):
type_object = _create_type(content="<schema></schema>")
mock_save.return_value = None
with self.assertRaises(exceptions.CoreError):
type_api.upsert(type_object)
def test_type_upsert_invalid_filename_raises_validation_error(self, mock_save):
type_object = _create_type(filename=1)
mock_save.side_effect = django_exceptions.ValidationError("")
with self.assertRaises(django_exceptions.ValidationError):
type_api.upsert(type_object)
def test_type_upsert_valid_returns_type(self, mock_save):
type_object = _create_type()
mock_save.return_value = type_object
result = type_api.upsert(type_object)
self.assertIsInstance(result, Type)