evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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 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={},

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

@ -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={},