Offline Caching

api/app/tests/fba/test_fba_endpoint.py

@@ -1,3 +1,4 @@
+import asyncio
 import json
 import math
 from collections import namedtuple
@@ -18,7 +19,14 @@
     TPIClassEnum,
 )
 from wps_shared.db.models.fuel_type_raster import FuelTypeRaster
-from wps_shared.schemas.fba import HfiThreshold
+from wps_shared.schemas.auto_spatial_advisory import SFMSRunType
+from wps_shared.schemas.fba import (
+    FireZoneHFIStats,
+    HFIStatsResponse,
+    HfiThreshold,
+    LatestSFMSRunParameterRangeResponse,
+    SFMSRunParameter,
+)
 from wps_shared.tests.common import default_mock_client_get
 
 mock_fire_centre_name = "PGFireCentre"
@@ -33,11 +41,11 @@
 )
 get_sfms_run_datetimes_url = "/api/fba/sfms-run-datetimes/forecast/2022-09-27"
 get_sfms_run_bounds_url = "/api/fba/sfms-run-bounds"
+get_tpi_stats_url = "api/fba/tpi-stats/forecast/2022-09-27/2022-09-27"
 
 decode_fn = "jwt.decode"
-mock_tpi_stats = AdvisoryTPIStats(
-    id=1, advisory_shape_id=1, valley_bottom=1, mid_slope=2, upper_slope=3, pixel_size_metres=50
-)
+
+mock_tpi_stats_empty = []
 
 mock_fire_centre_info = [(9.0, 11.0, 1, 1, 50, 100, 1)]
 mock_fire_centre_info_with_grass = [(9.0, 11.0, 12, 1, 50, 100, None)]
@@ -73,7 +81,6 @@
     ],
 )
 
-
 def create_mock_centre_tpi_stats(
     advisory_shape_id, source_identifier, valley_bottom, mid_slope, upper_slope, pixel_size_metres
 ):
@@ -90,6 +97,74 @@
 mock_centre_tpi_stats_1 = create_mock_centre_tpi_stats(1, 1, 1, 2, 3, 2)
 mock_centre_tpi_stats_2 = create_mock_centre_tpi_stats(2, 2, 1, 2, 3, 2)
 
