jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-26 10:17:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add input_button support to Google Assistant (#62593)

Browse Source
This commit is contained in:
Franck Nijhof 2021-12-23 07:38:35 +01:00 committed by GitHub
parent 99b2161365
commit dc47cbd01b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 27 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
homeassistant/components/google_assistant/const.py
View File

@ -10,6 +10,7 @@ from homeassistant.components import (
group,
humidifier,
input_boolean,
input_button,
input_select,
light,
lock,
@ -129,6 +130,7 @@ DOMAIN_TO_GOOGLE_TYPES = {
group.DOMAIN: TYPE_SWITCH,
humidifier.DOMAIN: TYPE_HUMIDIFIER,
input_boolean.DOMAIN: TYPE_SWITCH,
input_button.DOMAIN: TYPE_SCENE,
input_select.DOMAIN: TYPE_SENSOR,
light.DOMAIN: TYPE_LIGHT,
lock.DOMAIN: TYPE_LOCK,

13
homeassistant/components/google_assistant/trait.py
View File

@ -12,6 +12,7 @@ from homeassistant.components import (
fan,
group,
input_boolean,
input_button,
input_select,
light,
lock,
@ -514,7 +515,12 @@ class SceneTrait(_Trait):
@staticmethod
def supported(domain, features, device_class, _):
"""Test if state is supported."""
return domain in (button.DOMAIN, scene.DOMAIN, script.DOMAIN)
return domain in (
button.DOMAIN,
input_button.DOMAIN,
scene.DOMAIN,
script.DOMAIN,
)
def sync_attributes(self):
"""Return scene attributes for a sync request."""
@ -530,6 +536,8 @@ class SceneTrait(_Trait):
service = SERVICE_TURN_ON
if self.state.domain == button.DOMAIN:
service = button.SERVICE_PRESS
elif self.state.domain == input_button.DOMAIN:
service = input_button.SERVICE_PRESS
# Don't block for scripts or buttons, as they can be slow.
await self.hass.services.async_call(
@ -537,7 +545,8 @@ class SceneTrait(_Trait):
service,
{ATTR_ENTITY_ID: self.state.entity_id},
blocking=(not self.config.should_report_state)
and self.state.domain not in (button.DOMAIN, script.DOMAIN),
and self.state.domain
not in (button.DOMAIN, input_button.DOMAIN, script.DOMAIN),
context=data.context,
)

21
tests/components/google_assistant/test_trait.py
View File

@ -13,6 +13,7 @@ from homeassistant.components import (
fan,
group,
input_boolean,
input_button,
input_select,
light,
lock,
@ -768,24 +769,30 @@ async def test_light_modes(hass):
}
async def test_scene_button(hass):
"""Test Scene trait support for the button domain."""
assert helpers.get_google_type(button.DOMAIN, None) is not None
assert trait.SceneTrait.supported(button.DOMAIN, 0, None, None)
@pytest.mark.parametrize(
"component",
[button, input_button],
)
async def test_scene_button(hass, component):
"""Test Scene trait support for the (input) button domain."""
assert helpers.get_google_type(component.DOMAIN, None) is not None
assert trait.SceneTrait.supported(component.DOMAIN, 0, None, None)
trt = trait.SceneTrait(hass, State("button.bla", STATE_UNKNOWN), BASIC_CONFIG)
trt = trait.SceneTrait(
hass, State(f"{component.DOMAIN}.bla", STATE_UNKNOWN), BASIC_CONFIG
)
assert trt.sync_attributes() == {}
assert trt.query_attributes() == {}
assert trt.can_execute(trait.COMMAND_ACTIVATE_SCENE, {})
calls = async_mock_service(hass, button.DOMAIN, button.SERVICE_PRESS)
calls = async_mock_service(hass, component.DOMAIN, component.SERVICE_PRESS)
await trt.execute(trait.COMMAND_ACTIVATE_SCENE, BASIC_DATA, {}, {})
# We don't wait till button press is done.
await hass.async_block_till_done()
assert len(calls) == 1
assert calls[0].data == {ATTR_ENTITY_ID: "button.bla"}
assert calls[0].data == {ATTR_ENTITY_ID: f"{component.DOMAIN}.bla"}
async def test_scene_scene(hass):