Add type hints to integration tests (part 2) (#87789)

* Add type hints to integration tests (part 2)

* typo

* Improve analytics

* Improve automation

* Imrpove bluetooth
This commit is contained in:
epenet 2023-02-10 16:05:26 +01:00 committed by GitHub
parent fa7acb4f0d
commit 0cf5e9fb4a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
54 changed files with 1075 additions and 676 deletions

View file

@ -3,7 +3,6 @@ from datetime import timedelta
from bond_async import Action, DeviceType
from homeassistant import core
from homeassistant.components.cover import (
ATTR_CURRENT_POSITION,
ATTR_POSITION,
@ -22,6 +21,7 @@ from homeassistant.const import (
STATE_OPEN,
STATE_UNKNOWN,
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.entity_registry import EntityRegistry
from homeassistant.util import utcnow
@ -72,7 +72,7 @@ def tilt_shades(name: str):
}
async def test_entity_registry(hass: core.HomeAssistant):
async def test_entity_registry(hass: HomeAssistant) -> None:
"""Tests that the devices are registered in the entity registry."""
await setup_platform(
hass,
@ -87,7 +87,7 @@ async def test_entity_registry(hass: core.HomeAssistant):
assert entity.unique_id == "test-hub-id_test-device-id"
async def test_open_cover(hass: core.HomeAssistant):
async def test_open_cover(hass: HomeAssistant) -> None:
"""Tests that open cover command delegates to API."""
await setup_platform(
hass, COVER_DOMAIN, shades("name-1"), bond_device_id="test-device-id"
@ -105,7 +105,7 @@ async def test_open_cover(hass: core.HomeAssistant):
mock_open.assert_called_once_with("test-device-id", Action.open())
async def test_close_cover(hass: core.HomeAssistant):
async def test_close_cover(hass: HomeAssistant) -> None:
"""Tests that close cover command delegates to API."""
await setup_platform(
hass, COVER_DOMAIN, shades("name-1"), bond_device_id="test-device-id"
@ -123,7 +123,7 @@ async def test_close_cover(hass: core.HomeAssistant):
mock_close.assert_called_once_with("test-device-id", Action.close())
async def test_stop_cover(hass: core.HomeAssistant):
async def test_stop_cover(hass: HomeAssistant) -> None:
"""Tests that stop cover command delegates to API."""
await setup_platform(
hass, COVER_DOMAIN, shades("name-1"), bond_device_id="test-device-id"
@ -141,7 +141,7 @@ async def test_stop_cover(hass: core.HomeAssistant):
mock_hold.assert_called_once_with("test-device-id", Action.hold())
async def test_tilt_open_cover(hass: core.HomeAssistant):
async def test_tilt_open_cover(hass: HomeAssistant) -> None:
"""Tests that tilt open cover command delegates to API."""
await setup_platform(
hass, COVER_DOMAIN, tilt_only_shades("name-1"), bond_device_id="test-device-id"
@ -160,7 +160,7 @@ async def test_tilt_open_cover(hass: core.HomeAssistant):
assert hass.states.get("cover.name_1").state == STATE_UNKNOWN
async def test_tilt_close_cover(hass: core.HomeAssistant):
async def test_tilt_close_cover(hass: HomeAssistant) -> None:
"""Tests that tilt close cover command delegates to API."""
await setup_platform(
hass, COVER_DOMAIN, tilt_only_shades("name-1"), bond_device_id="test-device-id"
@ -179,7 +179,7 @@ async def test_tilt_close_cover(hass: core.HomeAssistant):
assert hass.states.get("cover.name_1").state == STATE_UNKNOWN
async def test_tilt_stop_cover(hass: core.HomeAssistant):
async def test_tilt_stop_cover(hass: HomeAssistant) -> None:
"""Tests that tilt stop cover command delegates to API."""
await setup_platform(
hass,
@ -202,7 +202,7 @@ async def test_tilt_stop_cover(hass: core.HomeAssistant):
assert hass.states.get("cover.name_1").state == STATE_UNKNOWN
async def test_tilt_and_open(hass: core.HomeAssistant):
async def test_tilt_and_open(hass: HomeAssistant) -> None:
"""Tests that supports both tilt and open."""
await setup_platform(
hass,
@ -225,7 +225,7 @@ async def test_tilt_and_open(hass: core.HomeAssistant):
assert hass.states.get("cover.name_1").state == STATE_CLOSED
async def test_update_reports_open_cover(hass: core.HomeAssistant):
async def test_update_reports_open_cover(hass: HomeAssistant) -> None:
"""Tests that update command sets correct state when Bond API reports cover is open."""
await setup_platform(hass, COVER_DOMAIN, shades("name-1"))
@ -236,7 +236,7 @@ async def test_update_reports_open_cover(hass: core.HomeAssistant):
assert hass.states.get("cover.name_1").state == "open"
async def test_update_reports_closed_cover(hass: core.HomeAssistant):
async def test_update_reports_closed_cover(hass: HomeAssistant) -> None:
"""Tests that update command sets correct state when Bond API reports cover is closed."""
await setup_platform(hass, COVER_DOMAIN, shades("name-1"))
@ -247,14 +247,14 @@ async def test_update_reports_closed_cover(hass: core.HomeAssistant):
assert hass.states.get("cover.name_1").state == "closed"
async def test_cover_available(hass: core.HomeAssistant):
async def test_cover_available(hass: HomeAssistant) -> None:
"""Tests that available state is updated based on API errors."""
await help_test_entity_available(
hass, COVER_DOMAIN, shades("name-1"), "cover.name_1"
)
async def test_set_position_cover(hass: core.HomeAssistant):
async def test_set_position_cover(hass: HomeAssistant) -> None:
"""Tests that set position cover command delegates to API."""
await setup_platform(
hass,