This commit is contained in:
Kenneth Kehl
2025-09-10 13:59:17 -07:00
parent 353c6e1a1f
commit 50979ae95a

View File

@@ -8,8 +8,10 @@ import pytest
from freezegun import freeze_time from freezegun import freeze_time
from hypothesis import given from hypothesis import given
from hypothesis import strategies as st from hypothesis import strategies as st
from sqlalchemy.exc import SQLAlchemyError
import app.celery.tasks import app.celery.tasks
from app import db
from app.dao.templates_dao import dao_update_template from app.dao.templates_dao import dao_update_template
from app.enums import ( from app.enums import (
JobStatus, 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" path = f"/service/{service_id}/job/{job_id}/cancel"
auth_header = create_admin_authorization_header() 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 status = response.status_code
# 400 Bad Request, 403 Forbidden, 404 Not Found # 400 Bad Request, 403 Forbidden, 404 Not Found
assert status in ( assert status in (