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

Add type hints to integration tests (part 20) (#88179)

Browse source
This commit is contained in:
epenet 2023-02-15 15:23:34 +01:00 • committed by GitHub
parent 793e51ac76
commit 3a9adacdde
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
55 changed files with 537 additions and 292 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/components/sleepiq/test_button.py
View file

@ -1,12 +1,13 @@
"""The tests for SleepIQ binary sensor platform."""
from homeassistant.components.button import DOMAIN
from homeassistant.const import ATTR_ENTITY_ID, ATTR_FRIENDLY_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
from .conftest import BED_ID, BED_NAME, BED_NAME_LOWER, setup_platform
async def test_button_calibrate(hass, mock_asyncsleepiq):
async def test_button_calibrate(hass: HomeAssistant, mock_asyncsleepiq) -> None:
"""Test the SleepIQ calibrate button."""
await setup_platform(hass, DOMAIN)
entity_registry = er.async_get(hass)
@ -31,7 +32,7 @@ async def test_button_calibrate(hass, mock_asyncsleepiq):
mock_asyncsleepiq.beds[BED_ID].calibrate.assert_called_once()
async def test_button_stop_pump(hass, mock_asyncsleepiq):
async def test_button_stop_pump(hass: HomeAssistant, mock_asyncsleepiq) -> None:
"""Test the SleepIQ stop pump button."""
await setup_platform(hass, DOMAIN)
entity_registry = er.async_get(hass)