Update to not use deprecated method

PeteRager · Aug 2, 2024 · 08ba029 · 08ba029
1 parent d76e0c0
commit 08ba029
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/custom_components/lennoxs30/__init__.py b/custom_components/lennoxs30/__init__.py
@@ -548,10 +548,7 @@ async def s30_initialize(self):
  await self.create_devices()
  # Only add entities the first time, on reconnect we do not need to add them again
  if self._climate_entities_initialized is False:
- for platform in PLATFORMS:
- self._hass.async_create_task(
- self._hass.config_entries.async_forward_entry_setup(self._config, platform)
- )
+ await self._hass.config_entries.async_forward_entry_setups(self._config, PLATFORMS)
  self._climate_entities_initialized = True
  self.updateState(DS_CONNECTED)
 

diff --git a/tests/test_manager.py b/tests/test_manager.py
@@ -321,8 +321,8 @@ async def test_manager_s30_initialize(hass: HomeAssistant, manager_us_customary_
  with patch.object(manager, "configuration_initialization") as configuration_initialization:
  with patch("asyncio.create_task") as create_task:
  create_task.return_value = "AWAITABLE_TASK"
- with patch.object(manager, "create_devices") as create_devices:
- with patch.object(hass, "async_create_task") as hass_create_task:
+ with patch.object(hass.config_entries, "async_forward_entry_setups") as forward_entry_setups:
+ with patch.object(manager, "create_devices") as create_devices:
  await manager.s30_initialize()
 
  assert update_state.call_count == 2
@@ -341,7 +341,7 @@ async def test_manager_s30_initialize(hass: HomeAssistant, manager_us_customary_
  assert create_devices.call_count == 1
  assert len(create_devices.mock_calls[0].args) == 0
 
- assert hass_create_task.call_count == len(PLATFORMS)
+ assert forward_entry_setups.call_count == 1
 
 
 @pytest.mark.asyncio