diff --git a/tests/components/unifi/test_device_tracker.py b/tests/components/unifi/test_device_tracker.py index b22767a2914..4037d976430 100644 --- a/tests/components/unifi/test_device_tracker.py +++ b/tests/components/unifi/test_device_tracker.py @@ -5,7 +5,6 @@ from datetime import timedelta 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 @@ import homeassistant.util.dt as dt_util 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 index a6b787045bd..9b63113e750 100644 --- a/tests/components/unifi/test_switch.py +++ b/tests/components/unifi/test_switch.py @@ -6,7 +6,6 @@ from datetime import timedelta 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 homeassistant.util import dt as dt_util 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={},