From f8e02d929588c8792dadaeb75e6f41f10917370a Mon Sep 17 00:00:00 2001 From: Andrew Kettel Date: Mon, 16 Mar 2026 16:52:33 -0700 Subject: [PATCH] Fixing mypy issues in tests --- tests/test_heartbeat.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/test_heartbeat.py b/tests/test_heartbeat.py index 67c36e1..f8ebe34 100644 --- a/tests/test_heartbeat.py +++ b/tests/test_heartbeat.py @@ -1,20 +1,24 @@ +from typing import Any, Generator + +from flask.testing import FlaskClient import pytest + from src.app import app @pytest.fixture -def client(): +def client() -> Generator[FlaskClient, Any, Any]: app.config["TESTING"] = True with app.test_client() as client: yield client -def test_heartbeat_status_code(client): +def test_heartbeat_status_code(client: FlaskClient) -> None: response = client.get("/heartbeat") assert response.status_code == 200 -def test_heartbeat_payload(client): +def test_heartbeat_payload(client: FlaskClient) -> None: response = client.get("/heartbeat") data = response.get_json() assert data["status"] == "ok"