jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 11:17:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Parametrize Switcher button tests (#82849)

Browse Source
This commit is contained in:
Shay Levy 2022-11-28 15:57:52 +02:00 committed by GitHub
parent 15f95a0787
commit 4c3d481b7b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 22 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
tests/components/switcher_kis/test_button.py
View File

@ -20,8 +20,15 @@ SWING_ON_EID = BASE_ENTITY_ID + "_vertical_swing_on"
SWING_OFF_EID = BASE_ENTITY_ID + "_vertical_swing_off" SWING_OFF_EID = BASE_ENTITY_ID + "_vertical_swing_off"
@pytest.mark.parametrize(
"entity, state",
[
(ASSUME_ON_EID, DeviceState.ON),
(ASSUME_OFF_EID, DeviceState.OFF),
],
)
@pytest.mark.parametrize("mock_bridge", [[DEVICE]], indirect=True) @pytest.mark.parametrize("mock_bridge", [[DEVICE]], indirect=True)
async def test_assume_button(hass: HomeAssistant, mock_bridge, mock_api): async def test_assume_button(hass: HomeAssistant, entity, state, mock_bridge, mock_api):
"""Test assume on/off button.""" """Test assume on/off button."""
await init_integration(hass) await init_integration(hass)
assert mock_bridge assert mock_bridge
@ -37,29 +44,24 @@ async def test_assume_button(hass: HomeAssistant, mock_bridge, mock_api):
await hass.services.async_call( await hass.services.async_call(
BUTTON_DOMAIN, BUTTON_DOMAIN,
SERVICE_PRESS, SERVICE_PRESS,
{ATTR_ENTITY_ID: ASSUME_ON_EID}, {ATTR_ENTITY_ID: entity},
blocking=True, blocking=True,
) )
assert mock_api.call_count == 2 assert mock_api.call_count == 2
mock_control_device.assert_called_once_with( mock_control_device.assert_called_once_with(ANY, state=state, update_state=True)
ANY, state=DeviceState.ON, update_state=True
)
mock_control_device.reset_mock()
await hass.services.async_call(
BUTTON_DOMAIN,
SERVICE_PRESS,
{ATTR_ENTITY_ID: ASSUME_OFF_EID},
blocking=True,
)
assert mock_api.call_count == 4
mock_control_device.assert_called_once_with(
ANY, state=DeviceState.OFF, update_state=True
)
@pytest.mark.parametrize(
"entity, swing",
[
(SWING_ON_EID, ThermostatSwing.ON),
(SWING_OFF_EID, ThermostatSwing.OFF),
],
)
@pytest.mark.parametrize("mock_bridge", [[DEVICE]], indirect=True) @pytest.mark.parametrize("mock_bridge", [[DEVICE]], indirect=True)
async def test_swing_button(hass: HomeAssistant, mock_bridge, mock_api, monkeypatch): async def test_swing_button(
hass: HomeAssistant, entity, swing, mock_bridge, mock_api, monkeypatch
):
"""Test vertical swing on/off button.""" """Test vertical swing on/off button."""
monkeypatch.setattr(DEVICE, "remote_id", "ELEC7022") monkeypatch.setattr(DEVICE, "remote_id", "ELEC7022")
await init_integration(hass) await init_integration(hass)
@ -76,21 +78,11 @@ async def test_swing_button(hass: HomeAssistant, mock_bridge, mock_api, monkeypa
await hass.services.async_call( await hass.services.async_call(
BUTTON_DOMAIN, BUTTON_DOMAIN,
SERVICE_PRESS, SERVICE_PRESS,
{ATTR_ENTITY_ID: SWING_ON_EID}, {ATTR_ENTITY_ID: entity},
blocking=True, blocking=True,
) )
assert mock_api.call_count == 2 assert mock_api.call_count == 2
mock_control_device.assert_called_once_with(ANY, swing=ThermostatSwing.ON) mock_control_device.assert_called_once_with(ANY, swing=swing)
mock_control_device.reset_mock()
await hass.services.async_call(
BUTTON_DOMAIN,
SERVICE_PRESS,
{ATTR_ENTITY_ID: SWING_OFF_EID},
blocking=True,
)
assert mock_api.call_count == 4
mock_control_device.assert_called_once_with(ANY, swing=ThermostatSwing.OFF)
@pytest.mark.parametrize("mock_bridge", [[DEVICE]], indirect=True) @pytest.mark.parametrize("mock_bridge", [[DEVICE]], indirect=True)