From 10f732b3903f1785e5045c3523d40a943421b93a Mon Sep 17 00:00:00 2001 From: Andrew Shumway Date: Thu, 1 Feb 2024 11:37:31 -0700 Subject: [PATCH] Bundle stats with service object --- app/__init__.py | 4 ++++ tests/app/main/views/test_api_integration.py | 3 +++ tests/conftest.py | 2 ++ 3 files changed, 9 insertions(+) diff --git a/app/__init__.py b/app/__init__.py index af9b3f9a9..9879431e2 100644 --- a/app/__init__.py +++ b/app/__init__.py @@ -381,6 +381,10 @@ def load_service_before_request(): request_ctx.service = Service( service_api_client.get_service(service_id)["data"] ) + stats = service_api_client.get_service_statistics( + service_id, limit_days=7 + ) + request_ctx.service.stats = stats except HTTPError as exc: # if service id isn't real, then 404 rather than 500ing later because we expect service to be set if exc.status_code == 404: diff --git a/tests/app/main/views/test_api_integration.py b/tests/app/main/views/test_api_integration.py index 3b326d55f..c1ca57f22 100644 --- a/tests/app/main/views/test_api_integration.py +++ b/tests/app/main/views/test_api_integration.py @@ -325,6 +325,7 @@ def test_route_permissions( def _get(mocker): return {"count": 0} + mocker.patch("app.service_api_client.get_service_statistics") mocker.patch( "app.service_api_client.get_global_notification_count", side_effect=_get ) @@ -358,6 +359,8 @@ def test_route_invalid_permissions( def _get(mocker): return {"count": 0} + mocker.patch("app.service_api_client.get_service_statistics") + mocker.patch( "app.service_api_client.get_global_notification_count", side_effect=_get ) diff --git a/tests/conftest.py b/tests/conftest.py index ec4ff8ad1..122b76826 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -2319,6 +2319,8 @@ def client_request(logged_in_client, mocker, service_one): # noqa (C901 too com def _get(mocker): return {"count": 0} + mocker.patch("app.service_api_client.get_service_statistics") + mocker.patch( "app.service_api_client.get_global_notification_count", side_effect=_get )