evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Cleanup modbus test mocks (#115412)

Browse source
This commit is contained in:
jan iversen 2024-04-17 19:10:09 +02:00 • committed by GitHub
parent dcd61ac086
commit 3202743b6c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 96 additions and 137 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/components/modbus/test_light.py
View file

@ -262,7 +262,6 @@ async def test_light_service_turn(
hass: HomeAssistant,
caplog: pytest.LogCaptureFixture,
mock_modbus,
mock_pymodbus_return,
) -> None:
"""Run test for service turn_on/turn_off."""
@ -300,12 +299,6 @@ async def test_light_service_turn(
)
await hass.async_block_till_done()
assert hass.states.get(ENTITY_ID2).state == STATE_UNAVAILABLE
mock_modbus.write_coil.side_effect = ModbusException("fail write_")
await hass.services.async_call(
"light", "turn_off", service_data={"entity_id": ENTITY_ID}
)
await hass.async_block_till_done()
assert hass.states.get(ENTITY_ID).state == STATE_UNAVAILABLE
@pytest.mark.parametrize(
@ -323,13 +316,13 @@ async def test_light_service_turn(
},
],
)
async def test_service_light_update(hass: HomeAssistant, mock_modbus, mock_ha) -> None:
async def test_service_light_update(hass: HomeAssistant, mock_modbus_ha) -> None:
"""Run test for service homeassistant.update_entity."""
await hass.services.async_call(
"homeassistant", "update_entity", {"entity_id": ENTITY_ID}, blocking=True
)
assert hass.states.get(ENTITY_ID).state == STATE_OFF
mock_modbus.read_coils.return_value = ReadResult([0x01])
mock_modbus_ha.read_coils.return_value = ReadResult([0x01])
await hass.services.async_call(
"homeassistant", "update_entity", {"entity_id": ENTITY_ID}, blocking=True
)