Add invalid API key test

tenstorrent · Dec 31, 2024 · f12988d · f12988d
1 parent 297cdec
commit f12988d
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tt-metal-yolov4/tests/test_inference_api.py b/tt-metal-yolov4/tests/test_inference_api.py
@@ -27,6 +27,17 @@ def test_valid_api_call():
     assert isinstance(response.json(), list)
 
 
+def test_invalid_api_call():
+    # get sample image file
+    file = sample_file()
+    # make request with INVALID auth header
+    headers = get_auth_header()
+    headers.update(Authorization="INVALID API KEY")
+    response = requests.post(API_URL, files=file, headers=headers)
+    # assert request was unauthorized
+    assert response.status_code == HTTPStatus.UNAUTHORIZED
+
+
 @pytest.mark.skip(
     reason="Not implemented, see https://github.com/tenstorrent/tt-inference-server/issues/63"
 )