Feedback

Author: erwindouna

tests/components/portainer/test_system_health.py

@@ -1,15 +1,15 @@
 """Test Portainer system health."""
 
 import asyncio
+from unittest.mock import AsyncMock
 
 from aiohttp import ClientError
 
 from homeassistant.components.portainer.const import DOMAIN
-from homeassistant.config_entries import ConfigEntryState
 from homeassistant.core import HomeAssistant
 from homeassistant.setup import async_setup_component
 
-from .conftest import MOCK_TEST_CONFIG, TEST_ENTRY, TEST_INSTANCE_ID
+from . import setup_integration
 
 from tests.common import MockConfigEntry, get_system_health_info
 from tests.test_util.aiohttp import AiohttpClientMocker
@@ -20,22 +20,14 @@
 async def test_system_health(
     hass: HomeAssistant,
     aioclient_mock: AiohttpClientMocker,
+    mock_config_entry: MockConfigEntry,
+    mock_portainer_client: AsyncMock,
 ) -> None:
     """Test system health when server is reachable."""
     aioclient_mock.get(MOCK_HEALTH_URL, text="ok")
 
-    hass.config.components.add(DOMAIN)
     assert await async_setup_component(hass, "system_health", {})
-    await hass.async_block_till_done()
-
-    entry = MockConfigEntry(
-        domain=DOMAIN,
-        data=MOCK_TEST_CONFIG,
-        unique_id=TEST_INSTANCE_ID,
-        entry_id=TEST_ENTRY,
-        state=ConfigEntryState.LOADED,
-    )
-    entry.add_to_hass(hass)
+    await setup_integration(hass, mock_config_entry)
 
     info = await get_system_health_info(hass, DOMAIN)
 
@@ -49,22 +41,14 @@ async def test_system_health(
 async def test_system_health_failed_connect(
     hass: HomeAssistant,
     aioclient_mock: AiohttpClientMocker,
+    mock_config_entry: MockConfigEntry,
+    mock_portainer_client: AsyncMock,
 ) -> None:
     """Test system health when server is unreachable."""
     aioclient_mock.get(MOCK_HEALTH_URL, exc=ClientError)
 
-    hass.config.components.add(DOMAIN)
     assert await async_setup_component(hass, "system_health", {})
-    await hass.async_block_till_done()
-
-    entry = MockConfigEntry(
-        domain=DOMAIN,
-        data=MOCK_TEST_CONFIG,
-        unique_id=TEST_INSTANCE_ID,
-        entry_id=TEST_ENTRY,
-        state=ConfigEntryState.LOADED,
-    )
-    entry.add_to_hass(hass)
+    await setup_integration(hass, mock_config_entry)
 
     info = await get_system_health_info(hass, DOMAIN)