jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-20 03:37:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Do not directy import platform DOMAIN const in MQTT platform tests (#125589)

Browse Source
This commit is contained in:
Jan Bouwhuis 2024-09-09 14:36:15 +02:00 committed by GitHub
parent 1dc496a2dd
commit 3889482f0e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 36 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/components/mqtt/test_humidifier.py
View File

@ -12,7 +12,6 @@ from homeassistant.components.humidifier import (
ATTR_CURRENT_HUMIDITY, ATTR_CURRENT_HUMIDITY,
ATTR_HUMIDITY, ATTR_HUMIDITY,
ATTR_MODE, ATTR_MODE,
DOMAIN,
SERVICE_SET_HUMIDITY, SERVICE_SET_HUMIDITY,
SERVICE_SET_MODE, SERVICE_SET_MODE,
HumidifierAction, HumidifierAction,
@ -87,7 +86,9 @@ async def async_turn_on(hass: HomeAssistant, entity_id: str = ENTITY_MATCH_ALL)
"""Turn all or specified humidifier on.""" """Turn all or specified humidifier on."""
data = {ATTR_ENTITY_ID: entity_id} if entity_id else {} data = {ATTR_ENTITY_ID: entity_id} if entity_id else {}
await hass.services.async_call(DOMAIN, SERVICE_TURN_ON, data, blocking=True) await hass.services.async_call(
humidifier.DOMAIN, SERVICE_TURN_ON, data, blocking=True
)
async def async_turn_off( async def async_turn_off(
@ -96,7 +97,9 @@ async def async_turn_off(
"""Turn all or specified humidier off.""" """Turn all or specified humidier off."""
data = {ATTR_ENTITY_ID: entity_id} if entity_id else {} data = {ATTR_ENTITY_ID: entity_id} if entity_id else {}
await hass.services.async_call(DOMAIN, SERVICE_TURN_OFF, data, blocking=True) await hass.services.async_call(
humidifier.DOMAIN, SERVICE_TURN_OFF, data, blocking=True
)
async def async_set_mode( async def async_set_mode(
@ -109,7 +112,9 @@ async def async_set_mode(
if value is not None if value is not None
} }
await hass.services.async_call(DOMAIN, SERVICE_SET_MODE, data, blocking=True) await hass.services.async_call(
humidifier.DOMAIN, SERVICE_SET_MODE, data, blocking=True
)
async def async_set_humidity( async def async_set_humidity(
@ -122,7 +127,9 @@ async def async_set_humidity(
if value is not None if value is not None
} }
await hass.services.async_call(DOMAIN, SERVICE_SET_HUMIDITY, data, blocking=True) await hass.services.async_call(
humidifier.DOMAIN, SERVICE_SET_HUMIDITY, data, blocking=True
)
@pytest.mark.parametrize( @pytest.mark.parametrize(

37
tests/components/mqtt/test_vacuum.py
View File

@ -21,7 +21,6 @@ from homeassistant.components.vacuum import (
ATTR_BATTERY_LEVEL, ATTR_BATTERY_LEVEL,
ATTR_FAN_SPEED, ATTR_FAN_SPEED,
ATTR_FAN_SPEED_LIST, ATTR_FAN_SPEED_LIST,
DOMAIN,
SERVICE_CLEAN_SPOT, SERVICE_CLEAN_SPOT,
SERVICE_LOCATE, SERVICE_LOCATE,
SERVICE_PAUSE, SERVICE_PAUSE,
@ -122,31 +121,34 @@ async def test_all_commands(
mqtt_mock = await mqtt_mock_entry() mqtt_mock = await mqtt_mock_entry()
await hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_START, {"entity_id": ENTITY_MATCH_ALL}, blocking=True vacuum.DOMAIN, SERVICE_START, {"entity_id": ENTITY_MATCH_ALL}, blocking=True
) )
mqtt_mock.async_publish.assert_called_once_with(COMMAND_TOPIC, "start", 0, False) mqtt_mock.async_publish.assert_called_once_with(COMMAND_TOPIC, "start", 0, False)
mqtt_mock.async_publish.reset_mock() mqtt_mock.async_publish.reset_mock()
await hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_STOP, {"entity_id": ENTITY_MATCH_ALL}, blocking=True vacuum.DOMAIN, SERVICE_STOP, {"entity_id": ENTITY_MATCH_ALL}, blocking=True
) )
mqtt_mock.async_publish.assert_called_once_with(COMMAND_TOPIC, "stop", 0, False) mqtt_mock.async_publish.assert_called_once_with(COMMAND_TOPIC, "stop", 0, False)
mqtt_mock.async_publish.reset_mock() mqtt_mock.async_publish.reset_mock()
await hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_PAUSE, {"entity_id": ENTITY_MATCH_ALL}, blocking=True vacuum.DOMAIN, SERVICE_PAUSE, {"entity_id": ENTITY_MATCH_ALL}, blocking=True
) )
mqtt_mock.async_publish.assert_called_once_with(COMMAND_TOPIC, "pause", 0, False) mqtt_mock.async_publish.assert_called_once_with(COMMAND_TOPIC, "pause", 0, False)
mqtt_mock.async_publish.reset_mock() mqtt_mock.async_publish.reset_mock()
await hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_LOCATE, {"entity_id": ENTITY_MATCH_ALL}, blocking=True vacuum.DOMAIN, SERVICE_LOCATE, {"entity_id": ENTITY_MATCH_ALL}, blocking=True
) )
mqtt_mock.async_publish.assert_called_once_with(COMMAND_TOPIC, "locate", 0, False) mqtt_mock.async_publish.assert_called_once_with(COMMAND_TOPIC, "locate", 0, False)
mqtt_mock.async_publish.reset_mock() mqtt_mock.async_publish.reset_mock()
await hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_CLEAN_SPOT, {"entity_id": ENTITY_MATCH_ALL}, blocking=True vacuum.DOMAIN,
SERVICE_CLEAN_SPOT,
{"entity_id": ENTITY_MATCH_ALL},
blocking=True,
) )
mqtt_mock.async_publish.assert_called_once_with( mqtt_mock.async_publish.assert_called_once_with(
COMMAND_TOPIC, "clean_spot", 0, False COMMAND_TOPIC, "clean_spot", 0, False
@ -154,7 +156,10 @@ async def test_all_commands(
mqtt_mock.async_publish.reset_mock() mqtt_mock.async_publish.reset_mock()
await hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_RETURN_TO_BASE, {"entity_id": ENTITY_MATCH_ALL}, blocking=True vacuum.DOMAIN,
SERVICE_RETURN_TO_BASE,
{"entity_id": ENTITY_MATCH_ALL},
blocking=True,
) )
mqtt_mock.async_publish.assert_called_once_with( mqtt_mock.async_publish.assert_called_once_with(
COMMAND_TOPIC, "return_to_base", 0, False COMMAND_TOPIC, "return_to_base", 0, False
@ -205,37 +210,43 @@ async def test_commands_without_supported_features(
mqtt_mock = await mqtt_mock_entry() mqtt_mock = await mqtt_mock_entry()
await hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_START, {"entity_id": ENTITY_MATCH_ALL}, blocking=True vacuum.DOMAIN, SERVICE_START, {"entity_id": ENTITY_MATCH_ALL}, blocking=True
) )
mqtt_mock.async_publish.assert_not_called() mqtt_mock.async_publish.assert_not_called()
mqtt_mock.async_publish.reset_mock() mqtt_mock.async_publish.reset_mock()
await hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_PAUSE, {"entity_id": ENTITY_MATCH_ALL}, blocking=True vacuum.DOMAIN, SERVICE_PAUSE, {"entity_id": ENTITY_MATCH_ALL}, blocking=True
) )
mqtt_mock.async_publish.assert_not_called() mqtt_mock.async_publish.assert_not_called()
mqtt_mock.async_publish.reset_mock() mqtt_mock.async_publish.reset_mock()
await hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_STOP, {"entity_id": ENTITY_MATCH_ALL}, blocking=True vacuum.DOMAIN, SERVICE_STOP, {"entity_id": ENTITY_MATCH_ALL}, blocking=True
) )
mqtt_mock.async_publish.assert_not_called() mqtt_mock.async_publish.assert_not_called()
mqtt_mock.async_publish.reset_mock() mqtt_mock.async_publish.reset_mock()
await hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_RETURN_TO_BASE, {"entity_id": ENTITY_MATCH_ALL}, blocking=True vacuum.DOMAIN,
SERVICE_RETURN_TO_BASE,
{"entity_id": ENTITY_MATCH_ALL},
blocking=True,
) )
mqtt_mock.async_publish.assert_not_called() mqtt_mock.async_publish.assert_not_called()
mqtt_mock.async_publish.reset_mock() mqtt_mock.async_publish.reset_mock()
await hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_LOCATE, {"entity_id": ENTITY_MATCH_ALL}, blocking=True vacuum.DOMAIN, SERVICE_LOCATE, {"entity_id": ENTITY_MATCH_ALL}, blocking=True
) )
mqtt_mock.async_publish.assert_not_called() mqtt_mock.async_publish.assert_not_called()
mqtt_mock.async_publish.reset_mock() mqtt_mock.async_publish.reset_mock()
await hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_CLEAN_SPOT, {"entity_id": ENTITY_MATCH_ALL}, blocking=True vacuum.DOMAIN,
SERVICE_CLEAN_SPOT,
{"entity_id": ENTITY_MATCH_ALL},
blocking=True,
) )
mqtt_mock.async_publish.assert_not_called() mqtt_mock.async_publish.assert_not_called()
mqtt_mock.async_publish.reset_mock() mqtt_mock.async_publish.reset_mock()