+TPIFuelAreasResponse = namedtuple(
+    "TPIFuelAreasResponse",
+    [
+        "tpi_class",
+        "fuel_area",
+        "source_identifier",
+        "id",
+        "name",
+    ],
+)
+
+
+def mock_create_tpi_fuel_area(tpi_class, fuel_area, source_identifier, id, name):
+    return TPIFuelAreasResponse(
+        tpi_class=tpi_class,
+        fuel_area=fuel_area,
+        source_identifier=source_identifier,
+        id=id,
+        name=name,
+    )
+
+
+mock_tpi_fuel_area_1 = mock_create_tpi_fuel_area(
+    TPIClassEnum.valley_bottom, 100, "20", 2, "Coastal"
+)
+mock_tpi_fuel_area_2 = mock_create_tpi_fuel_area(TPIClassEnum.mid_slope, 200, "20", 2, "Coastal")
+mock_tpi_fuel_area_3 = mock_create_tpi_fuel_area(TPIClassEnum.upper_slope, 300, "20", 2, "Coastal")
+
+TPIStatsResponse = namedtuple(
+    "TPIStatsResponse",
+    [
+        "advisory_shape_id",
+        "source_identifier",
+        "valley_bottom",
+        "mid_slope",
+        "upper_slope",
+        "pixel_size_metres",
+        "fire_centre_id",
+        "fire_centre_name",
+    ],
+)
+
+
+def create_mock_tpi_stats(
+    advisory_shape_id,
+    source_identifier,
+    valley_bottom,
+    mid_slope,
+    upper_slope,
+    pixel_size_metres,
+    fire_centre_id,
+    fire_centre_name,
+):
+    return TPIStatsResponse(
+        advisory_shape_id=advisory_shape_id,
+        source_identifier=source_identifier,
+        valley_bottom=valley_bottom,
+        mid_slope=mid_slope,
+        upper_slope=upper_slope,
+        pixel_size_metres=pixel_size_metres,
+        fire_centre_id=fire_centre_id,
+        fire_centre_name=fire_centre_name,
+    )
+
+
+mock_tpi_stats_1 = create_mock_tpi_stats(1, 1, 1, 2, 3, 2, 1, "foo")
+mock_tpi_stats_2 = create_mock_tpi_stats(2, 2, 1, 2, 3, 2, 2, "bar")
+
 
 CentreTPIFuelAreasResponse = namedtuple(
     "CentreTPIFuelAreasResponse", ["tpi_class", "fuel_area", "source_identifier"]
@@ -125,8 +200,8 @@
     return {}
 
 
-async def mock_get_tpi_stats(*_, **__):
-    return mock_tpi_stats
+async def mock_get_tpi_stats_empty(*_, **__):
+    return mock_tpi_stats_empty
 
 
 async def mock_get_tpi_stats_none(*_, **__):
@@ -149,6 +224,10 @@
     return [mock_centre_tpi_stats_1, mock_centre_tpi_stats_2]
 
 
+async def mock_get_tpi_stats(*_, **__):
+    return [mock_tpi_stats_1, mock_tpi_stats_2]
+
+
 async def mock_get_fire_centre_tpi_fuel_areas(*_, **__):
     return [mock_centre_tpi_fuel_area_1, mock_centre_tpi_fuel_area_2, mock_centre_tpi_fuel_area_3]
 
@@ -168,6 +247,32 @@
     return []
 
 
+async def mock_get_most_recent_run_datetime_for_date_range(*_, **__):
+    for_date_1 = date(2025, 8, 25)
+    for_date_2 = date(2025, 8, 26)
+    run_datetime = datetime(2025, 8, 25)
+    run_parameter_1 = SFMSRunParameter(
+        for_date=for_date_1, run_datetime=run_datetime, run_type=SFMSRunType.FORECAST
+    )
+    run_parameter_2 = SFMSRunParameter(
+        for_date=for_date_2, run_datetime=run_datetime, run_type=SFMSRunType.FORECAST
+    )
+    return [run_parameter_1, run_parameter_2]
+
+
+async def mock_get_all_zone_source_ids(*_, **__):
+    return [1, 2, 3]
+
+
+async def mock_get_tpi_fuel_areas(*_, **__):
+    return [mock_tpi_fuel_area_1, mock_centre_tpi_fuel_area_2, mock_tpi_fuel_area_3]
+
+
+async def mock_get_hfi_fuels_data_for_run_parameter(*_, **__):
+    mock_fire_zone_hfi_stats = FireZoneHFIStats(min_wind_stats=[], fuel_area_stats=[])
+    return HFIStatsResponse(zone_data={1: mock_fire_zone_hfi_stats})
+
+
 @pytest.fixture()
 def client():
     from app.main import app as test_app
@@ -205,6 +310,7 @@
         get_fire_centre_info_url,
         get_sfms_run_datetimes_url,
         get_sfms_run_bounds_url,
+        get_tpi_stats_url,
     ],
 )
 def test_get_endpoints_unauthorized(client: TestClient, endpoint: str):
@@ -406,6 +512,32 @@
     assert json_response["firezone_tpi_stats"][1]["mid_slope_tpi"] is None
     assert json_response["firezone_tpi_stats"][1]["upper_slope_tpi"] is None
 
