use DOMAIN instead of const.DOMAIN

This commit is contained in:
starkillerOG
2024-08-23 14:40:08 +02:00
parent a4f0a04c59
commit befe8a75b5

View File

@@ -7,7 +7,8 @@ import pytest
from reolink_aio.api import Chime from reolink_aio.api import Chime
from reolink_aio.exceptions import ReolinkError from reolink_aio.exceptions import ReolinkError
from homeassistant.components.reolink import DEVICE_UPDATE_INTERVAL, const from homeassistant.components.reolink import DEVICE_UPDATE_INTERVAL
from homeassistant.components.reolink.const import DOMAIN
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
from homeassistant.config_entries import ConfigEntryState from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import ( from homeassistant.const import (
@@ -42,14 +43,14 @@ async def test_cleanup_hdr_switch_(
entity_registry.async_get_or_create( entity_registry.async_get_or_create(
domain=domain, domain=domain,
platform=const.DOMAIN, platform=DOMAIN,
unique_id=original_id, unique_id=original_id,
config_entry=config_entry, config_entry=config_entry,
suggested_object_id=original_id, suggested_object_id=original_id,
disabled_by=er.RegistryEntryDisabler.USER, disabled_by=er.RegistryEntryDisabler.USER,
) )
assert entity_registry.async_get_entity_id(domain, const.DOMAIN, original_id) assert entity_registry.async_get_entity_id(domain, DOMAIN, original_id)
# setup CH 0 and host entities/device # setup CH 0 and host entities/device
with patch("homeassistant.components.reolink.PLATFORMS", [domain]): with patch("homeassistant.components.reolink.PLATFORMS", [domain]):
@@ -57,7 +58,7 @@ async def test_cleanup_hdr_switch_(
await hass.async_block_till_done() await hass.async_block_till_done()
assert ( assert (
entity_registry.async_get_entity_id(domain, const.DOMAIN, original_id) is None entity_registry.async_get_entity_id(domain, DOMAIN, original_id) is None
) )
@@ -77,23 +78,23 @@ async def test_hdr_switch_deprecated_repair_issue(
entity_registry.async_get_or_create( entity_registry.async_get_or_create(
domain=domain, domain=domain,
platform=const.DOMAIN, platform=DOMAIN,
unique_id=original_id, unique_id=original_id,
config_entry=config_entry, config_entry=config_entry,
suggested_object_id=original_id, suggested_object_id=original_id,
disabled_by=None, disabled_by=None,
) )
assert entity_registry.async_get_entity_id(domain, const.DOMAIN, original_id) assert entity_registry.async_get_entity_id(domain, DOMAIN, original_id)
# setup CH 0 and host entities/device # setup CH 0 and host entities/device
with patch("homeassistant.components.reolink.PLATFORMS", [domain]): with patch("homeassistant.components.reolink.PLATFORMS", [domain]):
assert await hass.config_entries.async_setup(config_entry.entry_id) assert await hass.config_entries.async_setup(config_entry.entry_id)
await hass.async_block_till_done() await hass.async_block_till_done()
assert entity_registry.async_get_entity_id(domain, const.DOMAIN, original_id) assert entity_registry.async_get_entity_id(domain, DOMAIN, original_id)
assert (const.DOMAIN, "hdr_switch_deprecated") in issue_registry.issues assert (DOMAIN, "hdr_switch_deprecated") in issue_registry.issues
async def test_switch( async def test_switch(