Update signature of platforms' async_setup_entry in tests (#138271)

This commit is contained in:
Erik Montnemery 2025-02-11 13:17:39 +01:00 committed by GitHub
parent 89f157592d
commit 428cc1a951
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
29 changed files with 75 additions and 69 deletions

View File

@ -86,7 +86,10 @@ from homeassistant.helpers.dispatcher import (
async_dispatcher_send, async_dispatcher_send,
) )
from homeassistant.helpers.entity import Entity from homeassistant.helpers.entity import Entity
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import (
AddConfigEntryEntitiesCallback,
AddEntitiesCallback,
)
from homeassistant.helpers.json import JSONEncoder, _orjson_default_encoder, json_dumps from homeassistant.helpers.json import JSONEncoder, _orjson_default_encoder, json_dumps
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import dt as dt_util, ulid as ulid_util from homeassistant.util import dt as dt_util, ulid as ulid_util
@ -1813,7 +1816,7 @@ def setup_test_component_platform(
async def _async_setup_entry( async def _async_setup_entry(
hass: HomeAssistant, hass: HomeAssistant,
entry: ConfigEntry, entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up a test component platform.""" """Set up a test component platform."""
async_add_entities(entities) async_add_entities(entities)

View File

@ -14,7 +14,7 @@ from homeassistant.components.alarm_control_panel.const import CodeFormat
from homeassistant.config_entries import ConfigEntry, ConfigFlow from homeassistant.config_entries import ConfigEntry, ConfigFlow
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er, frame from homeassistant.helpers import entity_registry as er, frame
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import MockAlarm from .common import MockAlarm
@ -194,7 +194,7 @@ async def setup_alarm_control_panel_platform_test_entity(
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test alarm control panel platform via config entry.""" """Set up test alarm control panel platform via config entry."""
async_add_entities([entity]) async_add_entities([entity])

View File

@ -25,7 +25,7 @@ from homeassistant.config_entries import ConfigEntry, ConfigFlow
from homeassistant.const import Platform from homeassistant.const import Platform
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.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from tests.common import ( from tests.common import (
@ -225,7 +225,7 @@ async def init_supporting_components(
async def async_setup_entry_stt_platform( async def async_setup_entry_stt_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test stt platform via config entry.""" """Set up test stt platform via config entry."""
async_add_entities([mock_stt_provider_entity]) async_add_entities([mock_stt_provider_entity])
@ -233,7 +233,7 @@ async def init_supporting_components(
async def async_setup_entry_wake_word_platform( async def async_setup_entry_wake_word_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test wake word platform via config entry.""" """Set up test wake word platform via config entry."""
async_add_entities( async_add_entities(
@ -325,7 +325,7 @@ async def assist_device(
async def async_setup_entry_select_platform( async def async_setup_entry_select_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test select platform via config entry.""" """Set up test select platform via config entry."""
entities = [ entities = [

View File

@ -19,7 +19,7 @@ from homeassistant.config_entries import ConfigEntry, ConfigEntryState
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.device_registry import DeviceInfo from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from tests.common import MockConfigEntry, MockPlatform, mock_platform from tests.common import MockConfigEntry, MockPlatform, mock_platform
@ -31,7 +31,7 @@ class SelectPlatform(MockPlatform):
self, self,
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up fake select platform.""" """Set up fake select platform."""
pipeline_entity = AssistPipelineSelect(hass, "test-domain", "test-prefix") pipeline_entity = AssistPipelineSelect(hass, "test-domain", "test-prefix")

View File

@ -9,7 +9,7 @@ from homeassistant.components import binary_sensor
from homeassistant.config_entries import ConfigEntry, ConfigFlow from homeassistant.config_entries import ConfigEntry, ConfigFlow
from homeassistant.const import STATE_OFF, STATE_ON, EntityCategory from homeassistant.const import STATE_OFF, STATE_ON, EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .common import MockBinarySensor from .common import MockBinarySensor
@ -102,7 +102,7 @@ async def test_name(hass: HomeAssistant) -> None:
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test binary_sensor platform via config entry.""" """Set up test binary_sensor platform via config entry."""
async_add_entities([entity1, entity2, entity3, entity4]) async_add_entities([entity1, entity2, entity3, entity4])
@ -172,7 +172,7 @@ async def test_entity_category_config_raises_error(
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test binary_sensor platform via config entry.""" """Set up test binary_sensor platform via config entry."""
async_add_entities([entity1, entity2]) async_add_entities([entity1, entity2])

View File

@ -6,7 +6,7 @@ import pytest
from homeassistant.components.button import DOMAIN, ButtonEntity from homeassistant.components.button import DOMAIN, ButtonEntity
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .const import TEST_DOMAIN from .const import TEST_DOMAIN
@ -31,7 +31,7 @@ async def setup_platform(hass: HomeAssistant) -> None:
async def async_setup_platform( async def async_setup_platform(
hass: HomeAssistant, hass: HomeAssistant,
config: ConfigType, config: ConfigType,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
discovery_info: DiscoveryInfoType | None = None, discovery_info: DiscoveryInfoType | None = None,
) -> None: ) -> None:
"""Set up test button platform.""" """Set up test button platform."""

View File

@ -22,7 +22,7 @@ from homeassistant.const import (
STATE_UNKNOWN, STATE_UNKNOWN,
) )
from homeassistant.core import HomeAssistant, State from homeassistant.core import HomeAssistant, State
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from homeassistant.util import dt as dt_util from homeassistant.util import dt as dt_util
@ -175,7 +175,7 @@ async def test_name(hass: HomeAssistant) -> None:
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test button platform via config entry.""" """Set up test button platform via config entry."""
async_add_entities([entity1, entity2, entity3, entity4]) async_add_entities([entity1, entity2, entity3, entity4])

View File

@ -12,7 +12,7 @@ from homeassistant.components.calendar import DOMAIN, CalendarEntity, CalendarEv
from homeassistant.config_entries import ConfigEntry, ConfigFlow from homeassistant.config_entries import ConfigEntry, ConfigFlow
from homeassistant.const import Platform from homeassistant.const import Platform
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util import dt as dt_util from homeassistant.util import dt as dt_util
from tests.common import ( from tests.common import (
@ -145,7 +145,7 @@ def mock_setup_integration(
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test event platform via config entry.""" """Set up test event platform via config entry."""
new_entities = create_test_entities() new_entities = create_test_entities()

View File

@ -42,7 +42,7 @@ from homeassistant.const import ATTR_TEMPERATURE, PRECISION_WHOLE, UnitOfTempera
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers import issue_registry as ir from homeassistant.helpers import issue_registry as ir
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from tests.common import ( from tests.common import (
MockConfigEntry, MockConfigEntry,
@ -582,7 +582,7 @@ async def test_issue_aux_property_deprecated(
async def async_setup_entry_climate_platform( async def async_setup_entry_climate_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test weather platform via config entry.""" """Set up test weather platform via config entry."""
async_add_entities([climate_entity]) async_add_entities([climate_entity])

View File

@ -24,7 +24,7 @@ from homeassistant.helpers import (
floor_registry as fr, floor_registry as fr,
intent, intent,
) )
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from tests.common import ( from tests.common import (
@ -90,7 +90,7 @@ async def create_mock_platform(
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test event platform via config entry.""" """Set up test event platform via config entry."""
async_add_entities(entities) async_add_entities(entities)

View File

@ -35,7 +35,7 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr, entity_registry as er from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_connect from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity import Entity from homeassistant.helpers.entity import Entity
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from tests.common import ( from tests.common import (
MockConfigEntry, MockConfigEntry,
@ -114,7 +114,7 @@ async def create_mock_platform(
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test event platform via config entry.""" """Set up test event platform via config entry."""
async_add_entities(entities) async_add_entities(entities)

View File

@ -17,7 +17,7 @@ from homeassistant.components.event import (
from homeassistant.config_entries import ConfigEntry, ConfigFlow from homeassistant.config_entries import ConfigEntry, ConfigFlow
from homeassistant.const import CONF_PLATFORM, STATE_UNKNOWN from homeassistant.const import CONF_PLATFORM, STATE_UNKNOWN
from homeassistant.core import HomeAssistant, State from homeassistant.core import HomeAssistant, State
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import STORAGE_KEY as RESTORE_STATE_KEY from homeassistant.helpers.restore_state import STORAGE_KEY as RESTORE_STATE_KEY
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from homeassistant.util import dt as dt_util from homeassistant.util import dt as dt_util
@ -297,7 +297,7 @@ async def test_name(hass: HomeAssistant) -> None:
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test event platform via config entry.""" """Set up test event platform via config entry."""
async_add_entities([entity1, entity2, entity3, entity4]) async_add_entities([entity1, entity2, entity3, entity4])

View File

@ -7,7 +7,10 @@ import pytest
from homeassistant.components import image from homeassistant.components import image
from homeassistant.config_entries import ConfigEntry, ConfigFlow from homeassistant.config_entries import ConfigEntry, ConfigFlow
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import (
AddConfigEntryEntitiesCallback,
AddEntitiesCallback,
)
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from homeassistant.util import dt as dt_util from homeassistant.util import dt as dt_util
@ -123,7 +126,7 @@ class MockImageConfigEntry:
self, self,
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test image platform via config entry.""" """Set up test image platform via config entry."""
async_add_entities([self._entities]) async_add_entities([self._entities])

View File

@ -14,7 +14,7 @@ from homeassistant.components.lawn_mower import (
from homeassistant.config_entries import ConfigEntry, ConfigEntryState, ConfigFlow from homeassistant.config_entries import ConfigEntry, ConfigEntryState, ConfigFlow
from homeassistant.const import STATE_UNAVAILABLE, Platform from homeassistant.const import STATE_UNAVAILABLE, Platform
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from tests.common import ( from tests.common import (
MockConfigEntry, MockConfigEntry,
@ -97,7 +97,7 @@ async def test_lawn_mower_setup(hass: HomeAssistant) -> None:
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test platform via config entry.""" """Set up test platform via config entry."""
async_add_entities([entity1]) async_add_entities([entity1])

View File

@ -14,7 +14,7 @@ from homeassistant.components.lock import (
from homeassistant.config_entries import ConfigEntry, ConfigFlow from homeassistant.config_entries import ConfigEntry, ConfigFlow
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from tests.common import ( from tests.common import (
MockConfigEntry, MockConfigEntry,
@ -120,7 +120,7 @@ async def setup_lock_platform_test_entity(
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test lock platform via config entry.""" """Set up test lock platform via config entry."""
async_add_entities([entity]) async_add_entities([entity])

View File

@ -38,7 +38,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, State from homeassistant.core import HomeAssistant, State
from homeassistant.exceptions import ServiceValidationError from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import STORAGE_KEY as RESTORE_STATE_KEY from homeassistant.helpers.restore_state import STORAGE_KEY as RESTORE_STATE_KEY
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from homeassistant.util.unit_system import METRIC_SYSTEM, US_CUSTOMARY_SYSTEM from homeassistant.util.unit_system import METRIC_SYSTEM, US_CUSTOMARY_SYSTEM
@ -974,7 +974,7 @@ async def test_name(hass: HomeAssistant) -> None:
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test number platform via config entry.""" """Set up test number platform via config entry."""
async_add_entities([entity1, entity2, entity3, entity4]) async_add_entities([entity1, entity2, entity3, entity4])

View File

@ -45,7 +45,7 @@ from homeassistant.const import (
) )
from homeassistant.core import HomeAssistant, State from homeassistant.core import HomeAssistant, State
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.restore_state import STORAGE_KEY as RESTORE_STATE_KEY from homeassistant.helpers.restore_state import STORAGE_KEY as RESTORE_STATE_KEY
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from homeassistant.util import dt as dt_util from homeassistant.util import dt as dt_util
@ -2584,7 +2584,7 @@ async def test_name(hass: HomeAssistant) -> None:
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test sensor platform via config entry.""" """Set up test sensor platform via config entry."""
async_add_entities([entity1, entity2, entity3, entity4]) async_add_entities([entity1, entity2, entity3, entity4])

View File

@ -16,7 +16,7 @@ from homeassistant.components.stt import (
) )
from homeassistant.config_entries import ConfigEntry, ConfigEntryState, ConfigFlow from homeassistant.config_entries import ConfigEntry, ConfigEntryState, ConfigFlow
from homeassistant.core import HomeAssistant, State from homeassistant.core import HomeAssistant, State
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from .common import ( from .common import (
@ -145,7 +145,7 @@ async def mock_config_entry_setup(
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test stt platform via config entry.""" """Set up test stt platform via config entry."""
async_add_entities([mock_provider_entity]) async_add_entities([mock_provider_entity])

View File

@ -24,7 +24,7 @@ from homeassistant.const import (
from homeassistant.core import Context, CoreState, HomeAssistant, State, callback from homeassistant.core import Context, CoreState, HomeAssistant, State, callback
from homeassistant.helpers import device_registry as dr, entity_registry as er from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.entity_component import async_update_entity from homeassistant.helpers.entity_component import async_update_entity
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.template import Template from homeassistant.helpers.template import Template
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.setup import ATTR_COMPONENT, async_setup_component from homeassistant.setup import ATTR_COMPONENT, async_setup_component
@ -393,7 +393,7 @@ async def test_creating_sensor_loads_group(hass: HomeAssistant) -> None:
async def async_setup_template( async def async_setup_template(
hass: HomeAssistant, hass: HomeAssistant,
config: ConfigType, config: ConfigType,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
discovery_info: DiscoveryInfoType | None = None, discovery_info: DiscoveryInfoType | None = None,
) -> bool: ) -> bool:
order.append("sensor.template") order.append("sensor.template")

View File

@ -3,7 +3,7 @@
from homeassistant.components.todo import DOMAIN, TodoItem, TodoListEntity from homeassistant.components.todo import DOMAIN, TodoItem, TodoListEntity
from homeassistant.config_entries import ConfigEntry, ConfigFlow from homeassistant.config_entries import ConfigEntry, ConfigFlow
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from tests.common import MockConfigEntry, MockPlatform, mock_platform from tests.common import MockConfigEntry, MockPlatform, mock_platform
@ -44,7 +44,7 @@ async def create_mock_platform(
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test event platform via config entry.""" """Set up test event platform via config entry."""
async_add_entities(entities) async_add_entities(entities)

View File

@ -24,7 +24,7 @@ from homeassistant.components.tts import (
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -249,7 +249,7 @@ async def mock_config_entry_setup(
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test tts platform via config entry.""" """Set up test tts platform via config entry."""
async_add_entities([tts_entity]) async_add_entities([tts_entity])

View File

@ -43,7 +43,7 @@ from homeassistant.const import (
) )
from homeassistant.core import HomeAssistant, State, callback from homeassistant.core import HomeAssistant, State, callback
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.event import async_track_state_change_event from homeassistant.helpers.event import async_track_state_change_event
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -857,7 +857,7 @@ async def test_name(hass: HomeAssistant) -> None:
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test update platform via config entry.""" """Set up test update platform via config entry."""
async_add_entities([entity1, entity2, entity3, entity4]) async_add_entities([entity1, entity2, entity3, entity4])

View File

@ -9,7 +9,7 @@ from homeassistant.components.vacuum import DOMAIN as VACUUM_DOMAIN, VacuumEntit
from homeassistant.config_entries import ConfigEntry, ConfigFlow from homeassistant.config_entries import ConfigEntry, ConfigFlow
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er, frame from homeassistant.helpers import entity_registry as er, frame
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from . import MockVacuum from . import MockVacuum
@ -87,7 +87,7 @@ async def setup_vacuum_platform_test_entity(
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test vacuum platform via config entry.""" """Set up test vacuum platform via config entry."""
async_add_entities([entity]) async_add_entities([entity])

View File

@ -22,7 +22,7 @@ from homeassistant.const import (
Platform, Platform,
) )
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from tests.common import ( from tests.common import (
MockConfigEntry, MockConfigEntry,
@ -174,7 +174,7 @@ def mock_config_entry(hass: HomeAssistant) -> tuple[MockConfigEntry, list[ValveE
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test platform via config entry.""" """Set up test platform via config entry."""
async_add_entities(entities) async_add_entities(entities)

View File

@ -13,7 +13,7 @@ from homeassistant.components import wake_word
from homeassistant.config_entries import ConfigEntry, ConfigEntryState, ConfigFlow from homeassistant.config_entries import ConfigEntry, ConfigEntryState, ConfigFlow
from homeassistant.const import EntityCategory from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, State from homeassistant.core import HomeAssistant, State
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from .common import mock_wake_word_entity_platform from .common import mock_wake_word_entity_platform
@ -143,7 +143,7 @@ async def mock_config_entry_setup(
async def async_setup_entry_platform( async def async_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test stt platform via config entry.""" """Set up test stt platform via config entry."""
async_add_entities([mock_provider_entity]) async_add_entities([mock_provider_entity])

View File

@ -23,7 +23,7 @@ from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers import config_validation as cv from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from tests.common import ( from tests.common import (
MockConfigEntry, MockConfigEntry,
@ -145,7 +145,7 @@ async def test_operation_mode_validation(
async def async_setup_entry_water_heater_platform( async def async_setup_entry_water_heater_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test water_heater platform via config entry.""" """Set up test water_heater platform via config entry."""
async_add_entities([water_heater_entity]) async_add_entities([water_heater_entity])

View File

@ -21,7 +21,7 @@ from homeassistant.components.weather import (
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from tests.common import ( from tests.common import (
MockConfigEntry, MockConfigEntry,
@ -90,7 +90,7 @@ async def create_entity(
async def async_setup_entry_weather_platform( async def async_setup_entry_weather_platform(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Set up test weather platform via config entry.""" """Set up test weather platform via config entry."""
async_add_entities([weather_entity]) async_add_entities([weather_entity])

View File

@ -35,7 +35,7 @@ from homeassistant.core import (
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import device_registry as dr, entity, entity_registry as er from homeassistant.helpers import device_registry as dr, entity, entity_registry as er
from homeassistant.helpers.entity_component import async_update_entity from homeassistant.helpers.entity_component import async_update_entity
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.typing import UNDEFINED, UndefinedType from homeassistant.helpers.typing import UNDEFINED, UndefinedType
from tests.common import ( from tests.common import (
@ -986,7 +986,7 @@ async def _test_friendly_name(
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Mock setup entry method.""" """Mock setup entry method."""
async_add_entities([ent]) async_add_entities([ent])
@ -1314,7 +1314,7 @@ async def test_entity_name_translation_placeholder_errors(
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Mock setup entry method.""" """Mock setup entry method."""
async_add_entities([ent]) async_add_entities([ent])
@ -1542,7 +1542,7 @@ async def test_friendly_name_updated(
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Mock setup entry method.""" """Mock setup entry method."""
async_add_entities( async_add_entities(

View File

@ -39,7 +39,7 @@ from homeassistant.exceptions import (
) )
from homeassistant.helpers import entity_registry as er, frame, issue_registry as ir from homeassistant.helpers import entity_registry as er, frame, issue_registry as ir
from homeassistant.helpers.discovery_flow import DiscoveryKey from homeassistant.helpers.discovery_flow import DiscoveryKey
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.json import json_dumps from homeassistant.helpers.json import json_dumps
from homeassistant.helpers.service_info.dhcp import DhcpServiceInfo from homeassistant.helpers.service_info.dhcp import DhcpServiceInfo
from homeassistant.helpers.service_info.hassio import HassioServiceInfo from homeassistant.helpers.service_info.hassio import HassioServiceInfo
@ -469,7 +469,7 @@ async def test_remove_entry(
async def mock_setup_entry_platform( async def mock_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
entry: config_entries.ConfigEntry, entry: config_entries.ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Mock setting up platform.""" """Mock setting up platform."""
async_add_entities([entity]) async_add_entities([entity])
@ -6625,7 +6625,7 @@ async def test_raise_wrong_exception_in_forwarded_platform(
async def mock_setup_entry_platform( async def mock_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
entry: config_entries.ConfigEntry, entry: config_entries.ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Mock setting up platform.""" """Mock setting up platform."""
raise exc raise exc
@ -6699,7 +6699,7 @@ async def test_config_entry_unloaded_during_platform_setups(
async def mock_setup_entry_platform( async def mock_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
entry: config_entries.ConfigEntry, entry: config_entries.ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Mock setting up platform.""" """Mock setting up platform."""
await asyncio.sleep(0) await asyncio.sleep(0)
@ -6771,7 +6771,7 @@ async def test_non_awaited_async_forward_entry_setups(
async def mock_setup_entry_platform( async def mock_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
entry: config_entries.ConfigEntry, entry: config_entries.ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Mock setting up platform.""" """Mock setting up platform."""
await forward_event.wait() await forward_event.wait()
@ -6843,7 +6843,7 @@ async def test_non_awaited_async_forward_entry_setup(
async def mock_setup_entry_platform( async def mock_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
entry: config_entries.ConfigEntry, entry: config_entries.ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Mock setting up platform.""" """Mock setting up platform."""
await forward_event.wait() await forward_event.wait()
@ -6918,7 +6918,7 @@ async def test_config_entry_unloaded_during_platform_setup(
async def mock_setup_entry_platform( async def mock_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
entry: config_entries.ConfigEntry, entry: config_entries.ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Mock setting up platform.""" """Mock setting up platform."""
await asyncio.sleep(0) await asyncio.sleep(0)
@ -6993,7 +6993,7 @@ async def test_config_entry_late_platform_setup(
async def mock_setup_entry_platform( async def mock_setup_entry_platform(
hass: HomeAssistant, hass: HomeAssistant,
entry: config_entries.ConfigEntry, entry: config_entries.ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddConfigEntryEntitiesCallback,
) -> None: ) -> None:
"""Mock setting up platform.""" """Mock setting up platform."""
await asyncio.sleep(0) await asyncio.sleep(0)