fix: switch to async_forward_entry_setup (#70)

firstof9 · Jan 28, 2023 · c7ce04e · c7ce04e
1 parent 85e94d4
commit c7ce04e
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 1 deletion.
diff --git a/custom_components/openei/__init__.py b/custom_components/openei/__init__.py
@@ -67,7 +67,10 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
         raise ConfigEntryNotReady
 
     hass.data[DOMAIN][entry.entry_id] = coordinator
-    hass.config_entries.async_setup_platforms(entry, PLATFORMS)
+    for platform in PLATFORMS:
+        hass.async_add_job(
+            hass.config_entries.async_forward_entry_setup(entry, platform)
+        )
 
     return True
 

diff --git a/tests/test_config_flow.py b/tests/test_config_flow.py
@@ -7,6 +7,8 @@
 
 from custom_components.openei.const import DOMAIN
 
+pytestmark = pytest.mark.asyncio
+
 
 @pytest.mark.parametrize(
     "input_1,step_id_2,input_2,step_id_3,input_3,title,data",

diff --git a/tests/test_init.py b/tests/test_init.py
@@ -12,6 +12,8 @@
 from custom_components.openei.const import DOMAIN
 from tests.const import CONFIG_DATA, CONFIG_DATA_MISSING_PLAN, CONFIG_DATA_WITH_SENSOR
 
+pytestmark = pytest.mark.asyncio
+
 async def test_setup_entry(hass, mock_sensors, mock_api):
     """Test settting up entities."""
     entry = MockConfigEntry(

diff --git a/tests/test_sensors.py b/tests/test_sensors.py
@@ -1,4 +1,5 @@
 """Tests for sensors."""
+import pytest
 
 from pytest_homeassistant_custom_component.common import MockConfigEntry
 
@@ -8,6 +9,8 @@
 FAKE_MINCHARGE_SENSOR = "sensor.fake_utility_co_minimum_charge"
 FAKE_CURRENT_RATE_SENSOR = "sensor.fake_utility_co_current_energy_rate"
 
+pytestmark = pytest.mark.asyncio
+
 
 async def test_sensors(hass, mock_sensors, mock_api):
     """Test settting up entities."""