mirror of
https://github.com/azaion/detections.git
synced 2026-04-23 04:16:32 +00:00
Oleksandr Bezdieniezhnykh
86b8f076b7
- Modified the health endpoint to return "None" for AI availability when inference is not initialized, improving clarity on system status. - Enhanced the test documentation to include handling of skipped tests, emphasizing the need for investigation before proceeding. - Updated test assertions to ensure proper execution order and prevent premature engine initialization. - Refactored test cases to streamline performance testing and improve readability, removing unnecessary complexity. These changes aim to enhance the robustness of the health check and improve the overall testing framework.
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
import pytest
|
|
import requests
|
|
|
|
_DETECT_TIMEOUT = 60
|
|
|
|
|
|
def test_nft_res_01_loader_outage_after_init(
|
|
warm_engine, http_client, mock_loader_url, image_small
|
|
):
|
|
requests.post(
|
|
f"{mock_loader_url}/mock/config", json={"mode": "error"}, timeout=10
|
|
).raise_for_status()
|
|
files = {"file": ("r1.jpg", image_small, "image/jpeg")}
|
|
r = http_client.post("/detect", files=files, timeout=_DETECT_TIMEOUT)
|
|
assert r.status_code == 200
|
|
assert isinstance(r.json(), list)
|
|
h = http_client.get("/health")
|
|
assert h.status_code == 200
|
|
hd = h.json()
|
|
assert hd["status"] == "healthy"
|
|
assert hd.get("errorMessage") is None
|
|
|
|
|
|
def test_nft_res_03_transient_loader_first_fail(
|
|
mock_loader_url, http_client, image_small
|
|
):
|
|
requests.post(
|
|
f"{mock_loader_url}/mock/config", json={"mode": "first_fail"}, timeout=10
|
|
).raise_for_status()
|
|
files = {"file": ("r3a.jpg", image_small, "image/jpeg")}
|
|
r1 = http_client.post("/detect", files=files, timeout=_DETECT_TIMEOUT)
|
|
files2 = {"file": ("r3b.jpg", image_small, "image/jpeg")}
|
|
r2 = http_client.post("/detect", files=files2, timeout=_DETECT_TIMEOUT)
|
|
assert r2.status_code == 200
|
|
if r1.status_code != 200:
|
|
assert r1.status_code != 500
|
|
|
|
|