diff --git a/tests/app/job/test_rest.py b/tests/app/job/test_rest.py index f35d2774c..abcd4d8cd 100644 --- a/tests/app/job/test_rest.py +++ b/tests/app/job/test_rest.py @@ -8,8 +8,10 @@ import pytest from freezegun import freeze_time from hypothesis import given from hypothesis import strategies as st +from sqlalchemy.exc import SQLAlchemyError import app.celery.tasks +from app import db from app.dao.templates_dao import dao_update_template from app.enums import ( JobStatus, @@ -89,7 +91,12 @@ def test_fuzz_cancel_job(fuzzed_job_id, fuzzed_service_id, request): path = f"/service/{service_id}/job/{job_id}/cancel" auth_header = create_admin_authorization_header() - response = client.post(path, headers=[auth_header]) + try: + response = client.post(path, headers=[auth_header]) + except SQLAlchemyError: + db.session.rollback() + raise + status = response.status_code # 400 Bad Request, 403 Forbidden, 404 Not Found assert status in (