def insert(template_version_manager, template):
"""Add a version to a template version manager.
Args:
template_version_manager:
template:
Returns:
"""
# save the template in database
template_api.upsert(template)
try:
# insert the initial template in the version manager
version_manager_api.insert_version(template_version_manager, template)
# insert the version manager in database
version_manager_api.upsert(template_version_manager)
# get template display name
display_name = get_latest_version_name(template_version_manager)
# update saved template
template_api.set_display_name(template, display_name)
# return version manager
return template_version_manager
except Exception, e:
template_api.delete(template)
raise e
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_version_manager_insert_first_version_sets_current(
self, mock_save):
# Arrange
version_manager = _create_version_manager()
version = _create_mock_object()
mock_save.return_value = None
# Act + Assert
version_manager_api.insert_version(version_manager, version)
self.assertEquals(version_manager.current, str(version.id))
def test_version_manager_insert_first_version_sets_current(
self, mock_save):
# Arrange
mock_user = create_mock_user("1", is_superuser=True)
mock_request = create_mock_request(mock_user)
version_manager = _create_version_manager()
version = _create_mock_object()
mock_save.return_value = None
# Act + Assert
version_manager_api.insert_version(version_manager,
version,
request=mock_request)
self.assertEquals(version_manager.current, str(version.id))