From 42793927f7931155b23cf9810c7c1677bf3d98f5 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Mon, 25 Oct 2021 15:42:40 +0200 Subject: [PATCH] Use ATTR_VIA_DEVICE constant in onewire tests (#58405) Co-authored-by: epenet --- tests/components/onewire/__init__.py | 3 ++- tests/components/onewire/const.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/components/onewire/__init__.py b/tests/components/onewire/__init__.py index 25ff4a15cfd..36035c1c85b 100644 --- a/tests/components/onewire/__init__.py +++ b/tests/components/onewire/__init__.py @@ -15,6 +15,7 @@ from homeassistant.const import ( ATTR_MODEL, ATTR_NAME, ATTR_STATE, + ATTR_VIA_DEVICE, ) from homeassistant.core import HomeAssistant from homeassistant.helpers.device_registry import DeviceRegistry @@ -57,7 +58,7 @@ def check_device_registry( assert registry_entry.manufacturer == expected_device[ATTR_MANUFACTURER] assert registry_entry.name == expected_device[ATTR_NAME] assert registry_entry.model == expected_device[ATTR_MODEL] - if expected_via_device := expected_device.get("via_device"): + if expected_via_device := expected_device.get(ATTR_VIA_DEVICE): assert registry_entry.via_device_id is not None parent_entry = device_registry.async_get_device({expected_via_device}) assert parent_entry is not None diff --git a/tests/components/onewire/const.py b/tests/components/onewire/const.py index 91b7b618c37..2642e4e7454 100644 --- a/tests/components/onewire/const.py +++ b/tests/components/onewire/const.py @@ -20,6 +20,7 @@ from homeassistant.const import ( ATTR_NAME, ATTR_STATE, ATTR_UNIT_OF_MEASUREMENT, + ATTR_VIA_DEVICE, DEVICE_CLASS_CURRENT, DEVICE_CLASS_HUMIDITY, DEVICE_CLASS_ILLUMINANCE, @@ -224,7 +225,7 @@ MOCK_OWPROXY_DEVICES = { ATTR_MANUFACTURER: MANUFACTURER, ATTR_MODEL: "DS2423", ATTR_NAME: "1D.111111111111", - "via_device": (DOMAIN, "1F.111111111111"), + ATTR_VIA_DEVICE: (DOMAIN, "1F.111111111111"), }, ], "branches": {