jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-21 12:17:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix assertions in zeroconf tests (#96957)

Browse Source
This commit is contained in:
J. Nick Koston 2023-07-20 13:36:46 -05:00 committed by GitHub
parent 2a13515759
commit e9620c62b8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 78 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

81
tests/components/zeroconf/test_init.py
View File

@ -12,7 +12,6 @@ from zeroconf import (
from zeroconf.asyncio import AsyncServiceInfo from zeroconf.asyncio import AsyncServiceInfo
from homeassistant.components import zeroconf from homeassistant.components import zeroconf
from homeassistant.components.zeroconf import CONF_DEFAULT_INTERFACE, CONF_IPV6
from homeassistant.const import ( from homeassistant.const import (
EVENT_COMPONENT_LOADED, EVENT_COMPONENT_LOADED,
EVENT_HOMEASSISTANT_START, EVENT_HOMEASSISTANT_START,
@ -231,82 +230,10 @@ async def test_setup_with_overly_long_url_and_name(
assert "German Umlaut" in caplog.text assert "German Umlaut" in caplog.text
async def test_setup_with_default_interface( async def test_setup_with_defaults(
hass: HomeAssistant, mock_async_zeroconf: None hass: HomeAssistant, mock_zeroconf: None, mock_async_zeroconf: None
) -> None: ) -> None:
"""Test default interface config.""" """Test default interface config."""
with patch.object(hass.config_entries.flow, "async_init"), patch.object(
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
), patch(
"homeassistant.components.zeroconf.AsyncServiceInfo",
side_effect=get_service_info_mock,
):
assert await async_setup_component(
hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {CONF_DEFAULT_INTERFACE: True}}
)
hass.bus.async_fire(EVENT_HOMEASSISTANT_STARTED)
await hass.async_block_till_done()
assert mock_async_zeroconf.called_with(interface_choice=InterfaceChoice.Default)
async def test_setup_without_default_interface(
hass: HomeAssistant, mock_async_zeroconf: None
) -> None:
"""Test without default interface config."""
with patch.object(hass.config_entries.flow, "async_init"), patch.object(
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
), patch(
"homeassistant.components.zeroconf.AsyncServiceInfo",
side_effect=get_service_info_mock,
):
assert await async_setup_component(
hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {CONF_DEFAULT_INTERFACE: False}}
)
assert mock_async_zeroconf.called_with()
async def test_setup_without_ipv6(
hass: HomeAssistant, mock_async_zeroconf: None
) -> None:
"""Test without ipv6."""
with patch.object(hass.config_entries.flow, "async_init"), patch.object(
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
), patch(
"homeassistant.components.zeroconf.AsyncServiceInfo",
side_effect=get_service_info_mock,
):
assert await async_setup_component(
hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {CONF_IPV6: False}}
)
hass.bus.async_fire(EVENT_HOMEASSISTANT_STARTED)
await hass.async_block_till_done()
assert mock_async_zeroconf.called_with(ip_version=IPVersion.V4Only)
async def test_setup_with_ipv6(hass: HomeAssistant, mock_async_zeroconf: None) -> None:
"""Test without ipv6."""
with patch.object(hass.config_entries.flow, "async_init"), patch.object(
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
), patch(
"homeassistant.components.zeroconf.AsyncServiceInfo",
side_effect=get_service_info_mock,
):
assert await async_setup_component(
hass, zeroconf.DOMAIN, {zeroconf.DOMAIN: {CONF_IPV6: True}}
)
hass.bus.async_fire(EVENT_HOMEASSISTANT_STARTED)
await hass.async_block_till_done()
assert mock_async_zeroconf.called_with()
async def test_setup_with_ipv6_default(
hass: HomeAssistant, mock_async_zeroconf: None
) -> None:
"""Test without ipv6 as default."""
with patch.object(hass.config_entries.flow, "async_init"), patch.object( with patch.object(hass.config_entries.flow, "async_init"), patch.object(
zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock zeroconf, "HaAsyncServiceBrowser", side_effect=service_update_mock
), patch( ), patch(
@ -317,7 +244,9 @@ async def test_setup_with_ipv6_default(
hass.bus.async_fire(EVENT_HOMEASSISTANT_STARTED) hass.bus.async_fire(EVENT_HOMEASSISTANT_STARTED)
await hass.async_block_till_done() await hass.async_block_till_done()
assert mock_async_zeroconf.called_with() mock_zeroconf.assert_called_with(
interfaces=InterfaceChoice.Default, ip_version=IPVersion.V4Only
)
async def test_zeroconf_match_macaddress( async def test_zeroconf_match_macaddress(

6
tests/conftest.py
View File

@ -1104,10 +1104,12 @@ def mock_get_source_ip() -> Generator[None, None, None]:
@pytest.fixture @pytest.fixture
def mock_zeroconf() -> Generator[None, None, None]: def mock_zeroconf() -> Generator[None, None, None]:
"""Mock zeroconf.""" """Mock zeroconf."""
with patch("homeassistant.components.zeroconf.HaZeroconf", autospec=True), patch( with patch(
"homeassistant.components.zeroconf.HaZeroconf", autospec=True
) as mock_zc, patch(
"homeassistant.components.zeroconf.HaAsyncServiceBrowser", autospec=True "homeassistant.components.zeroconf.HaAsyncServiceBrowser", autospec=True
): ):
yield yield mock_zc
@pytest.fixture @pytest.fixture