Added delete endpoint and tests.

This commit is contained in:
Nicholas Staples
2016-01-12 10:39:49 +00:00
parent 752a359d3d
commit 31f10c7457
8 changed files with 88 additions and 29 deletions

View File

@@ -1,6 +1,6 @@
import pytest
from app.dao.services_dao import (
save_model_service, get_model_services, DAOException)
save_model_service, get_model_services, DAOException, delete_model_service)
from tests.app.conftest import sample_service as create_sample_service
from app.models import Service
@@ -62,3 +62,9 @@ def test_missing_user_attribute(notify_api, notify_db, notify_db_session):
pytest.fail("DAOException not thrown")
except DAOException as e:
assert "Missing data for required attribute" in str(e)
def test_delete_service(notify_api, notify_db, notify_db_session, sample_service):
assert Service.query.count() == 1
delete_model_service(sample_service)
assert Service.query.count() == 0