Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use new ServiceInfo location in component tests (part 4) (#136065)

pull/135220/head^2
epenet 2025-01-20 13:19:17 +01:00 committed by GitHub
parent af40b6524e
commit c5efad3a2d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
14 changed files with 120 additions and 103 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/components/airzone/test_config_flow.py
View File

@ -12,7 +12,6 @@ from aioairzone.exceptions import (
) )
from homeassistant import config_entries from homeassistant import config_entries
from homeassistant.components import dhcp
from homeassistant.components.airzone.config_flow import short_mac from homeassistant.components.airzone.config_flow import short_mac
from homeassistant.components.airzone.const import DOMAIN from homeassistant.components.airzone.const import DOMAIN
from homeassistant.config_entries import SOURCE_USER, ConfigEntryState from homeassistant.config_entries import SOURCE_USER, ConfigEntryState
@ -20,6 +19,7 @@ from homeassistant.const import CONF_HOST, CONF_ID, CONF_PORT
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import FlowResultType from homeassistant.data_entry_flow import FlowResultType
from homeassistant.helpers import device_registry as dr from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.service_info.dhcp import DhcpServiceInfo
from .util import ( from .util import (
CONFIG, CONFIG,
@ -32,7 +32,7 @@ from .util import (
from tests.common import MockConfigEntry from tests.common import MockConfigEntry
DHCP_SERVICE_INFO = dhcp.DhcpServiceInfo( DHCP_SERVICE_INFO = DhcpServiceInfo(
hostname="airzone", hostname="airzone",
ip="192.168.1.100", ip="192.168.1.100",
macaddress=dr.format_mac("E84F25000000").replace(":", ""), macaddress=dr.format_mac("E84F25000000").replace(":", ""),

16
tests/components/androidtv_remote/test_config_flow.py
View File

@ -6,7 +6,6 @@ from unittest.mock import AsyncMock, MagicMock
from androidtvremote2 import CannotConnect, ConnectionClosed, InvalidAuth from androidtvremote2 import CannotConnect, ConnectionClosed, InvalidAuth
from homeassistant import config_entries from homeassistant import config_entries
from homeassistant.components import zeroconf
from homeassistant.components.androidtv_remote.config_flow import ( from homeassistant.components.androidtv_remote.config_flow import (
APPS_NEW_ID, APPS_NEW_ID,
CONF_APP_DELETE, CONF_APP_DELETE,
@ -22,6 +21,7 @@ from homeassistant.components.androidtv_remote.const import (
from homeassistant.config_entries import ConfigEntryState from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import FlowResultType from homeassistant.data_entry_flow import FlowResultType
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
from tests.common import MockConfigEntry from tests.common import MockConfigEntry
@ -444,7 +444,7 @@ async def test_zeroconf_flow_success(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address(host), ip_address=ip_address(host),
ip_addresses=[ip_address(host)], ip_addresses=[ip_address(host)],
port=6466, port=6466,
@ -522,7 +522,7 @@ async def test_zeroconf_flow_cannot_connect(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address(host), ip_address=ip_address(host),
ip_addresses=[ip_address(host)], ip_addresses=[ip_address(host)],
port=6466, port=6466,
@ -573,7 +573,7 @@ async def test_zeroconf_flow_pairing_invalid_auth(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address(host), ip_address=ip_address(host),
ip_addresses=[ip_address(host)], ip_addresses=[ip_address(host)],
port=6466, port=6466,
@ -657,7 +657,7 @@ async def test_zeroconf_flow_already_configured_host_changed_reloads_entry(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address(host), ip_address=ip_address(host),
ip_addresses=[ip_address(host)], ip_addresses=[ip_address(host)],
port=6466, port=6466,
@ -710,7 +710,7 @@ async def test_zeroconf_flow_already_configured_host_not_changed_no_reload_entry
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address(host), ip_address=ip_address(host),
ip_addresses=[ip_address(host)], ip_addresses=[ip_address(host)],
port=6466, port=6466,
@ -743,7 +743,7 @@ async def test_zeroconf_flow_abort_if_mac_is_missing(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address(host), ip_address=ip_address(host),
ip_addresses=[ip_address(host)], ip_addresses=[ip_address(host)],
port=6466, port=6466,
@ -787,7 +787,7 @@ async def test_zeroconf_flow_already_configured_zeroconf_has_multiple_invalid_ip
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("1.2.3.5"), ip_address=ip_address("1.2.3.5"),
ip_addresses=[ip_address("1.2.3.5"), ip_address(host)], ip_addresses=[ip_address("1.2.3.5"), ip_address(host)],
port=6466, port=6466,

32
tests/components/apple_tv/test_config_flow.py
View File

@ -9,7 +9,6 @@ from pyatv.const import PairingRequirement, Protocol
import pytest import pytest
from homeassistant import config_entries from homeassistant import config_entries
from homeassistant.components import zeroconf
from homeassistant.components.apple_tv import CONF_ADDRESS, config_flow from homeassistant.components.apple_tv import CONF_ADDRESS, config_flow
from homeassistant.components.apple_tv.const import ( from homeassistant.components.apple_tv.const import (
CONF_IDENTIFIERS, CONF_IDENTIFIERS,
@ -19,12 +18,13 @@ from homeassistant.components.apple_tv.const import (
from homeassistant.const import CONF_NAME from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import FlowResultType from homeassistant.data_entry_flow import FlowResultType
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
from .common import airplay_service, create_conf, mrp_service, raop_service from .common import airplay_service, create_conf, mrp_service, raop_service
from tests.common import MockConfigEntry from tests.common import MockConfigEntry
DMAP_SERVICE = zeroconf.ZeroconfServiceInfo( DMAP_SERVICE = ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -35,7 +35,7 @@ DMAP_SERVICE = zeroconf.ZeroconfServiceInfo(
) )
RAOP_SERVICE = zeroconf.ZeroconfServiceInfo( RAOP_SERVICE = ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -566,7 +566,7 @@ async def test_zeroconf_unsupported_service_aborts(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -586,7 +586,7 @@ async def test_zeroconf_add_mrp_device(hass: HomeAssistant) -> None:
unrelated_result = await hass.config_entries.flow.async_init( unrelated_result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.2"), ip_address=ip_address("127.0.0.2"),
ip_addresses=[ip_address("127.0.0.2")], ip_addresses=[ip_address("127.0.0.2")],
hostname="mock_hostname", hostname="mock_hostname",
@ -601,7 +601,7 @@ async def test_zeroconf_add_mrp_device(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -883,7 +883,7 @@ async def test_zeroconf_abort_if_other_in_progress(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -906,7 +906,7 @@ async def test_zeroconf_abort_if_other_in_progress(
result2 = await hass.config_entries.flow.async_init( result2 = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -933,7 +933,7 @@ async def test_zeroconf_missing_device_during_protocol_resolve(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -955,7 +955,7 @@ async def test_zeroconf_missing_device_during_protocol_resolve(
await hass.config_entries.flow.async_init( await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -992,7 +992,7 @@ async def test_zeroconf_additional_protocol_resolve_failure(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -1014,7 +1014,7 @@ async def test_zeroconf_additional_protocol_resolve_failure(
await hass.config_entries.flow.async_init( await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -1053,7 +1053,7 @@ async def test_zeroconf_pair_additionally_found_protocols(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -1096,7 +1096,7 @@ async def test_zeroconf_pair_additionally_found_protocols(
await hass.config_entries.flow.async_init( await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -1158,7 +1158,7 @@ async def test_zeroconf_mismatch(hass: HomeAssistant, mock_scan: AsyncMock) -> N
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -1242,7 +1242,7 @@ async def test_zeroconf_rejects_ipv6(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("fd00::b27c:63bb:cc85:4ea0"), ip_address=ip_address("fd00::b27c:63bb:cc85:4ea0"),
ip_addresses=[ip_address("fd00::b27c:63bb:cc85:4ea0")], ip_addresses=[ip_address("fd00::b27c:63bb:cc85:4ea0")],
hostname="mock_hostname", hostname="mock_hostname",

29
tests/components/arcam_fmj/test_config_flow.py
View File

@ -7,12 +7,21 @@ from unittest.mock import AsyncMock, MagicMock, patch
from arcam.fmj.client import ConnectionFailed from arcam.fmj.client import ConnectionFailed
import pytest import pytest
from homeassistant.components import ssdp
from homeassistant.components.arcam_fmj.const import DOMAIN from homeassistant.components.arcam_fmj.const import DOMAIN
from homeassistant.config_entries import SOURCE_SSDP, SOURCE_USER from homeassistant.config_entries import SOURCE_SSDP, SOURCE_USER
from homeassistant.const import CONF_HOST, CONF_PORT, CONF_SOURCE from homeassistant.const import CONF_HOST, CONF_PORT, CONF_SOURCE
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import FlowResultType from homeassistant.data_entry_flow import FlowResultType
from homeassistant.helpers.service_info.ssdp import (
ATTR_UPNP_DEVICE_TYPE,
ATTR_UPNP_FRIENDLY_NAME,
ATTR_UPNP_MANUFACTURER,
ATTR_UPNP_MODEL_NAME,
ATTR_UPNP_MODEL_NUMBER,
ATTR_UPNP_SERIAL,
ATTR_UPNP_UDN,
SsdpServiceInfo,
)
from .conftest import ( from .conftest import (
MOCK_CONFIG_ENTRY, MOCK_CONFIG_ENTRY,
@ -36,18 +45,18 @@ MOCK_UPNP_DEVICE = f"""
MOCK_UPNP_LOCATION = f"http://{MOCK_HOST}:8080/dd.xml" MOCK_UPNP_LOCATION = f"http://{MOCK_HOST}:8080/dd.xml"
MOCK_DISCOVER = ssdp.SsdpServiceInfo( MOCK_DISCOVER = SsdpServiceInfo(
ssdp_usn="mock_usn", ssdp_usn="mock_usn",
ssdp_st="mock_st", ssdp_st="mock_st",
ssdp_location=f"http://{MOCK_HOST}:8080/dd.xml", ssdp_location=f"http://{MOCK_HOST}:8080/dd.xml",
upnp={ upnp={
ssdp.ATTR_UPNP_MANUFACTURER: "ARCAM", ATTR_UPNP_MANUFACTURER: "ARCAM",
ssdp.ATTR_UPNP_MODEL_NAME: " ", ATTR_UPNP_MODEL_NAME: " ",
ssdp.ATTR_UPNP_MODEL_NUMBER: "AVR450, AVR750", ATTR_UPNP_MODEL_NUMBER: "AVR450, AVR750",
ssdp.ATTR_UPNP_FRIENDLY_NAME: f"Arcam media client {MOCK_UUID}", ATTR_UPNP_FRIENDLY_NAME: f"Arcam media client {MOCK_UUID}",
ssdp.ATTR_UPNP_SERIAL: "12343", ATTR_UPNP_SERIAL: "12343",
ssdp.ATTR_UPNP_UDN: MOCK_UDN, ATTR_UPNP_UDN: MOCK_UDN,
ssdp.ATTR_UPNP_DEVICE_TYPE: "urn:schemas-upnp-org:device:MediaRenderer:1", ATTR_UPNP_DEVICE_TYPE: "urn:schemas-upnp-org:device:MediaRenderer:1",
}, },
) )
@ -115,7 +124,7 @@ async def test_ssdp_unable_to_connect(
async def test_ssdp_invalid_id(hass: HomeAssistant) -> None: async def test_ssdp_invalid_id(hass: HomeAssistant) -> None:
"""Test a ssdp with invalid UDN.""" """Test a ssdp with invalid UDN."""
discover = replace( discover = replace(
MOCK_DISCOVER, upnp=MOCK_DISCOVER.upnp | {ssdp.ATTR_UPNP_UDN: "invalid"} MOCK_DISCOVER, upnp=MOCK_DISCOVER.upnp | {ATTR_UPNP_UDN: "invalid"}
) )
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(

4
tests/components/awair/const.py
View File

@ -2,15 +2,15 @@
from ipaddress import ip_address from ipaddress import ip_address
from homeassistant.components import zeroconf
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_HOST from homeassistant.const import CONF_ACCESS_TOKEN, CONF_HOST
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
AWAIR_UUID = "awair_24947" AWAIR_UUID = "awair_24947"
CLOUD_CONFIG = {CONF_ACCESS_TOKEN: "12345"} CLOUD_CONFIG = {CONF_ACCESS_TOKEN: "12345"}
LOCAL_CONFIG = {CONF_HOST: "192.0.2.5"} LOCAL_CONFIG = {CONF_HOST: "192.0.2.5"}
CLOUD_UNIQUE_ID = "foo@bar.com" CLOUD_UNIQUE_ID = "foo@bar.com"
LOCAL_UNIQUE_ID = "00:B0:D0:63:C2:26" LOCAL_UNIQUE_ID = "00:B0:D0:63:C2:26"
ZEROCONF_DISCOVERY = zeroconf.ZeroconfServiceInfo( ZEROCONF_DISCOVERY = ZeroconfServiceInfo(
ip_address=ip_address("192.0.2.5"), ip_address=ip_address("192.0.2.5"),
ip_addresses=[ip_address("192.0.2.5")], ip_addresses=[ip_address("192.0.2.5")],
hostname="mock_hostname", hostname="mock_hostname",

34
tests/components/axis/test_config_flow.py
View File

@ -6,7 +6,6 @@ from unittest.mock import patch
import pytest import pytest
from homeassistant.components import dhcp, ssdp, zeroconf
from homeassistant.components.axis import config_flow from homeassistant.components.axis import config_flow
from homeassistant.components.axis.const import ( from homeassistant.components.axis.const import (
CONF_STREAM_PROFILE, CONF_STREAM_PROFILE,
@ -33,6 +32,9 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import BaseServiceInfo, FlowResultType from homeassistant.data_entry_flow import BaseServiceInfo, FlowResultType
from homeassistant.helpers import device_registry as dr from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.service_info.dhcp import DhcpServiceInfo
from homeassistant.helpers.service_info.ssdp import SsdpServiceInfo
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
from .const import DEFAULT_HOST, MAC, MODEL, NAME from .const import DEFAULT_HOST, MAC, MODEL, NAME
@ -268,7 +270,7 @@ async def test_reconfiguration_flow_update_configuration(
[ [
( (
SOURCE_DHCP, SOURCE_DHCP,
dhcp.DhcpServiceInfo( DhcpServiceInfo(
hostname=f"axis-{MAC}", hostname=f"axis-{MAC}",
ip=DEFAULT_HOST, ip=DEFAULT_HOST,
macaddress=DHCP_FORMATTED_MAC, macaddress=DHCP_FORMATTED_MAC,
@ -276,7 +278,7 @@ async def test_reconfiguration_flow_update_configuration(
), ),
( (
SOURCE_SSDP, SOURCE_SSDP,
ssdp.SsdpServiceInfo( SsdpServiceInfo(
ssdp_usn="mock_usn", ssdp_usn="mock_usn",
ssdp_st="mock_st", ssdp_st="mock_st",
upnp={ upnp={
@ -312,7 +314,7 @@ async def test_reconfiguration_flow_update_configuration(
), ),
( (
SOURCE_ZEROCONF, SOURCE_ZEROCONF,
zeroconf.ZeroconfServiceInfo( ZeroconfServiceInfo(
ip_address=ip_address(DEFAULT_HOST), ip_address=ip_address(DEFAULT_HOST),
ip_addresses=[ip_address(DEFAULT_HOST)], ip_addresses=[ip_address(DEFAULT_HOST)],
port=80, port=80,
@ -376,7 +378,7 @@ async def test_discovery_flow(
[ [
( (
SOURCE_DHCP, SOURCE_DHCP,
dhcp.DhcpServiceInfo( DhcpServiceInfo(
hostname=f"axis-{MAC}", hostname=f"axis-{MAC}",
ip=DEFAULT_HOST, ip=DEFAULT_HOST,
macaddress=DHCP_FORMATTED_MAC, macaddress=DHCP_FORMATTED_MAC,
@ -384,7 +386,7 @@ async def test_discovery_flow(
), ),
( (
SOURCE_SSDP, SOURCE_SSDP,
ssdp.SsdpServiceInfo( SsdpServiceInfo(
ssdp_usn="mock_usn", ssdp_usn="mock_usn",
ssdp_st="mock_st", ssdp_st="mock_st",
upnp={ upnp={
@ -396,7 +398,7 @@ async def test_discovery_flow(
), ),
( (
SOURCE_ZEROCONF, SOURCE_ZEROCONF,
zeroconf.ZeroconfServiceInfo( ZeroconfServiceInfo(
ip_address=ip_address(DEFAULT_HOST), ip_address=ip_address(DEFAULT_HOST),
ip_addresses=[ip_address(DEFAULT_HOST)], ip_addresses=[ip_address(DEFAULT_HOST)],
hostname="mock_hostname", hostname="mock_hostname",
@ -431,7 +433,7 @@ async def test_discovered_device_already_configured(
[ [
( (
SOURCE_DHCP, SOURCE_DHCP,
dhcp.DhcpServiceInfo( DhcpServiceInfo(
hostname=f"axis-{MAC}", hostname=f"axis-{MAC}",
ip="2.3.4.5", ip="2.3.4.5",
macaddress=DHCP_FORMATTED_MAC, macaddress=DHCP_FORMATTED_MAC,
@ -440,7 +442,7 @@ async def test_discovered_device_already_configured(
), ),
( (
SOURCE_SSDP, SOURCE_SSDP,
ssdp.SsdpServiceInfo( SsdpServiceInfo(
ssdp_usn="mock_usn", ssdp_usn="mock_usn",
ssdp_st="mock_st", ssdp_st="mock_st",
upnp={ upnp={
@ -453,7 +455,7 @@ async def test_discovered_device_already_configured(
), ),
( (
SOURCE_ZEROCONF, SOURCE_ZEROCONF,
zeroconf.ZeroconfServiceInfo( ZeroconfServiceInfo(
ip_address=ip_address("2.3.4.5"), ip_address=ip_address("2.3.4.5"),
ip_addresses=[ip_address("2.3.4.5")], ip_addresses=[ip_address("2.3.4.5")],
hostname="mock_hostname", hostname="mock_hostname",
@ -507,7 +509,7 @@ async def test_discovery_flow_updated_configuration(
[ [
( (
SOURCE_DHCP, SOURCE_DHCP,
dhcp.DhcpServiceInfo( DhcpServiceInfo(
hostname="", hostname="",
ip="", ip="",
macaddress=dr.format_mac("01234567890").replace(":", ""), macaddress=dr.format_mac("01234567890").replace(":", ""),
@ -515,7 +517,7 @@ async def test_discovery_flow_updated_configuration(
), ),
( (
SOURCE_SSDP, SOURCE_SSDP,
ssdp.SsdpServiceInfo( SsdpServiceInfo(
ssdp_usn="mock_usn", ssdp_usn="mock_usn",
ssdp_st="mock_st", ssdp_st="mock_st",
upnp={ upnp={
@ -527,7 +529,7 @@ async def test_discovery_flow_updated_configuration(
), ),
( (
SOURCE_ZEROCONF, SOURCE_ZEROCONF,
zeroconf.ZeroconfServiceInfo( ZeroconfServiceInfo(
ip_address=None, ip_address=None,
ip_addresses=[], ip_addresses=[],
hostname="mock_hostname", hostname="mock_hostname",
@ -556,7 +558,7 @@ async def test_discovery_flow_ignore_non_axis_device(
[ [
( (
SOURCE_DHCP, SOURCE_DHCP,
dhcp.DhcpServiceInfo( DhcpServiceInfo(
hostname=f"axis-{MAC}", hostname=f"axis-{MAC}",
ip="169.254.3.4", ip="169.254.3.4",
macaddress=DHCP_FORMATTED_MAC, macaddress=DHCP_FORMATTED_MAC,
@ -564,7 +566,7 @@ async def test_discovery_flow_ignore_non_axis_device(
), ),
( (
SOURCE_SSDP, SOURCE_SSDP,
ssdp.SsdpServiceInfo( SsdpServiceInfo(
ssdp_usn="mock_usn", ssdp_usn="mock_usn",
ssdp_st="mock_st", ssdp_st="mock_st",
upnp={ upnp={
@ -576,7 +578,7 @@ async def test_discovery_flow_ignore_non_axis_device(
), ),
( (
SOURCE_ZEROCONF, SOURCE_ZEROCONF,
zeroconf.ZeroconfServiceInfo( ZeroconfServiceInfo(
ip_address=ip_address("169.254.3.4"), ip_address=ip_address("169.254.3.4"),
ip_addresses=[ip_address("169.254.3.4")], ip_addresses=[ip_address("169.254.3.4")],
hostname="mock_hostname", hostname="mock_hostname",

5
tests/components/axis/test_hub.py
View File

@ -11,13 +11,14 @@ import axis as axislib
import pytest import pytest
from syrupy import SnapshotAssertion from syrupy import SnapshotAssertion
from homeassistant.components import axis, zeroconf from homeassistant.components import axis
from homeassistant.components.axis.const import DOMAIN as AXIS_DOMAIN from homeassistant.components.axis.const import DOMAIN as AXIS_DOMAIN
from homeassistant.components.binary_sensor import DOMAIN as BINARY_SENSOR_DOMAIN from homeassistant.components.binary_sensor import DOMAIN as BINARY_SENSOR_DOMAIN
from homeassistant.config_entries import SOURCE_ZEROCONF, ConfigEntryState from homeassistant.config_entries import SOURCE_ZEROCONF, ConfigEntryState
from homeassistant.const import STATE_OFF, STATE_ON, STATE_UNAVAILABLE from homeassistant.const import STATE_OFF, STATE_ON, STATE_UNAVAILABLE
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
from .conftest import RtspEventMock, RtspStateType from .conftest import RtspEventMock, RtspStateType
from .const import ( from .const import (
@ -93,7 +94,7 @@ async def test_update_address(
mock_requests("2.3.4.5") mock_requests("2.3.4.5")
await hass.config_entries.flow.async_init( await hass.config_entries.flow.async_init(
AXIS_DOMAIN, AXIS_DOMAIN,
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("2.3.4.5"), ip_address=ip_address("2.3.4.5"),
ip_addresses=[ip_address("2.3.4.5")], ip_addresses=[ip_address("2.3.4.5")],
hostname="mock_hostname", hostname="mock_hostname",

10
tests/components/baf/test_config_flow.py
View File

@ -4,11 +4,11 @@ from ipaddress import ip_address
from unittest.mock import patch from unittest.mock import patch
from homeassistant import config_entries from homeassistant import config_entries
from homeassistant.components import zeroconf
from homeassistant.components.baf.const import DOMAIN from homeassistant.components.baf.const import DOMAIN
from homeassistant.const import CONF_IP_ADDRESS from homeassistant.const import CONF_IP_ADDRESS
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import FlowResultType from homeassistant.data_entry_flow import FlowResultType
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
from . import MOCK_NAME, MOCK_UUID, MockBAFDevice from . import MOCK_NAME, MOCK_UUID, MockBAFDevice
@ -90,7 +90,7 @@ async def test_zeroconf_discovery(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -128,7 +128,7 @@ async def test_zeroconf_updates_existing_ip(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -148,7 +148,7 @@ async def test_zeroconf_rejects_ipv6(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("fd00::b27c:63bb:cc85:4ea0"), ip_address=ip_address("fd00::b27c:63bb:cc85:4ea0"),
ip_addresses=[ip_address("fd00::b27c:63bb:cc85:4ea0")], ip_addresses=[ip_address("fd00::b27c:63bb:cc85:4ea0")],
hostname="mock_hostname", hostname="mock_hostname",
@ -167,7 +167,7 @@ async def test_user_flow_is_not_blocked_by_discovery(hass: HomeAssistant) -> Non
discovery_result = await hass.config_entries.flow.async_init( discovery_result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",

10
tests/components/blebox/test_config_flow.py
View File

@ -7,12 +7,12 @@ import blebox_uniapi
import pytest import pytest
from homeassistant import config_entries from homeassistant import config_entries
from homeassistant.components import zeroconf
from homeassistant.components.blebox import config_flow from homeassistant.components.blebox import config_flow
from homeassistant.config_entries import ConfigEntryState from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import CONF_IP_ADDRESS from homeassistant.const import CONF_IP_ADDRESS
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import FlowResultType from homeassistant.data_entry_flow import FlowResultType
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from .conftest import mock_config, mock_feature, mock_only_feature, setup_product_mock from .conftest import mock_config, mock_feature, mock_only_feature, setup_product_mock
@ -227,7 +227,7 @@ async def test_flow_with_zeroconf(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN, config_flow.DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("172.100.123.4"), ip_address=ip_address("172.100.123.4"),
ip_addresses=[ip_address("172.100.123.4")], ip_addresses=[ip_address("172.100.123.4")],
port=80, port=80,
@ -267,7 +267,7 @@ async def test_flow_with_zeroconf_when_already_configured(hass: HomeAssistant) -
result2 = await hass.config_entries.flow.async_init( result2 = await hass.config_entries.flow.async_init(
config_flow.DOMAIN, config_flow.DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("172.100.123.4"), ip_address=ip_address("172.100.123.4"),
ip_addresses=[ip_address("172.100.123.4")], ip_addresses=[ip_address("172.100.123.4")],
port=80, port=80,
@ -291,7 +291,7 @@ async def test_flow_with_zeroconf_when_device_unsupported(hass: HomeAssistant) -
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN, config_flow.DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("172.100.123.4"), ip_address=ip_address("172.100.123.4"),
ip_addresses=[ip_address("172.100.123.4")], ip_addresses=[ip_address("172.100.123.4")],
port=80, port=80,
@ -317,7 +317,7 @@ async def test_flow_with_zeroconf_when_device_response_unsupported(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN, config_flow.DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("172.100.123.4"), ip_address=ip_address("172.100.123.4"),
ip_addresses=[ip_address("172.100.123.4")], ip_addresses=[ip_address("172.100.123.4")],
port=80, port=80,

22
tests/components/bond/test_config_flow.py
View File

@ -10,12 +10,12 @@ from unittest.mock import MagicMock, Mock, patch
from aiohttp import ClientConnectionError, ClientResponseError from aiohttp import ClientConnectionError, ClientResponseError
from homeassistant import config_entries from homeassistant import config_entries
from homeassistant.components import zeroconf
from homeassistant.components.bond.const import DOMAIN from homeassistant.components.bond.const import DOMAIN
from homeassistant.config_entries import ConfigEntryState from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_HOST from homeassistant.const import CONF_ACCESS_TOKEN, CONF_HOST
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import FlowResultType from homeassistant.data_entry_flow import FlowResultType
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
from .common import ( from .common import (
patch_bond_bridge, patch_bond_bridge,
@ -219,7 +219,7 @@ async def test_zeroconf_form(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -260,7 +260,7 @@ async def test_zeroconf_form_token_unavailable(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -302,7 +302,7 @@ async def test_zeroconf_form_token_times_out(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -349,7 +349,7 @@ async def test_zeroconf_form_with_token_available(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -393,7 +393,7 @@ async def test_zeroconf_form_with_token_available_name_unavailable(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",
@ -437,7 +437,7 @@ async def test_zeroconf_already_configured(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.2"), ip_address=ip_address("127.0.0.2"),
ip_addresses=[ip_address("127.0.0.2")], ip_addresses=[ip_address("127.0.0.2")],
hostname="mock_hostname", hostname="mock_hostname",
@ -475,7 +475,7 @@ async def test_zeroconf_in_setup_retry_state(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.2"), ip_address=ip_address("127.0.0.2"),
ip_addresses=[ip_address("127.0.0.2")], ip_addresses=[ip_address("127.0.0.2")],
hostname="mock_hostname", hostname="mock_hostname",
@ -522,7 +522,7 @@ async def test_zeroconf_already_configured_refresh_token(hass: HomeAssistant) ->
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.2"), ip_address=ip_address("127.0.0.2"),
ip_addresses=[ip_address("127.0.0.2")], ip_addresses=[ip_address("127.0.0.2")],
hostname="mock_hostname", hostname="mock_hostname",
@ -561,7 +561,7 @@ async def test_zeroconf_already_configured_no_reload_same_host(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_ZEROCONF}, context={"source": config_entries.SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.3"), ip_address=ip_address("127.0.0.3"),
ip_addresses=[ip_address("127.0.0.3")], ip_addresses=[ip_address("127.0.0.3")],
hostname="mock_hostname", hostname="mock_hostname",
@ -583,7 +583,7 @@ async def test_zeroconf_form_unexpected_error(hass: HomeAssistant) -> None:
await _help_test_form_unexpected_error( await _help_test_form_unexpected_error(
hass, hass,
source=config_entries.SOURCE_ZEROCONF, source=config_entries.SOURCE_ZEROCONF,
initial_input=zeroconf.ZeroconfServiceInfo( initial_input=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="mock_hostname", hostname="mock_hostname",

6
tests/components/bosch_shc/test_config_flow.py
View File

@ -13,11 +13,11 @@ from boschshcpy.information import SHCInformation
import pytest import pytest
from homeassistant import config_entries from homeassistant import config_entries
from homeassistant.components import zeroconf
from homeassistant.components.bosch_shc.config_flow import write_tls_asset from homeassistant.components.bosch_shc.config_flow import write_tls_asset
from homeassistant.components.bosch_shc.const import CONF_SHC_CERT, CONF_SHC_KEY, DOMAIN from homeassistant.components.bosch_shc.const import CONF_SHC_CERT, CONF_SHC_KEY, DOMAIN
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import FlowResultType from homeassistant.data_entry_flow import FlowResultType
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
from tests.common import MockConfigEntry from tests.common import MockConfigEntry
@ -25,7 +25,7 @@ MOCK_SETTINGS = {
"name": "Test name", "name": "Test name",
"device": {"mac": "test-mac", "hostname": "test-host"}, "device": {"mac": "test-mac", "hostname": "test-host"},
} }
DISCOVERY_INFO = zeroconf.ZeroconfServiceInfo( DISCOVERY_INFO = ZeroconfServiceInfo(
ip_address=ip_address("1.1.1.1"), ip_address=ip_address("1.1.1.1"),
ip_addresses=[ip_address("1.1.1.1")], ip_addresses=[ip_address("1.1.1.1")],
hostname="shc012345.local.", hostname="shc012345.local.",
@ -615,7 +615,7 @@ async def test_zeroconf_not_bosch_shc(hass: HomeAssistant) -> None:
"""Test we filter out non-bosch_shc devices.""" """Test we filter out non-bosch_shc devices."""
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("1.1.1.1"), ip_address=ip_address("1.1.1.1"),
ip_addresses=[ip_address("1.1.1.1")], ip_addresses=[ip_address("1.1.1.1")],
hostname="mock_hostname", hostname="mock_hostname",

23
tests/components/braviatv/test_config_flow.py
View File

@ -10,7 +10,6 @@ from pybravia import (
) )
import pytest import pytest
from homeassistant.components import ssdp
from homeassistant.components.braviatv.const import ( from homeassistant.components.braviatv.const import (
CONF_NICKNAME, CONF_NICKNAME,
CONF_USE_PSK, CONF_USE_PSK,
@ -22,6 +21,12 @@ from homeassistant.const import CONF_CLIENT_ID, CONF_HOST, CONF_MAC, CONF_PIN
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import FlowResultType from homeassistant.data_entry_flow import FlowResultType
from homeassistant.helpers import instance_id from homeassistant.helpers import instance_id
from homeassistant.helpers.service_info.ssdp import (
ATTR_UPNP_FRIENDLY_NAME,
ATTR_UPNP_MODEL_NAME,
ATTR_UPNP_UDN,
SsdpServiceInfo,
)
from tests.common import MockConfigEntry from tests.common import MockConfigEntry
@ -46,14 +51,14 @@ BRAVIA_SOURCES = [
{"title": "AV/Component", "uri": "extInput:component?port=1"}, {"title": "AV/Component", "uri": "extInput:component?port=1"},
] ]
BRAVIA_SSDP = ssdp.SsdpServiceInfo( BRAVIA_SSDP = SsdpServiceInfo(
ssdp_usn="mock_usn", ssdp_usn="mock_usn",
ssdp_st="mock_st", ssdp_st="mock_st",
ssdp_location="http://bravia-host:52323/dmr.xml", ssdp_location="http://bravia-host:52323/dmr.xml",
upnp={ upnp={
ssdp.ATTR_UPNP_UDN: "uuid:1234", ATTR_UPNP_UDN: "uuid:1234",
ssdp.ATTR_UPNP_FRIENDLY_NAME: "Living TV", ATTR_UPNP_FRIENDLY_NAME: "Living TV",
ssdp.ATTR_UPNP_MODEL_NAME: "KE-55XH9096", ATTR_UPNP_MODEL_NAME: "KE-55XH9096",
"X_ScalarWebAPI_DeviceInfo": { "X_ScalarWebAPI_DeviceInfo": {
"X_ScalarWebAPI_ServiceList": { "X_ScalarWebAPI_ServiceList": {
"X_ScalarWebAPI_ServiceType": [ "X_ScalarWebAPI_ServiceType": [
@ -68,14 +73,14 @@ BRAVIA_SSDP = ssdp.SsdpServiceInfo(
}, },
) )
FAKE_BRAVIA_SSDP = ssdp.SsdpServiceInfo( FAKE_BRAVIA_SSDP = SsdpServiceInfo(
ssdp_usn="mock_usn", ssdp_usn="mock_usn",
ssdp_st="mock_st", ssdp_st="mock_st",
ssdp_location="http://soundbar-host:52323/dmr.xml", ssdp_location="http://soundbar-host:52323/dmr.xml",
upnp={ upnp={
ssdp.ATTR_UPNP_UDN: "uuid:1234", ATTR_UPNP_UDN: "uuid:1234",
ssdp.ATTR_UPNP_FRIENDLY_NAME: "Sony Audio Device", ATTR_UPNP_FRIENDLY_NAME: "Sony Audio Device",
ssdp.ATTR_UPNP_MODEL_NAME: "HT-S700RF", ATTR_UPNP_MODEL_NAME: "HT-S700RF",
"X_ScalarWebAPI_DeviceInfo": { "X_ScalarWebAPI_DeviceInfo": {
"X_ScalarWebAPI_ServiceList": { "X_ScalarWebAPI_ServiceList": {
"X_ScalarWebAPI_ServiceType": ["guide", "system", "audio", "avContent"], "X_ScalarWebAPI_ServiceType": ["guide", "system", "audio", "avContent"],

16
tests/components/broadlink/test_config_flow.py
View File

@ -8,10 +8,10 @@ import broadlink.exceptions as blke
import pytest import pytest
from homeassistant import config_entries from homeassistant import config_entries
from homeassistant.components import dhcp
from homeassistant.components.broadlink.const import DOMAIN from homeassistant.components.broadlink.const import DOMAIN
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import FlowResultType from homeassistant.data_entry_flow import FlowResultType
from homeassistant.helpers.service_info.dhcp import DhcpServiceInfo
from . import get_device from . import get_device
@ -828,7 +828,7 @@ async def test_dhcp_can_finish(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_DHCP}, context={"source": config_entries.SOURCE_DHCP},
data=dhcp.DhcpServiceInfo( data=DhcpServiceInfo(
hostname="broadlink", hostname="broadlink",
ip="1.2.3.4", ip="1.2.3.4",
macaddress=device.mac, macaddress=device.mac,
@ -862,7 +862,7 @@ async def test_dhcp_fails_to_connect(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_DHCP}, context={"source": config_entries.SOURCE_DHCP},
data=dhcp.DhcpServiceInfo( data=DhcpServiceInfo(
hostname="broadlink", hostname="broadlink",
ip="1.2.3.4", ip="1.2.3.4",
macaddress="34ea34b43b5a", macaddress="34ea34b43b5a",
@ -881,7 +881,7 @@ async def test_dhcp_unreachable(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_DHCP}, context={"source": config_entries.SOURCE_DHCP},
data=dhcp.DhcpServiceInfo( data=DhcpServiceInfo(
hostname="broadlink", hostname="broadlink",
ip="1.2.3.4", ip="1.2.3.4",
macaddress="34ea34b43b5a", macaddress="34ea34b43b5a",
@ -900,7 +900,7 @@ async def test_dhcp_connect_unknown_error(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_DHCP}, context={"source": config_entries.SOURCE_DHCP},
data=dhcp.DhcpServiceInfo( data=DhcpServiceInfo(
hostname="broadlink", hostname="broadlink",
ip="1.2.3.4", ip="1.2.3.4",
macaddress="34ea34b43b5a", macaddress="34ea34b43b5a",
@ -922,7 +922,7 @@ async def test_dhcp_device_not_supported(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_DHCP}, context={"source": config_entries.SOURCE_DHCP},
data=dhcp.DhcpServiceInfo( data=DhcpServiceInfo(
hostname="broadlink", hostname="broadlink",
ip=device.host, ip=device.host,
macaddress=device.mac, macaddress=device.mac,
@ -946,7 +946,7 @@ async def test_dhcp_already_exists(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_DHCP}, context={"source": config_entries.SOURCE_DHCP},
data=dhcp.DhcpServiceInfo( data=DhcpServiceInfo(
hostname="broadlink", hostname="broadlink",
ip="1.2.3.4", ip="1.2.3.4",
macaddress="34ea34b43b5a", macaddress="34ea34b43b5a",
@ -971,7 +971,7 @@ async def test_dhcp_updates_host(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": config_entries.SOURCE_DHCP}, context={"source": config_entries.SOURCE_DHCP},
data=dhcp.DhcpServiceInfo( data=DhcpServiceInfo(
hostname="broadlink", hostname="broadlink",
ip="4.5.6.7", ip="4.5.6.7",
macaddress="34ea34b43b5a", macaddress="34ea34b43b5a",

12
tests/components/brother/test_config_flow.py
View File

@ -6,12 +6,12 @@ from unittest.mock import AsyncMock, patch
from brother import SnmpError, UnsupportedModelError from brother import SnmpError, UnsupportedModelError
import pytest import pytest
from homeassistant.components import zeroconf
from homeassistant.components.brother.const import DOMAIN from homeassistant.components.brother.const import DOMAIN
from homeassistant.config_entries import SOURCE_USER, SOURCE_ZEROCONF from homeassistant.config_entries import SOURCE_USER, SOURCE_ZEROCONF
from homeassistant.const import CONF_HOST, CONF_TYPE from homeassistant.const import CONF_HOST, CONF_TYPE
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import FlowResultType from homeassistant.data_entry_flow import FlowResultType
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
from . import init_integration from . import init_integration
@ -121,7 +121,7 @@ async def test_zeroconf_exception(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": SOURCE_ZEROCONF}, context={"source": SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="example.local.", hostname="example.local.",
@ -145,7 +145,7 @@ async def test_zeroconf_unsupported_model(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": SOURCE_ZEROCONF}, context={"source": SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="example.local.", hostname="example.local.",
@ -171,7 +171,7 @@ async def test_zeroconf_device_exists_abort(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": SOURCE_ZEROCONF}, context={"source": SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="example.local.", hostname="example.local.",
@ -200,7 +200,7 @@ async def test_zeroconf_no_probe_existing_device(hass: HomeAssistant) -> None:
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": SOURCE_ZEROCONF}, context={"source": SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="example.local.", hostname="example.local.",
@ -224,7 +224,7 @@ async def test_zeroconf_confirm_create_entry(
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, DOMAIN,
context={"source": SOURCE_ZEROCONF}, context={"source": SOURCE_ZEROCONF},
data=zeroconf.ZeroconfServiceInfo( data=ZeroconfServiceInfo(
ip_address=ip_address("127.0.0.1"), ip_address=ip_address("127.0.0.1"),
ip_addresses=[ip_address("127.0.0.1")], ip_addresses=[ip_address("127.0.0.1")],
hostname="example.local.", hostname="example.local.",