Refactor tests in GIOS (home-assistant#155756)

Co-authored-by: mik-laj <[email protected]>

Author: mik-laj

tests/components/gios/__init__.py

@@ -1,74 +1,16 @@
 """Tests for GIOS."""
 
-from unittest.mock import patch
-
-from homeassistant.components.gios.const import DOMAIN
 from homeassistant.core import HomeAssistant
 
-from tests.common import (
-    MockConfigEntry,
-    async_load_json_array_fixture,
-    async_load_json_object_fixture,
-)
-
-STATIONS = [
-    {
-        "Identyfikator stacji": 123,
-        "Nazwa stacji": "Test Name 1",
-        "WGS84 φ N": "99.99",
-        "WGS84 λ E": "88.88",
-    },
-    {
-        "Identyfikator stacji": 321,
-        "Nazwa stacji": "Test Name 2",
-        "WGS84 φ N": "77.77",
-        "WGS84 λ E": "66.66",
-    },
-]
-
-
-async def init_integration(
-    hass: HomeAssistant, incomplete_data=False, invalid_indexes=False
-) -> MockConfigEntry:
-    """Set up the GIOS integration in Home Assistant."""
-    entry = MockConfigEntry(
-        domain=DOMAIN,
-        title="Home",
-        unique_id="123",
-        data={"station_id": 123, "name": "Home"},
-        entry_id="86129426118ae32020417a53712d6eef",
-    )
+from tests.common import MockConfigEntry
 
-    indexes = await async_load_json_object_fixture(hass, "indexes.json", DOMAIN)
-    station = await async_load_json_array_fixture(hass, "station.json", DOMAIN)
-    sensors = await async_load_json_object_fixture(hass, "sensors.json", DOMAIN)
-    if incomplete_data:
-        indexes["AqIndex"] = "foo"
-        sensors["pm10"]["Lista danych pomiarowych"][0]["Wartość"] = None
-        sensors["pm10"]["Lista danych pomiarowych"][1]["Wartość"] = None
-    if invalid_indexes:
-        indexes = {}
 
-    with (
-        patch(
-            "homeassistant.components.gios.coordinator.Gios._get_stations",
-            return_value=STATIONS,
-        ),
-        patch(
-            "homeassistant.components.gios.coordinator.Gios._get_station",
-            return_value=station,
-        ),
-        patch(
-            "homeassistant.components.gios.coordinator.Gios._get_all_sensors",
-            return_value=sensors,
-        ),
-        patch(
-            "homeassistant.components.gios.coordinator.Gios._get_indexes",
-            return_value=indexes,
-        ),
-    ):
-        entry.add_to_hass(hass)
-        await hass.config_entries.async_setup(entry.entry_id)
-        await hass.async_block_till_done()
+async def setup_integration(
+    hass: HomeAssistant,
+    mock_config_entry: MockConfigEntry,
+) -> None:
+    """Set up the GIOS integration for testing."""
+    mock_config_entry.add_to_hass(hass)
 
-    return entry
+    await hass.config_entries.async_setup(mock_config_entry.entry_id)
+    await hass.async_block_till_done()

tests/components/gios/conftest.py

@@ -0,0 +1,84 @@
+"""Fixtures for GIOS integration tests."""
+
+from collections.abc import AsyncGenerator
+from unittest.mock import AsyncMock, MagicMock, patch
+
+from gios.model import GiosSensors, GiosStation, Sensor as GiosSensor
+import pytest
+
+from homeassistant.components.gios.const import DOMAIN
+from homeassistant.core import HomeAssistant
+
+from . import setup_integration
+
+from tests.common import MockConfigEntry
+
+
+@pytest.fixture
+def mock_config_entry() -> MockConfigEntry:
+    """Return the default mocked config entry."""
+    return MockConfigEntry(
+        domain=DOMAIN,
+        title="Home",
+        unique_id="123",
+        data={"station_id": 123, "name": "Home"},
+        entry_id="86129426118ae32020417a53712d6eef",
+    )
+
+
+@pytest.fixture
+def mock_gios_sensors() -> GiosSensors:
+    """Return the default mocked gios sensors."""
+    return GiosSensors(
+        aqi=GiosSensor(name="AQI", id=None, index=None, value="good"),
+        c6h6=GiosSensor(name="benzene", id=658, index="very_good", value=0.23789),
+        co=GiosSensor(name="carbon monoxide", id=660, index="good", value=251.874),
+        no=GiosSensor(name="nitrogen monoxide", id=664, index=None, value=5.1),
+        no2=GiosSensor(name="nitrogen dioxide", id=665, index="good", value=7.13411),
+        nox=GiosSensor(name="nitrogen oxides", id=666, index=None, value=5.5),
+        o3=GiosSensor(name="ozone", id=667, index="good", value=95.7768),
+        pm10=GiosSensor(
+            name="particulate matter 10", id=14395, index="good", value=16.8344
+        ),
+        pm25=GiosSensor(name="particulate matter 2.5", id=670, index="good", value=4),
+        so2=GiosSensor(name="sulfur dioxide", id=672, index="very_good", value=4.35478),
+    )
+
+
+@pytest.fixture
+def mock_gios_stations() -> dict[int, GiosStation]:
+    """Return the default mocked gios stations."""
+    return {
+        123: GiosStation(id=123, name="Test Name 1", latitude=99.99, longitude=88.88),
+        321: GiosStation(id=321, name="Test Name 2", latitude=77.77, longitude=66.66),
+    }
+
+
+@pytest.fixture
+async def mock_gios(
+    hass: HomeAssistant,
+    mock_gios_stations: dict[int, GiosStation],
+    mock_gios_sensors: GiosSensors,
+) -> AsyncGenerator[MagicMock]:
+    """Yield a mocked GIOS client."""
+    with (
+        patch("homeassistant.components.gios.Gios", autospec=True) as mock_gios,
+        patch("homeassistant.components.gios.config_flow.Gios", new=mock_gios),
+    ):
+        mock_gios.create = AsyncMock(return_value=mock_gios)
+        mock_gios.async_update = AsyncMock(return_value=mock_gios_sensors)
+        mock_gios.measurement_stations = mock_gios_stations
+        mock_gios.station_id = 123
+        mock_gios.station_name = mock_gios_stations[mock_gios.station_id].name
+
+        yield mock_gios
+
+
+@pytest.fixture
+async def init_integration(
+    hass: HomeAssistant,
+    mock_config_entry: MockConfigEntry,
+    mock_gios: MagicMock,
+) -> None:
+    """Set up the GIOS integration for testing."""
+    await setup_integration(hass, mock_config_entry)