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

Improve switch template tests, remove common test helpers (#41322)

Browse source
This commit is contained in:
Franck Nijhof 2020-10-06 14:54:39 +02:00 • committed by GitHub
parent ae54c21945
commit 357a0e9d7e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 33 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

43
tests/components/template/test_switch.py
View file

@ -3,7 +3,15 @@
import pytest
from homeassistant import setup
from homeassistant.const import STATE_OFF, STATE_ON, STATE_UNAVAILABLE
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
from homeassistant.const import (
ATTR_ENTITY_ID,
SERVICE_TURN_OFF,
SERVICE_TURN_ON,
STATE_OFF,
STATE_ON,
STATE_UNAVAILABLE,
)
from homeassistant.core import CoreState, State
from homeassistant.setup import async_setup_component
@ -13,7 +21,6 @@ from tests.common import (
mock_component,
mock_restore_cache,
)
from tests.components.switch import common
@pytest.fixture
@ -418,8 +425,12 @@ async def test_on_action(hass, calls):
state = hass.states.get("switch.test_template_switch")
assert state.state == STATE_OFF
await common.async_turn_on(hass, "switch.test_template_switch")
await hass.async_block_till_done()
await hass.services.async_call(
SWITCH_DOMAIN,
SERVICE_TURN_ON,
{ATTR_ENTITY_ID: "switch.test_template_switch"},
blocking=True,
)
assert len(calls) == 1
@ -454,8 +465,12 @@ async def test_on_action_optimistic(hass, calls):
state = hass.states.get("switch.test_template_switch")
assert state.state == STATE_OFF
await common.async_turn_on(hass, "switch.test_template_switch")
await hass.async_block_till_done()
await hass.services.async_call(
SWITCH_DOMAIN,
SERVICE_TURN_ON,
{ATTR_ENTITY_ID: "switch.test_template_switch"},
blocking=True,
)
state = hass.states.get("switch.test_template_switch")
assert len(calls) == 1
@ -494,8 +509,12 @@ async def test_off_action(hass, calls):
state = hass.states.get("switch.test_template_switch")
assert state.state == STATE_ON
await common.async_turn_off(hass, "switch.test_template_switch")
await hass.async_block_till_done()
await hass.services.async_call(
SWITCH_DOMAIN,
SERVICE_TURN_OFF,
{ATTR_ENTITY_ID: "switch.test_template_switch"},
blocking=True,
)
assert len(calls) == 1
@ -530,8 +549,12 @@ async def test_off_action_optimistic(hass, calls):
state = hass.states.get("switch.test_template_switch")
assert state.state == STATE_ON
await common.async_turn_off(hass, "switch.test_template_switch")
await hass.async_block_till_done()
await hass.services.async_call(
SWITCH_DOMAIN,
SERVICE_TURN_OFF,
{ATTR_ENTITY_ID: "switch.test_template_switch"},
blocking=True,
)
state = hass.states.get("switch.test_template_switch")
assert len(calls) == 1