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

Add type hints to integration tests (zha) (#88309)

Browse source
This commit is contained in:
epenet 2023-02-17 18:54:26 +01:00 • committed by GitHub
parent 185cd61cbd
commit 7427d4f323
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
28 changed files with 411 additions and 238 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/components/zha/test_diagnostics.py
View file

@ -1,6 +1,4 @@
"""Tests for the diagnostics data provided by the ESPHome integration."""
from unittest.mock import patch
import pytest
@ -20,6 +18,7 @@ from tests.components.diagnostics import (
get_diagnostics_for_config_entry,
get_diagnostics_for_device,
)
from tests.typing import ClientSessionGenerator
CONFIG_ENTRY_DIAGNOSTICS_KEYS = [
"config",
@ -56,11 +55,11 @@ def zigpy_device(zigpy_device_mock):
async def test_diagnostics_for_config_entry(
hass: HomeAssistant,
hass_client,
hass_client: ClientSessionGenerator,
config_entry,
zha_device_joined,
zigpy_device,
):
) -> None:
"""Test diagnostics for config entry."""
await zha_device_joined(zigpy_device)
diagnostics_data = await get_diagnostics_for_config_entry(
@ -74,11 +73,11 @@ async def test_diagnostics_for_config_entry(
async def test_diagnostics_for_device(
hass: HomeAssistant,
hass_client,
hass_client: ClientSessionGenerator,
config_entry,
zha_device_joined,
zigpy_device,
):
) -> None:
"""Test diagnostics for device."""
zha_device: ZHADevice = await zha_device_joined(zigpy_device)