From f12988d55417b36f78e999cd4eb26936d0413c8c Mon Sep 17 00:00:00 2001 From: bgoelTT Date: Mon, 30 Dec 2024 20:24:44 -0500 Subject: [PATCH] Add invalid API key test --- tt-metal-yolov4/tests/test_inference_api.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/tt-metal-yolov4/tests/test_inference_api.py b/tt-metal-yolov4/tests/test_inference_api.py index d4f630a..d38602f 100644 --- 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" )