Add integration test for usergroup costs (#96)

* Add output data to run assertion against

* Add integration test

Author: jnywong

tests/conftest.py

@@ -55,6 +55,16 @@ def output_cost_per_user():
     return data
 
 
+@pytest.fixture(scope="function")
+def output_cost_per_group():
+    """
+    Output to assert against query_total_costs_per_group function.
+    """
+    with open("tests/data/test_output_cost_per_group.json") as f:
+        data = json.load(f)
+    return data
+
+
 @pytest.fixture(autouse=True)
 def env_vars(monkeypatch):
     """

tests/data/test_output_cost_per_group.json

@@ -0,0 +1,32 @@
+[
+    {
+        "cost": 9.804900000000002,
+        "date": "2025-09-01",
+        "usergroup": "none"
+    },
+    {
+        "cost": 1.9541,
+        "date": "2025-09-01",
+        "usergroup": "group_1"
+    },
+    {
+        "cost": 4.3109,
+        "date": "2025-09-01",
+        "usergroup": "group_2"
+    },
+    {
+        "cost": 14.564999999999998,
+        "date": "2025-09-02",
+        "usergroup": "none"
+    },
+    {
+        "cost": 3.8275,
+        "date": "2025-09-02",
+        "usergroup": "group_1"
+    },
+    {
+        "cost": 2.6275,
+        "date": "2025-09-02",
+        "usergroup": "group_2"
+    }
+]

tests/test_integration.py

@@ -120,6 +120,32 @@ def test_costs_per_user(
             assert r["value"] == lookup[key]
 
 
+@pytest.mark.parametrize("mock_prometheus_usage", [None], indirect=True)
+def test_costs_per_group(
+    mock_prometheus_usage,
+    mock_prometheus_usage_share,
+    mock_prometheus_user_group_info,
+    mock_ce,
+    output_cost_per_group,
+):
+    """
+    Test cost logic for costs per group.
+    """
+    from src.jupyterhub_cost_monitoring.query_cost_aws import (
+        query_total_costs_per_group,
+    )
+
+    result = query_total_costs_per_group(date_range)
+    logger.info(f"Cost per group: {result}")
+
+    lookup = {(o["date"], o["usergroup"]): o["cost"] for o in output_cost_per_group}
+
+    for r in result:
+        key = (r["date"], r["usergroup"])
+        if key in lookup:
+            assert r["cost"] == lookup[key]
+
+
 @pytest.mark.parametrize("mock_prometheus_usage", [None], indirect=True)
 def test_costs_per_user_limit(
     mock_ce,