Minor fixes to A. O. Smith integration (#107421)

2024-01-31 05:22:25 -05:00 · 2024-01-31 05:22:25 -05:00 · a3352ce457
commit a3352ce457
parent 60fbb8b698
5 changed files with 131 additions and 34 deletions
--- a/tests/components/aosmith/test_init.py
+++ b/tests/components/aosmith/test_init.py
@ -50,7 +50,14 @@ async def test_config_entry_not_ready_get_energy_use_data_error(
    """Test the config entry not ready when get_energy_use_data fails."""
    mock_config_entry.add_to_hass(hass)

-    get_devices_fixture = [build_device_fixture(False, False, True)]
+    get_devices_fixture = [
+        build_device_fixture(
+            heat_pump=True,
+            mode_pending=False,
+            setpoint_pending=False,
+            has_vacation_mode=True,
+        )
+    ]

    with patch(
        "homeassistant.components.aosmith.config_flow.AOSmithAPIClient.get_devices",