From 228beacca8470e1e676b56356da128bfb60eb68c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ab=C3=ADlio=20Costa?= <abmantis@users.noreply.github.com>
Date: Thu, 22 May 2025 20:20:57 +0100
Subject: [PATCH] Add default sensor data for Tesla Wall Connector tests
 (#145462)

---
 .../tesla_wall_connector/conftest.py           | 18 +++++++++++++++++-
 .../tesla_wall_connector/test_binary_sensor.py |  2 --
 .../tesla_wall_connector/test_init.py          |  6 +++---
 .../tesla_wall_connector/test_sensor.py        | 13 -------------
 4 files changed, 20 insertions(+), 19 deletions(-)

diff --git a/tests/components/tesla_wall_connector/conftest.py b/tests/components/tesla_wall_connector/conftest.py
index e4499d6e308..4cb03f2bb1e 100644
--- a/tests/components/tesla_wall_connector/conftest.py
+++ b/tests/components/tesla_wall_connector/conftest.py
@@ -88,7 +88,23 @@ async def create_wall_connector_entry(
 
 def get_vitals_mock() -> Vitals:
     """Get mocked vitals object."""
-    return MagicMock(auto_spec=Vitals)
+    mock = MagicMock(auto_spec=Vitals)
+    mock.evse_state = 1
+    mock.handle_temp_c = 25.51
+    mock.pcba_temp_c = 30.5
+    mock.mcu_temp_c = 42.0
+    mock.grid_v = 230.15
+    mock.grid_hz = 50.021
+    mock.voltageA_v = 230.1
+    mock.voltageB_v = 231
+    mock.voltageC_v = 232.1
+    mock.currentA_a = 10
+    mock.currentB_a = 11.1
+    mock.currentC_a = 12
+    mock.session_energy_wh = 1234.56
+    mock.contactor_closed = False
+    mock.vehicle_connected = True
+    return mock
 
 
 def get_lifetime_mock() -> Lifetime:
diff --git a/tests/components/tesla_wall_connector/test_binary_sensor.py b/tests/components/tesla_wall_connector/test_binary_sensor.py
index 22100bbb1c1..3990369262d 100644
--- a/tests/components/tesla_wall_connector/test_binary_sensor.py
+++ b/tests/components/tesla_wall_connector/test_binary_sensor.py
@@ -23,8 +23,6 @@ async def test_sensors(hass: HomeAssistant) -> None:
     ]
 
     mock_vitals_first_update = get_vitals_mock()
-    mock_vitals_first_update.contactor_closed = False
-    mock_vitals_first_update.vehicle_connected = True
 
     mock_vitals_second_update = get_vitals_mock()
     mock_vitals_second_update.contactor_closed = True
diff --git a/tests/components/tesla_wall_connector/test_init.py b/tests/components/tesla_wall_connector/test_init.py
index 2b37924b2e4..e16180c328a 100644
--- a/tests/components/tesla_wall_connector/test_init.py
+++ b/tests/components/tesla_wall_connector/test_init.py
@@ -5,13 +5,13 @@ from tesla_wall_connector.exceptions import WallConnectorConnectionError
 from homeassistant.config_entries import ConfigEntryState
 from homeassistant.core import HomeAssistant
 
-from .conftest import create_wall_connector_entry
+from .conftest import create_wall_connector_entry, get_vitals_mock
 
 
 async def test_init_success(hass: HomeAssistant) -> None:
     """Test setup and that we get the device info, including firmware version."""
 
-    entry = await create_wall_connector_entry(hass)
+    entry = await create_wall_connector_entry(hass, vitals_data=get_vitals_mock())
 
     assert entry.state is ConfigEntryState.LOADED
 
@@ -28,7 +28,7 @@ async def test_init_while_offline(hass: HomeAssistant) -> None:
 async def test_load_unload(hass: HomeAssistant) -> None:
     """Config entry can be unloaded."""
 
-    entry = await create_wall_connector_entry(hass)
+    entry = await create_wall_connector_entry(hass, vitals_data=get_vitals_mock())
 
     assert entry.state is ConfigEntryState.LOADED
 
diff --git a/tests/components/tesla_wall_connector/test_sensor.py b/tests/components/tesla_wall_connector/test_sensor.py
index 62eca46c388..c6c93006896 100644
--- a/tests/components/tesla_wall_connector/test_sensor.py
+++ b/tests/components/tesla_wall_connector/test_sensor.py
@@ -59,19 +59,6 @@ async def test_sensors(hass: HomeAssistant) -> None:
     ]
 
     mock_vitals_first_update = get_vitals_mock()
-    mock_vitals_first_update.evse_state = 1
-    mock_vitals_first_update.handle_temp_c = 25.51
-    mock_vitals_first_update.pcba_temp_c = 30.5
-    mock_vitals_first_update.mcu_temp_c = 42.0
-    mock_vitals_first_update.grid_v = 230.15
-    mock_vitals_first_update.grid_hz = 50.021
-    mock_vitals_first_update.voltageA_v = 230.1
-    mock_vitals_first_update.voltageB_v = 231
-    mock_vitals_first_update.voltageC_v = 232.1
-    mock_vitals_first_update.currentA_a = 10
-    mock_vitals_first_update.currentB_a = 11.1
-    mock_vitals_first_update.currentC_a = 12
-    mock_vitals_first_update.session_energy_wh = 1234.56
 
     mock_vitals_second_update = get_vitals_mock()
     mock_vitals_second_update.evse_state = 3