jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 13:17:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use MockConfigEntry in unifi tests (#117238)

Browse Source
This commit is contained in:
Erik Montnemery 2024-05-11 11:11:53 +02:00 committed by GitHub
parent daef625985
commit 90a50c162d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/components/unifi/test_device_tracker.py
View File

@ -5,7 +5,6 @@ from datetime import timedelta
from aiounifi.models.message import MessageKey from aiounifi.models.message import MessageKey
from freezegun.api import FrozenDateTimeFactory, freeze_time 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.device_tracker import DOMAIN as TRACKER_DOMAIN
from homeassistant.components.unifi.const import ( from homeassistant.components.unifi.const import (
CONF_BLOCK_CLIENT, CONF_BLOCK_CLIENT,
@ -26,7 +25,7 @@ import homeassistant.util.dt as dt_util
from .test_hub import ENTRY_CONFIG, setup_unifi_integration 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 from tests.test_util.aiohttp import AiohttpClientMocker
@ -959,11 +958,8 @@ async def test_restoring_client(
"mac": "00:00:00:00:00:03", "mac": "00:00:00:00:00:03",
} }
config_entry = config_entries.ConfigEntry( config_entry = MockConfigEntry(
version=1,
minor_version=1,
domain=UNIFI_DOMAIN, domain=UNIFI_DOMAIN,
title="Mock Title",
data=ENTRY_CONFIG, data=ENTRY_CONFIG,
source="test", source="test",
options={}, options={},

8
tests/components/unifi/test_switch.py
View File

@ -6,7 +6,6 @@ from datetime import timedelta
from aiounifi.models.message import MessageKey from aiounifi.models.message import MessageKey
import pytest import pytest
from homeassistant import config_entries
from homeassistant.components.switch import ( from homeassistant.components.switch import (
DOMAIN as SWITCH_DOMAIN, DOMAIN as SWITCH_DOMAIN,
SERVICE_TURN_OFF, 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 .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 from tests.test_util.aiohttp import AiohttpClientMocker
CLIENT_1 = { CLIENT_1 = {
@ -1628,11 +1627,8 @@ async def test_updating_unique_id(
], ],
} }
config_entry = config_entries.ConfigEntry( config_entry = MockConfigEntry(
version=1,
minor_version=1,
domain=UNIFI_DOMAIN, domain=UNIFI_DOMAIN,
title="Mock Title",
data=ENTRY_CONFIG, data=ENTRY_CONFIG,
source="test", source="test",
options={}, options={},