+@pytest.mark.usefixtures("mock_jwt_decode")
+@patch("app.routers.fba.get_auth_header", mock_get_auth_header)
+@patch("app.routers.fba.get_tpi_stats", mock_get_tpi_stats)
+@patch("app.routers.fba.get_fuel_type_raster_by_year", mock_get_fuel_type_raster_by_year)
+@patch("app.routers.fba.get_tpi_fuel_areas", mock_get_tpi_fuel_areas)
+def test_get_tpi_stats_authorized(client: TestClient):
+    """Allowed to get tpi stats for run parameters when authorized"""
+    response = client.get(get_tpi_stats_url)
+
+    json_response = response.json()
+    assert response.status_code == 200
+    assert json_response["firezone_tpi_stats"][0]["fire_zone_id"] == 1
+    assert json_response["firezone_tpi_stats"][0]["valley_bottom_hfi"] == 4
+    assert json_response["firezone_tpi_stats"][0]["valley_bottom_tpi"] is None
+    assert json_response["firezone_tpi_stats"][0]["mid_slope_hfi"] == 8
+    assert math.isclose(json_response["firezone_tpi_stats"][0]["mid_slope_tpi"], 2.0)
+    assert json_response["firezone_tpi_stats"][0]["upper_slope_hfi"] == 12
+    assert json_response["firezone_tpi_stats"][0]["upper_slope_tpi"] is None
+    assert json_response["firezone_tpi_stats"][1]["fire_zone_id"] == 2
+    assert json_response["firezone_tpi_stats"][1]["valley_bottom_hfi"] == 4
+    assert json_response["firezone_tpi_stats"][1]["valley_bottom_tpi"] is None
+    assert json_response["firezone_tpi_stats"][1]["mid_slope_hfi"] == 8
+    assert json_response["firezone_tpi_stats"][1]["mid_slope_tpi"] is None
+    assert json_response["firezone_tpi_stats"][1]["upper_slope_hfi"] == 12
+    assert json_response["firezone_tpi_stats"][1]["upper_slope_tpi"] is None
+
 
 @pytest.mark.usefixtures("mock_jwt_decode")
 @patch("app.routers.fba.get_auth_header", mock_get_auth_header)
@@ -440,6 +572,9 @@
     "/api/fba/fire-centre-tpi-stats/forecast/2024-08-10/2024-08-10/PGFireCentre",
     "/api/fba/sfms-run-datetimes/forecast/2022-09-27",
     "/api/fba/sfms-run-bounds",
+    "/api/fba/latest-sfms-run-parameters/2025-08-25/2025-08-26",
+    "/api/fba/hfi-stats/forecast/2025-08-25T15:01:47.340947Z/2025-08-26",
+    "/api/fba/tpi-stats/forecast/2025-08-25T15:01:47.340947Z/2025-08-26",
 ]
 
 
@@ -456,8 +591,19 @@
 @patch("app.routers.fba.get_fuel_type_raster_by_year", mock_get_fuel_type_raster_by_year)
 @patch("app.routers.fba.get_fire_centre_tpi_fuel_areas", mock_get_fire_centre_tpi_fuel_areas)
 @patch("app.routers.fba.get_centre_tpi_stats", mock_get_centre_tpi_stats)
+@patch("app.routers.fba.get_tpi_stats", mock_get_tpi_stats)
 @patch("app.routers.fba.get_run_datetimes", mock_get_sfms_run_datetimes)
 @patch("app.routers.fba.get_sfms_bounds", mock_get_sfms_bounds)
+@patch(
+    "app.routers.fba.get_most_recent_run_datetime_for_date_range",
+    mock_get_most_recent_run_datetime_for_date_range,
+)
+@patch(
+    "app.routers.fba.get_all_zone_source_ids",
+    mock_get_all_zone_source_ids,
+)
+@patch("app.routers.fba.get_tpi_fuel_areas", mock_get_tpi_fuel_areas)
+@patch("app.routers.fba.get_tpi_stats", mock_get_tpi_stats)
 def test_fba_endpoints_allowed_for_test_idir(client, endpoint):
     headers = {"Authorization": "Bearer token"}
     response = client.get(endpoint, headers=headers)