File size: 2,715 Bytes
cd0c7a9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 | """
Smoke tests for the MD Simulation endpoints.
Covers:
- POST /api/md/run creates a job (auth required)
- GET /api/md/status/{job_id} returns status (auth required)
- Invalid PDB ID returns 422
- Missing auth returns 401
- Status for nonexistent job returns 404/422
"""
import pytest
BASE_RUN = "/api/md/run"
BASE_STATUS = "/api/md/status"
class TestMDAuth:
"""Verify auth enforcement."""
def test_no_auth_returns_401(self, client):
resp = client.post(BASE_RUN, json={"pdb_id": "1crn", "mode": "minimize"})
assert resp.status_code == 401
def test_invalid_token_returns_401(self, client):
resp = client.post(
BASE_RUN,
json={"pdb_id": "1crn", "mode": "minimize"},
headers={"Authorization": "Bearer garbage.token.here"},
)
assert resp.status_code == 401
class TestMDRun:
"""POST /api/md/run validation."""
def test_valid_request_returns_job(self, client, auth_headers):
resp = client.post(BASE_RUN, json={"pdb_id": "1crn", "mode": "minimize"}, headers=auth_headers)
assert resp.status_code == 200
body = resp.json()
assert "job_id" in body
assert body["status"] in ("queued", "submitted", "running")
def test_invalid_pdb_id_too_short(self, client, auth_headers):
resp = client.post(BASE_RUN, json={"pdb_id": "1cr", "mode": "minimize"}, headers=auth_headers)
assert resp.status_code == 422
def test_invalid_pdb_id_special_chars(self, client, auth_headers):
resp = client.post(BASE_RUN, json={"pdb_id": "1!@#", "mode": "minimize"}, headers=auth_headers)
assert resp.status_code == 422
def test_invalid_mode_returns_422(self, client, auth_headers):
resp = client.post(BASE_RUN, json={"pdb_id": "1crn", "mode": "invalid_mode"}, headers=auth_headers)
assert resp.status_code == 422
def test_missing_pdb_id_returns_422(self, client, auth_headers):
resp = client.post(BASE_RUN, json={"mode": "minimize"}, headers=auth_headers)
assert resp.status_code == 422
def test_default_mode_is_minimize(self, client, auth_headers):
resp = client.post(BASE_RUN, json={"pdb_id": "1crn"}, headers=auth_headers)
assert resp.status_code == 200
class TestMDStatus:
"""GET /api/md/status/{job_id} validation."""
def test_no_auth_returns_401(self, client):
resp = client.get(f"{BASE_STATUS}/fake-job-id")
assert resp.status_code == 401
def test_nonexistent_job_returns_error(self, client, auth_headers):
resp = client.get(f"{BASE_STATUS}/00000000-0000-0000-0000-000000000000", headers=auth_headers)
assert resp.status_code in (404, 422, 500)
|