def deprecate_dataset_version(datasetVersionId, deprecationReasonObj):
reason = deprecationReasonObj["deprecationReason"]
try:
models_controller.deprecate_dataset_version(datasetVersionId, reason)
# TODO: Manage case by case exceptions => NotFound for example
except:
error = sys.exc_info()[1]
flask.abort(500,
"Error in deprecate_dataset_version: {}".format(error))
return flask.jsonify({})
def test_state_deprecated_to_approved(session: SessionBase, new_dataset,
new_dataset_version):
dataset_version_id = new_dataset_version.id
# TODO: Might not be the best way to test this, since we rely on the deprecation function to work
mc.deprecate_dataset_version(new_dataset_version.id, "test deprecation")
assert new_dataset_version.state == DatasetVersion.DatasetVersionState.deprecated
mc.approve_dataset_version(dataset_version_id)
assert new_dataset_version.state == DatasetVersion.DatasetVersionState.approved
def test_state_approved_to_deprecated(session: SessionBase, new_dataset,
new_dataset_version):
assert new_dataset_version.state == DatasetVersion.DatasetVersionState.approved
dataset_version_id = new_dataset_version.id
mc.deprecate_dataset_version(dataset_version_id, "test deprecation")
updated_dataset_version = mc.get_dataset_version(
dataset_version_id=dataset_version_id)
assert (updated_dataset_version.state ==
DatasetVersion.DatasetVersionState.deprecated)
def test_error_de_deprecate_dataset_version(session: SessionBase, new_dataset: Dataset):
# Deprecate the dataset version first
dataset_version_id = new_dataset.dataset_versions[0].id
models_controller.deprecate_dataset_version(
dataset_version_id, "Test de-deprecation"
)
error_raised = False
try:
endpoint.de_deprecate_dataset_version("")
except:
error_raised = True
assert error_raised, "Error not raised despite no dataset_version id passed"
def test_de_deprecate_dataset_version(session: SessionBase, new_dataset: Dataset):
# Deprecate the dataset version first
dataset_version_id = new_dataset.dataset_versions[0].id
models_controller.deprecate_dataset_version(
dataset_version_id, "Test de-deprecation"
)
flask_answer = endpoint.de_deprecate_dataset_version(dataset_version_id)
ack = get_data_from_flask_jsonify(flask_answer)
updated_dataset_version = models_controller.get_dataset_version(dataset_version_id)
assert updated_dataset_version.state == DatasetVersion.DatasetVersionState.approved
assert updated_dataset_version.reason_state == "Test de-deprecation"
def test_de_delete_dataset_version(session: SessionBase, new_dataset: Dataset):
dataset_version = new_dataset.dataset_versions[0]
dataset_version_id = dataset_version.id
deprecation_reason = "DeprecationReason"
reason_obj_should_not_be = {"deprecationReason": "ShouldNotBeSet"}
models_controller.deprecate_dataset_version(
dataset_version_id=dataset_version_id, reason="DeprecationReason"
)
models_controller.delete_dataset_version(dataset_version_id=dataset_version_id)
endpoint.deprecate_dataset_version(
dataset_version_id, deprecationReasonObj=reason_obj_should_not_be
)
assert dataset_version.state == DatasetVersion.DatasetVersionState.deprecated
assert dataset_version.reason_state == deprecation_reason