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

Fix incorrect zeroconf type hint in tests (#118465)

Browse source 
* Fix incorrect `mock_async_zeroconf` type hint

* Adjust thread

* One more

* Fix mock_zeroconf also

* Adjust

* Adjust
This commit is contained in:
epenet 2024-05-30 21:57:09 +02:00 • committed by GitHub
parent 1352c4e427
commit a5dc4cb1c7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 40 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/components/thread/test_discovery.py
View file

@ -1,6 +1,6 @@
"""Test the thread websocket API."""
from unittest.mock import ANY, AsyncMock, Mock
from unittest.mock import ANY, AsyncMock, MagicMock, Mock
import pytest
from zeroconf.asyncio import AsyncServiceInfo
@ -24,7 +24,9 @@ from . import (
)
async def test_discover_routers(hass: HomeAssistant, mock_async_zeroconf: None) -> None:
async def test_discover_routers(
hass: HomeAssistant, mock_async_zeroconf: MagicMock
) -> None:
"""Test discovering thread routers."""
mock_async_zeroconf.async_add_service_listener = AsyncMock()
mock_async_zeroconf.async_remove_service_listener = AsyncMock()
@ -151,7 +153,7 @@ async def test_discover_routers(hass: HomeAssistant, mock_async_zeroconf: None)
],
)
async def test_discover_routers_unconfigured(
hass: HomeAssistant, mock_async_zeroconf: None, data, unconfigured
hass: HomeAssistant, mock_async_zeroconf: MagicMock, data, unconfigured
) -> None:
"""Test discovering thread routers and setting the unconfigured flag."""
mock_async_zeroconf.async_add_service_listener = AsyncMock()
@ -197,7 +199,7 @@ async def test_discover_routers_unconfigured(
"data", [ROUTER_DISCOVERY_HASS_BAD_DATA, ROUTER_DISCOVERY_HASS_MISSING_DATA]
)
async def test_discover_routers_bad_or_missing_optional_data(
hass: HomeAssistant, mock_async_zeroconf: None, data
hass: HomeAssistant, mock_async_zeroconf: MagicMock, data
) -> None:
"""Test discovering thread routers with bad or missing vendor mDNS data."""
mock_async_zeroconf.async_add_service_listener = AsyncMock()
@ -247,7 +249,7 @@ async def test_discover_routers_bad_or_missing_optional_data(
],
)
async def test_discover_routers_bad_or_missing_mandatory_data(
hass: HomeAssistant, mock_async_zeroconf: None, service
hass: HomeAssistant, mock_async_zeroconf: MagicMock, service
) -> None:
"""Test discovering thread routers with missing mandatory mDNS data."""
mock_async_zeroconf.async_add_service_listener = AsyncMock()
@ -281,7 +283,7 @@ async def test_discover_routers_bad_or_missing_mandatory_data(
async def test_discover_routers_get_service_info_fails(
hass: HomeAssistant, mock_async_zeroconf: None
hass: HomeAssistant, mock_async_zeroconf: MagicMock
) -> None:
"""Test discovering thread routers with invalid mDNS data."""
mock_async_zeroconf.async_add_service_listener = AsyncMock()
@ -311,7 +313,7 @@ async def test_discover_routers_get_service_info_fails(
async def test_discover_routers_update_unchanged(
hass: HomeAssistant, mock_async_zeroconf: None
hass: HomeAssistant, mock_async_zeroconf: MagicMock
) -> None:
"""Test discovering thread routers with identical mDNS data in update."""
mock_async_zeroconf.async_add_service_listener = AsyncMock()
@ -353,7 +355,7 @@ async def test_discover_routers_update_unchanged(
async def test_discover_routers_stop_twice(
hass: HomeAssistant, mock_async_zeroconf: None
hass: HomeAssistant, mock_async_zeroconf: MagicMock
) -> None:
"""Test discovering thread routers stopping discovery twice."""
mock_async_zeroconf.async_add_service_listener = AsyncMock()