Make device entry disabled by an enum (#60239)

This commit is contained in:
Ville Skyttä 2021-11-24 23:32:16 +02:00 committed by GitHub
parent 42389fc81b
commit 9f4de8df18
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 52 additions and 39 deletions

View file

@ -3,6 +3,7 @@ import pytest
from homeassistant.components.config import entity_registry
from homeassistant.const import ATTR_ICON
from homeassistant.helpers.device_registry import DeviceEntryDisabler
from homeassistant.helpers.entity_registry import DISABLED_USER, RegistryEntry
from tests.common import (
@ -325,7 +326,7 @@ async def test_enable_entity_disabled_device(hass, client, device_registry):
identifiers={("bridgeid", "0123")},
manufacturer="manufacturer",
model="model",
disabled_by=DISABLED_USER,
disabled_by=DeviceEntryDisabler.USER,
)
mock_registry(