diff --git a/tests/helpers/test_collection.py b/tests/helpers/test_collection.py index fe403f8db1b..64c83757a7b 100644 --- a/tests/helpers/test_collection.py +++ b/tests/helpers/test_collection.py @@ -256,6 +256,7 @@ async def test_storage_collection(hass: HomeAssistant) -> None: async def test_attach_entity_component_collection(hass: HomeAssistant) -> None: """Test attaching collection to entity component.""" ent_comp = entity_component.EntityComponent(_LOGGER, "test", hass) + await ent_comp.async_setup({}) coll = MockObservableCollection(_LOGGER) collection.sync_entity_lifecycle(hass, "test", "test", ent_comp, coll, MockEntity) @@ -295,6 +296,7 @@ async def test_attach_entity_component_collection(hass: HomeAssistant) -> None: async def test_entity_component_collection_abort(hass: HomeAssistant) -> None: """Test aborted entity adding is handled.""" ent_comp = entity_component.EntityComponent(_LOGGER, "test", hass) + await ent_comp.async_setup({}) coll = MockObservableCollection(_LOGGER) async_update_config_calls = [] @@ -361,6 +363,7 @@ async def test_entity_component_collection_abort(hass: HomeAssistant) -> None: async def test_entity_component_collection_entity_removed(hass: HomeAssistant) -> None: """Test entity removal is handled.""" ent_comp = entity_component.EntityComponent(_LOGGER, "test", hass) + await ent_comp.async_setup({}) coll = MockObservableCollection(_LOGGER) async_update_config_calls = []