diff --git a/tests/components/huum/test_climate.py b/tests/components/huum/test_climate.py index 3e0a07f92325..c8eb7c775669 100644 --- a/tests/components/huum/test_climate.py +++ b/tests/components/huum/test_climate.py @@ -7,6 +7,7 @@ from syrupy.assertion import SnapshotAssertion from homeassistant.components.climate import ( ATTR_HVAC_MODE, + DOMAIN as CLIMATE_DOMAIN, SERVICE_SET_HVAC_MODE, SERVICE_SET_TEMPERATURE, HVACMode, @@ -48,7 +49,7 @@ async def test_set_hvac_mode( mock_huum.status = SaunaStatus.ONLINE_HEATING await hass.services.async_call( - Platform.CLIMATE, + CLIMATE_DOMAIN, SERVICE_SET_HVAC_MODE, {ATTR_ENTITY_ID: ENTITY_ID, ATTR_HVAC_MODE: HVACMode.HEAT}, blocking=True, @@ -70,7 +71,7 @@ async def test_set_temperature( mock_huum.status = SaunaStatus.ONLINE_HEATING await hass.services.async_call( - Platform.CLIMATE, + CLIMATE_DOMAIN, SERVICE_SET_TEMPERATURE, { ATTR_ENTITY_ID: ENTITY_ID, diff --git a/tests/components/huum/test_light.py b/tests/components/huum/test_light.py index 8ad12a36f4ea..db9ee0f69527 100644 --- a/tests/components/huum/test_light.py +++ b/tests/components/huum/test_light.py @@ -4,6 +4,7 @@ from unittest.mock import AsyncMock from syrupy.assertion import SnapshotAssertion +from homeassistant.components.light import DOMAIN as LIGHT_DOMAIN from homeassistant.const import ( ATTR_ENTITY_ID, SERVICE_TURN_OFF, @@ -46,7 +47,7 @@ async def test_light_turn_off( assert state.state == STATE_ON await hass.services.async_call( - Platform.LIGHT, + LIGHT_DOMAIN, SERVICE_TURN_OFF, {ATTR_ENTITY_ID: ENTITY_ID}, blocking=True, @@ -68,7 +69,7 @@ async def test_light_turn_on( assert state.state == STATE_OFF await hass.services.async_call( - Platform.LIGHT, + LIGHT_DOMAIN, SERVICE_TURN_ON, {ATTR_ENTITY_ID: ENTITY_ID}, blocking=True,