Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-25 05:26:47 +00:00
Code Activity

Add type hints to integration tests (part 4) (#87848)

Browse Source
This commit is contained in:
epenet
2023-02-11 08:26:13 +01:00
committed by GitHub
parent a385a00d08
commit 9f688a564f
52 changed files with 733 additions and 336 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/components/demo/test_switch.py
View File

@@ -8,6 +8,7 @@ from homeassistant.components.switch import (
SERVICE_TURN_ON,
)
from homeassistant.const import ATTR_ENTITY_ID, STATE_OFF, STATE_ON
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component
SWITCH_ENTITY_IDS = ["switch.decorative_lights", "switch.ac"]
@@ -23,7 +24,7 @@ async def setup_comp(hass):
@pytest.mark.parametrize("switch_entity_id", SWITCH_ENTITY_IDS)
async def test_turn_on(hass, switch_entity_id):
async def test_turn_on(hass: HomeAssistant, switch_entity_id) -> None:
"""Test switch turn on method."""
await hass.services.async_call(
SWITCH_DOMAIN,
@@ -47,7 +48,7 @@ async def test_turn_on(hass, switch_entity_id):
@pytest.mark.parametrize("switch_entity_id", SWITCH_ENTITY_IDS)
async def test_turn_off(hass, switch_entity_id):
async def test_turn_off(hass: HomeAssistant, switch_entity_id) -> None:
"""Test switch turn off method."""
await hass.services.async_call(
SWITCH_DOMAIN,
@@ -71,7 +72,9 @@ async def test_turn_off(hass, switch_entity_id):
@pytest.mark.parametrize("switch_entity_id", SWITCH_ENTITY_IDS)
async def test_turn_off_without_entity_id(hass, switch_entity_id):
async def test_turn_off_without_entity_id(
hass: HomeAssistant, switch_entity_id
) -> None:
"""Test switch turn off all switches."""
await hass.services.async_call(
SWITCH_DOMAIN, SERVICE_TURN_ON, {ATTR_ENTITY_ID: "all"}, blocking=True