Remove unnecessary class from wemo test_fan (#72377)

pull/72379/head
Eric Severance 2022-05-23 10:02:42 -07:00 committed by GitHub
parent f25663067c
commit e2a80e7a45
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 18 deletions

View File

@ -19,6 +19,7 @@ from homeassistant.const import ATTR_ENTITY_ID, SERVICE_TURN_ON, STATE_OFF, STAT
from homeassistant.setup import async_setup_component
from . import entity_test_helpers
from .conftest import async_create_wemo_entity
@pytest.fixture
@ -161,21 +162,14 @@ async def test_fan_set_percentage(
pywemo_device.set_state.assert_called_with(expected_fan_mode)
class TestInitialFanMode:
"""Test that the FanMode is set to High when turned on the first time."""
@pytest.fixture
def pywemo_device(self, pywemo_device):
"""Set the FanMode to off initially."""
pywemo_device.fan_mode = FanMode.Off
yield pywemo_device
async def test_fan_mode_high_initially(self, hass, pywemo_device, wemo_entity):
"""Verify the FanMode is set to High when turned on."""
assert await hass.services.async_call(
FAN_DOMAIN,
SERVICE_TURN_ON,
{ATTR_ENTITY_ID: [wemo_entity.entity_id]},
blocking=True,
)
pywemo_device.set_state.assert_called_with(FanMode.High)
async def test_fan_mode_high_initially(hass, pywemo_device):
"""Verify the FanMode is set to High when turned on."""
pywemo_device.fan_mode = FanMode.Off
wemo_entity = await async_create_wemo_entity(hass, pywemo_device, "")
assert await hass.services.async_call(
FAN_DOMAIN,
SERVICE_TURN_ON,
{ATTR_ENTITY_ID: [wemo_entity.entity_id]},
blocking=True,
)
pywemo_device.set_state.assert_called_with(FanMode.High)