Add default sensor data for Tesla Wall Connector tests (#145462)

This commit is contained in:
Abílio Costa 2025-05-22 20:20:57 +01:00 committed by GitHub
parent c130a9f31c
commit 228beacca8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 20 additions and 19 deletions

View File

@ -88,7 +88,23 @@ async def create_wall_connector_entry(
def get_vitals_mock() -> Vitals: def get_vitals_mock() -> Vitals:
"""Get mocked vitals object.""" """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: def get_lifetime_mock() -> Lifetime:

View File

@ -23,8 +23,6 @@ async def test_sensors(hass: HomeAssistant) -> None:
] ]
mock_vitals_first_update = get_vitals_mock() 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 = get_vitals_mock()
mock_vitals_second_update.contactor_closed = True mock_vitals_second_update.contactor_closed = True

View File

@ -5,13 +5,13 @@ from tesla_wall_connector.exceptions import WallConnectorConnectionError
from homeassistant.config_entries import ConfigEntryState from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant 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: async def test_init_success(hass: HomeAssistant) -> None:
"""Test setup and that we get the device info, including firmware version.""" """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 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: async def test_load_unload(hass: HomeAssistant) -> None:
"""Config entry can be unloaded.""" """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 assert entry.state is ConfigEntryState.LOADED

View File

@ -59,19 +59,6 @@ async def test_sensors(hass: HomeAssistant) -> None:
] ]
mock_vitals_first_update = get_vitals_mock() 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 = get_vitals_mock()
mock_vitals_second_update.evse_state = 3 mock_vitals_second_update.evse_state = 3