Add diagnostics to RainMachine (#64788)

tests/components/rainmachine/conftest.py

@@ -1,4 +1,5 @@
 """Define test fixtures for RainMachine."""
+import json
 from unittest.mock import AsyncMock, patch
 
 import pytest
@@ -7,15 +8,12 @@ from homeassistant.components.rainmachine import DOMAIN
 from homeassistant.const import CONF_IP_ADDRESS, CONF_PASSWORD, CONF_PORT, CONF_SSL
 from homeassistant.setup import async_setup_component
 
-from tests.common import MockConfigEntry
+from tests.common import MockConfigEntry, load_fixture
 
 
 @pytest.fixture(name="client")
-def client_fixture(controller_mac):
+def client_fixture(controller, controller_mac):
     """Define a regenmaschine client."""
-    controller = AsyncMock()
-    controller.name = "My RainMachine"
-    controller.mac = controller_mac
     return AsyncMock(load_local=AsyncMock(), controllers={controller_mac: controller})
 
 
@@ -38,12 +36,68 @@ def config_entry_fixture(hass, config, controller_mac):
     return entry
 
 
+@pytest.fixture(name="controller")
+def controller_fixture(
+    controller_mac,
+    data_programs,
+    data_provision_settings,
+    data_restrictions_current,
+    data_restrictions_universal,
+    data_zones,
+):
+    """Define a regenmaschine controller."""
+    controller = AsyncMock()
+    controller.api_version = "4.5.0"
+    controller.hardware_version = 3
+    controller.name = "My RainMachine"
+    controller.mac = controller_mac
+    controller.software_version = "4.0.925"
+
+    controller.programs.all.return_value = data_programs
+    controller.provisioning.settings.return_value = data_provision_settings
+    controller.restrictions.current.return_value = data_restrictions_current
+    controller.restrictions.universal.return_value = data_restrictions_universal
+    controller.zones.all.return_value = data_zones
+
+    return controller
+
+
 @pytest.fixture(name="controller_mac")
 def controller_mac_fixture():
     """Define a controller MAC address."""
     return "aa:bb:cc:dd:ee:ff"
 
 
+@pytest.fixture(name="data_programs", scope="session")
+def data_programs_fixture():
+    """Define program data."""
+    return json.loads(load_fixture("programs_data.json", "rainmachine"))
+
+
+@pytest.fixture(name="data_provision_settings", scope="session")
+def data_provision_settings_fixture():
+    """Define provisioning settings data."""
+    return json.loads(load_fixture("provision_settings_data.json", "rainmachine"))
+
+
+@pytest.fixture(name="data_restrictions_current", scope="session")
+def data_restrictions_current_fixture():
+    """Define current restrictions settings data."""
+    return json.loads(load_fixture("restrictions_current_data.json", "rainmachine"))
+
+
+@pytest.fixture(name="data_restrictions_universal", scope="session")
+def data_restrictions_universal_fixture():
+    """Define universal restrictions settings data."""
+    return json.loads(load_fixture("restrictions_universal_data.json", "rainmachine"))
+
+
+@pytest.fixture(name="data_zones", scope="session")
+def data_zones_fixture():
+    """Define zone data."""
+    return json.loads(load_fixture("zones_data.json", "rainmachine"))
+
+
 @pytest.fixture(name="setup_rainmachine")
 async def setup_rainmachine_fixture(hass, client, config):
     """Define a fixture to set up RainMachine."""