Use MockConfigEntry in unifi tests (#117238)

home-assistant · May 11, 2024 · 90a50c1 · 90a50c1
1 parent daef625
commit 90a50c1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 12 deletions.
diff --git a/tests/components/unifi/test_device_tracker.py b/tests/components/unifi/test_device_tracker.py
@@ -5,7 +5,6 @@
 from aiounifi.models.message import MessageKey
 from freezegun.api import FrozenDateTimeFactory, freeze_time
 
-from homeassistant import config_entries
 from homeassistant.components.device_tracker import DOMAIN as TRACKER_DOMAIN
 from homeassistant.components.unifi.const import (
     CONF_BLOCK_CLIENT,
@@ -26,7 +25,7 @@
 
 from .test_hub import ENTRY_CONFIG, setup_unifi_integration
 
-from tests.common import async_fire_time_changed
+from tests.common import MockConfigEntry, async_fire_time_changed
 from tests.test_util.aiohttp import AiohttpClientMocker
 
 
@@ -959,11 +958,8 @@ async def test_restoring_client(
         "mac": "00:00:00:00:00:03",
     }
 
-    config_entry = config_entries.ConfigEntry(
-        version=1,
-        minor_version=1,
+    config_entry = MockConfigEntry(
         domain=UNIFI_DOMAIN,
-        title="Mock Title",
         data=ENTRY_CONFIG,
         source="test",
         options={},

diff --git a/tests/components/unifi/test_switch.py b/tests/components/unifi/test_switch.py
@@ -6,7 +6,6 @@
 from aiounifi.models.message import MessageKey
 import pytest
 
-from homeassistant import config_entries
 from homeassistant.components.switch import (
     DOMAIN as SWITCH_DOMAIN,
     SERVICE_TURN_OFF,
@@ -38,7 +37,7 @@
 
 from .test_hub import CONTROLLER_HOST, ENTRY_CONFIG, SITE, setup_unifi_integration
 
-from tests.common import async_fire_time_changed
+from tests.common import MockConfigEntry, async_fire_time_changed
 from tests.test_util.aiohttp import AiohttpClientMocker
 
 CLIENT_1 = {
@@ -1628,11 +1627,8 @@ async def test_updating_unique_id(
         ],
     }
 
-    config_entry = config_entries.ConfigEntry(
-        version=1,
-        minor_version=1,
+    config_entry = MockConfigEntry(
         domain=UNIFI_DOMAIN,
-        title="Mock Title",
         data=ENTRY_CONFIG,
         source="test",
         options={},