async def test_async_set_speed():
"""Test setting speed."""
with patch(f"{WinixDriver_TypeName}.turn_on"), patch(
f"{WinixDriver_TypeName}.manual"
), patch(f"{WinixDriver_TypeName}.high") as high_speed, patch(
f"{WinixDriver_TypeName}.low"
) as low_speed:
wrapper = build_mock_wrapper()
await wrapper.async_set_speed(AIRFLOW_LOW)
assert high_speed.call_count == 0
assert low_speed.call_count == 1
assert wrapper.is_on
assert not wrapper.is_auto
assert wrapper.is_manual
# Calling again at same speed does nothing
await wrapper.async_set_speed(AIRFLOW_LOW)
assert high_speed.call_count == 0
assert low_speed.call_count == 1
assert wrapper.is_on
assert not wrapper.is_auto
assert wrapper.is_manual
# Setting a different speed
await wrapper.async_set_speed(AIRFLOW_HIGH)
assert high_speed.call_count == 1
assert low_speed.call_count == 1
assert wrapper.is_on
assert not wrapper.is_auto
assert wrapper.is_manual
async def test_async_plasmawave_on_off():
"""Test turning plasmawave on."""
# async_plasmawave does not need the device to be turned on
with patch(f"{WinixDriver_TypeName}.plasmawave_on") as plasmawave_on, patch(
f"{WinixDriver_TypeName}.plasmawave_off"
) as plasmawave_off:
wrapper = build_mock_wrapper()
await wrapper.async_plasmawave_on()
assert plasmawave_on.call_count == 1
assert plasmawave_off.call_count == 0
assert wrapper.is_plasma_on
assert wrapper.get_state().get(ATTR_PLASMA) == ON_VALUE
await wrapper.async_plasmawave_on() # Calling again should not do anything
assert plasmawave_on.call_count == 1
assert plasmawave_off.call_count == 0
await wrapper.async_plasmawave_off() # Turn plasma off
assert not wrapper.is_plasma_on
assert wrapper.get_state().get(ATTR_PLASMA) == OFF_VALUE
assert plasmawave_on.call_count == 1
assert plasmawave_off.call_count == 1
await wrapper.async_plasmawave_off() # Calling again should not do anything
assert plasmawave_on.call_count == 1
assert plasmawave_off.call_count == 1
async def test_async_turn_off():
"""Test turning off."""
with patch(f"{WinixDriver_TypeName}.turn_on") as turn_on, patch(
f"{WinixDriver_TypeName}.turn_off"
) as turn_off:
wrapper = build_mock_wrapper()
assert not wrapper.is_on # initially off
await wrapper.async_ensure_on()
assert wrapper.is_on
assert turn_on.call_count == 1
assert turn_off.call_count == 0
await wrapper.async_ensure_on() # Test turning it on again
assert turn_on.call_count == 1 # Should not do anything
assert turn_off.call_count == 0
await wrapper.async_turn_off() # Turn it off
assert turn_on.call_count == 1
assert turn_off.call_count == 1
await wrapper.async_turn_off() # Test turning it off agian
assert turn_on.call_count == 1
assert turn_off.call_count == 1 # Should not do anything
async def test_async_turn_on():
"""Test turning on."""
wrapper = build_mock_wrapper()
wrapper.async_ensure_on = AsyncMock()
wrapper.async_auto = AsyncMock()
await wrapper.async_turn_on()
assert wrapper.async_ensure_on.call_count == 1
assert wrapper.async_auto.call_count == 1
async def test_async_ensure_on():
"""Test ensuring device is on."""
with patch(f"{WinixDriver_TypeName}.turn_on") as turn_on:
wrapper = build_mock_wrapper()
assert not wrapper.is_on # initially off
await wrapper.async_ensure_on()
assert wrapper.is_on
assert turn_on.call_count == 1
await wrapper.async_ensure_on() # Test turning it on again
assert turn_on.call_count == 1 # Should not do anything
async def test_async_auto():
"""Test setting auto mode."""
# async_auto does not need the device to be turned on
with patch(f"{WinixDriver_TypeName}.auto") as auto:
wrapper = build_mock_wrapper()
await wrapper.async_auto()
assert auto.call_count == 1
assert wrapper.is_auto
assert not wrapper.is_manual
assert not wrapper.is_plasma_on # unchanged
assert not wrapper.is_sleep
assert wrapper.get_state().get(ATTR_AIRFLOW) == AIRFLOW_LOW
await wrapper.async_auto() # Calling again should not do anything
assert auto.call_count == 1
async def test_wrapper_update(
mock_state, is_auto, is_manual, is_on, is_plasma_on, is_sleep
):
"""Tests device wrapper states."""
with patch(
f"{WinixDriver_TypeName}.get_state",
AsyncMock(return_value=mock_state),
) as get_state:
wrapper = build_mock_wrapper()
await wrapper.update()
assert get_state.call_count == 1
assert wrapper.get_state() == mock_state
assert wrapper.is_auto == is_auto
assert wrapper.is_manual == is_manual
assert wrapper.is_on == is_on
assert wrapper.is_plasma_on == is_plasma_on
assert wrapper.is_sleep == is_sleep
async def test_async_sleep():
"""Test setting sleep mode."""
# async_sleep does not need the device to be turned on
with patch(f"{WinixDriver_TypeName}.sleep") as sleep:
wrapper = build_mock_wrapper()
await wrapper.async_sleep()
assert sleep.call_count == 1
assert not wrapper.is_auto
assert not wrapper.is_manual
assert not wrapper.is_plasma_on
assert wrapper.is_sleep
assert wrapper.get_state().get(ATTR_MODE) == MODE_MANUAL
assert wrapper.get_state().get(ATTR_AIRFLOW) == AIRFLOW_SLEEP
await wrapper.async_sleep() # Calling again should not do anything
assert sleep.call_count == 1
async def test_async_set_preset_mode(
preset_mode, sleep, auto, manual, plasmawave_off, plasmawave_on
):
"""Test setting preset mode."""
wrapper = build_mock_wrapper()
wrapper.async_ensure_on = AsyncMock()
wrapper.async_sleep = AsyncMock()
wrapper.async_auto = AsyncMock()
wrapper.async_manual = AsyncMock()
wrapper.async_plasmawave_off = AsyncMock()
wrapper.async_plasmawave_on = AsyncMock()
await wrapper.async_set_preset_mode(preset_mode)
assert wrapper.async_ensure_on.call_count == 1
assert wrapper.async_sleep.call_count == sleep
assert wrapper.async_auto.call_count == auto
assert wrapper.async_manual.call_count == manual
assert wrapper.async_plasmawave_off.call_count == plasmawave_off
assert wrapper.async_plasmawave_on.call_count == plasmawave_on