def test_delete_model(firebase_model):
ml.delete_model(firebase_model.model_id)
# Second delete of same model will fail
with pytest.raises(exceptions.NotFoundError) as excinfo:
ml.delete_model(firebase_model.model_id)
check_firebase_error(excinfo, 404, 'Requested entity was not found.')
def test_delete_model(self):
recorder = instrument_ml_service(status=200, payload=EMPTY_RESPONSE)
ml.delete_model(MODEL_ID_1) # no response for delete
assert len(recorder) == 1
assert recorder[0].method == 'DELETE'
assert recorder[0].url == TestDeleteModel._url(PROJECT_ID, MODEL_ID_1)
assert recorder[0].headers[HEADER_CLIENT_KEY] == HEADER_CLIENT_VALUE
def test_get_non_existing_model(firebase_model):
# Get a valid model_id that no longer exists
ml.delete_model(firebase_model.model_id)
with pytest.raises(exceptions.NotFoundError) as excinfo:
ml.get_model(firebase_model.model_id)
check_firebase_error(excinfo, 404, 'Requested entity was not found.')
def _clean_up_model(model):
try:
# Try to delete the model.
# Some tests delete the model as part of the test.
ml.delete_model(model.model_id)
except exceptions.NotFoundError:
pass
def test_update_non_existing_model(firebase_model):
ml.delete_model(firebase_model.model_id)
firebase_model.tags = ['tag987']
with pytest.raises(exceptions.NotFoundError) as excinfo:
ml.update_model(firebase_model)
check_operation_error(
excinfo,
'Model \'{0}\' was not found'.format(firebase_model.as_dict().get('name')))
def test_delete_model_error(self):
recorder = instrument_ml_service(status=404,
payload=ERROR_RESPONSE_NOT_FOUND)
with pytest.raises(exceptions.NotFoundError) as excinfo:
ml.delete_model(MODEL_ID_1)
check_firebase_error(excinfo, ERROR_STATUS_NOT_FOUND,
ERROR_CODE_NOT_FOUND, ERROR_MSG_NOT_FOUND)
assert len(recorder) == 1
assert recorder[0].method == 'DELETE'
assert recorder[0].url == self._url(PROJECT_ID, MODEL_ID_1)
assert recorder[0].headers[HEADER_CLIENT_KEY] == HEADER_CLIENT_VALUE
def test_publish_unpublish_non_existing_model(firebase_model):
ml.delete_model(firebase_model.model_id)
with pytest.raises(exceptions.NotFoundError) as excinfo:
ml.publish_model(firebase_model.model_id)
check_operation_error(
excinfo,
'Model \'{0}\' was not found'.format(firebase_model.as_dict().get('name')))
with pytest.raises(exceptions.NotFoundError) as excinfo:
ml.unpublish_model(firebase_model.model_id)
check_operation_error(
excinfo,
'Model \'{0}\' was not found'.format(firebase_model.as_dict().get('name')))
def delete_model(model_id):
"""Delete a model from the project."""
ml.delete_model(model_id)
def evaluate():
app = firebase_admin.initialize_app(testutils.MockCredential(),
name='no_project_id')
with pytest.raises(ValueError):
ml.delete_model(MODEL_ID_1, app)
def test_delete_model_validation_errors(self, model_id, exc_type):
with pytest.raises(exc_type) as excinfo:
ml.delete_model(model_id)
check_error(excinfo, exc_type)
