Add HomeKit Television functionality (#22968)
This commit is contained in:
Austin Drummond
cdce8p
parent
8da600adff
commit
e24d56aa5b
5 changed files with 447 additions and 18 deletions
|
|
@ -59,10 +59,6 @@ def test_customize_options(config, name):
|
|||
('Fan', 'fan.test', 'on', {}, {}),
|
||||
('Light', 'light.test', 'on', {}, {}),
|
||||
('Lock', 'lock.test', 'locked', {}, {ATTR_CODE: '1234'}),
|
||||
('MediaPlayer', 'media_player.test', 'on',
|
||||
{ATTR_SUPPORTED_FEATURES: media_player_c.SUPPORT_TURN_ON |
|
||||
media_player_c.SUPPORT_TURN_OFF}, {CONF_FEATURE_LIST:
|
||||
{FEATURE_ON_OFF: None}}),
|
||||
('SecuritySystem', 'alarm_control_panel.test', 'armed_away', {},
|
||||
{ATTR_CODE: '1234'}),
|
||||
('Thermostat', 'climate.test', 'auto', {}, {}),
|
||||
|
|
@ -101,6 +97,26 @@ def test_type_covers(type_name, entity_id, state, attrs):
|
|||
assert mock_type.called
|
||||
|
||||
|
||||
@pytest.mark.parametrize('type_name, entity_id, state, attrs, config', [
|
||||
('MediaPlayer', 'media_player.test', 'on',
|
||||
{ATTR_SUPPORTED_FEATURES: media_player_c.SUPPORT_TURN_ON |
|
||||
media_player_c.SUPPORT_TURN_OFF}, {CONF_FEATURE_LIST:
|
||||
{FEATURE_ON_OFF: None}}),
|
||||
('TelevisionMediaPlayer', 'media_player.tv', 'on',
|
||||
{ATTR_DEVICE_CLASS: 'tv'}, {}),
|
||||
])
|
||||
def test_type_media_player(type_name, entity_id, state, attrs, config):
|
||||
"""Test if media_player types are associated correctly."""
|
||||
mock_type = Mock()
|
||||
with patch.dict(TYPES, {type_name: mock_type}):
|
||||
entity_state = State(entity_id, state, attrs)
|
||||
get_accessory(None, None, entity_state, 2, config)
|
||||
assert mock_type.called
|
||||
|
||||
if config:
|
||||
assert mock_type.call_args[0][-1] == config
|
||||
|
||||
|
||||
@pytest.mark.parametrize('type_name, entity_id, state, attrs', [
|
||||
('BinarySensor', 'binary_sensor.opening', 'on',
|
||||
{ATTR_DEVICE_CLASS: 'opening'}),
|
||||
|
|
|
|||
|
|
@ -3,12 +3,15 @@
|
|||
from homeassistant.components.homekit.const import (
|
||||
ATTR_VALUE, CONF_FEATURE_LIST, FEATURE_ON_OFF, FEATURE_PLAY_PAUSE,
|
||||
FEATURE_PLAY_STOP, FEATURE_TOGGLE_MUTE)
|
||||
from homeassistant.components.homekit.type_media_players import MediaPlayer
|
||||
from homeassistant.components.media_player import DEVICE_CLASS_TV
|
||||
from homeassistant.components.homekit.type_media_players import (
|
||||
MediaPlayer, TelevisionMediaPlayer)
|
||||
from homeassistant.components.media_player.const import (
|
||||
ATTR_INPUT_SOURCE, ATTR_INPUT_SOURCE_LIST, ATTR_MEDIA_VOLUME_LEVEL,
|
||||
ATTR_MEDIA_VOLUME_MUTED, DOMAIN)
|
||||
from homeassistant.const import (
|
||||
ATTR_ENTITY_ID, ATTR_SUPPORTED_FEATURES, STATE_IDLE, STATE_OFF, STATE_ON,
|
||||
STATE_PAUSED, STATE_PLAYING)
|
||||
ATTR_DEVICE_CLASS, ATTR_ENTITY_ID, ATTR_SUPPORTED_FEATURES, STATE_IDLE,
|
||||
STATE_OFF, STATE_ON, STATE_PAUSED, STATE_PLAYING)
|
||||
|
||||
from tests.common import async_mock_service
|
||||
|
||||
|
|
@ -129,3 +132,183 @@ async def test_media_player_set_state(hass, hk_driver, events):
|
|||
assert call_toggle_mute[1].data[ATTR_MEDIA_VOLUME_MUTED] is False
|
||||
assert len(events) == 8
|
||||
assert events[-1].data[ATTR_VALUE] is None
|
||||
|
||||
|
||||
async def test_media_player_television(hass, hk_driver, events, caplog):
|
||||
"""Test if television accessory and HA are updated accordingly."""
|
||||
entity_id = 'media_player.television'
|
||||
|
||||
# Supports 'select_source', 'volume_step', 'turn_on', 'turn_off',
|
||||
# 'volume_mute', 'volume_set', 'pause'
|
||||
hass.states.async_set(entity_id, None, {
|
||||
ATTR_DEVICE_CLASS: DEVICE_CLASS_TV, ATTR_SUPPORTED_FEATURES: 3469,
|
||||
ATTR_MEDIA_VOLUME_MUTED: False, ATTR_INPUT_SOURCE_LIST: [
|
||||
'HDMI 1', 'HDMI 2', 'HDMI 3', 'HDMI 4']})
|
||||
await hass.async_block_till_done()
|
||||
acc = TelevisionMediaPlayer(hass, hk_driver, 'MediaPlayer', entity_id, 2,
|
||||
None)
|
||||
await hass.async_add_job(acc.run)
|
||||
|
||||
assert acc.aid == 2
|
||||
assert acc.category == 31 # Television
|
||||
|
||||
assert acc.char_active.value == 0
|
||||
assert acc.char_remote_key.value == 0
|
||||
assert acc.char_input_source.value == 0
|
||||
assert acc.char_mute.value is False
|
||||
|
||||
hass.states.async_set(entity_id, STATE_ON, {ATTR_MEDIA_VOLUME_MUTED: True})
|
||||
await hass.async_block_till_done()
|
||||
assert acc.char_active.value == 1
|
||||
assert acc.char_mute.value is True
|
||||
|
||||
hass.states.async_set(entity_id, STATE_OFF)
|
||||
await hass.async_block_till_done()
|
||||
assert acc.char_active.value == 0
|
||||
|
||||
hass.states.async_set(entity_id, STATE_ON, {ATTR_INPUT_SOURCE: 'HDMI 2'})
|
||||
await hass.async_block_till_done()
|
||||
assert acc.char_input_source.value == 1
|
||||
|
||||
hass.states.async_set(entity_id, STATE_ON, {ATTR_INPUT_SOURCE: 'HDMI 3'})
|
||||
await hass.async_block_till_done()
|
||||
assert acc.char_input_source.value == 2
|
||||
|
||||
hass.states.async_set(entity_id, STATE_ON, {ATTR_INPUT_SOURCE: 'HDMI 5'})
|
||||
await hass.async_block_till_done()
|
||||
assert acc.char_input_source.value == 0
|
||||
assert caplog.records[-2].levelname == 'WARNING'
|
||||
|
||||
# Set from HomeKit
|
||||
call_turn_on = async_mock_service(hass, DOMAIN, 'turn_on')
|
||||
call_turn_off = async_mock_service(hass, DOMAIN, 'turn_off')
|
||||
call_media_play = async_mock_service(hass, DOMAIN, 'media_play')
|
||||
call_media_pause = async_mock_service(hass, DOMAIN, 'media_pause')
|
||||
call_media_play_pause = async_mock_service(hass, DOMAIN,
|
||||
'media_play_pause')
|
||||
call_toggle_mute = async_mock_service(hass, DOMAIN, 'volume_mute')
|
||||
call_select_source = async_mock_service(hass, DOMAIN, 'select_source')
|
||||
call_volume_up = async_mock_service(hass, DOMAIN, 'volume_up')
|
||||
call_volume_down = async_mock_service(hass, DOMAIN, 'volume_down')
|
||||
call_volume_set = async_mock_service(hass, DOMAIN, 'volume_set')
|
||||
|
||||
await hass.async_add_job(acc.char_active.client_update_value, 1)
|
||||
await hass.async_block_till_done()
|
||||
assert call_turn_on
|
||||
assert call_turn_on[0].data[ATTR_ENTITY_ID] == entity_id
|
||||
assert len(events) == 1
|
||||
assert events[-1].data[ATTR_VALUE] is None
|
||||
|
||||
await hass.async_add_job(acc.char_active.client_update_value, 0)
|
||||
await hass.async_block_till_done()
|
||||
assert call_turn_off
|
||||
assert call_turn_off[0].data[ATTR_ENTITY_ID] == entity_id
|
||||
assert len(events) == 2
|
||||
assert events[-1].data[ATTR_VALUE] is None
|
||||
|
||||
await hass.async_add_job(acc.char_remote_key.client_update_value, 11)
|
||||
await hass.async_block_till_done()
|
||||
assert call_media_play_pause
|
||||
assert call_media_play_pause[0].data[ATTR_ENTITY_ID] == entity_id
|
||||
assert len(events) == 3
|
||||
assert events[-1].data[ATTR_VALUE] is None
|
||||
|
||||
hass.states.async_set(entity_id, STATE_PLAYING)
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_add_job(acc.char_remote_key.client_update_value, 11)
|
||||
await hass.async_block_till_done()
|
||||
assert call_media_pause
|
||||
assert call_media_pause[0].data[ATTR_ENTITY_ID] == entity_id
|
||||
assert len(events) == 4
|
||||
assert events[-1].data[ATTR_VALUE] is None
|
||||
|
||||
await hass.async_add_job(acc.char_remote_key.client_update_value, 10)
|
||||
await hass.async_block_till_done()
|
||||
assert len(events) == 4
|
||||
assert events[-1].data[ATTR_VALUE] is None
|
||||
|
||||
hass.states.async_set(entity_id, STATE_PAUSED)
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_add_job(acc.char_remote_key.client_update_value, 11)
|
||||
await hass.async_block_till_done()
|
||||
assert call_media_play
|
||||
assert call_media_play[0].data[ATTR_ENTITY_ID] == entity_id
|
||||
assert len(events) == 5
|
||||
assert events[-1].data[ATTR_VALUE] is None
|
||||
|
||||
await hass.async_add_job(acc.char_mute.client_update_value, True)
|
||||
await hass.async_block_till_done()
|
||||
assert call_toggle_mute
|
||||
assert call_toggle_mute[0].data[ATTR_ENTITY_ID] == entity_id
|
||||
assert call_toggle_mute[0].data[ATTR_MEDIA_VOLUME_MUTED] is True
|
||||
assert len(events) == 6
|
||||
assert events[-1].data[ATTR_VALUE] is None
|
||||
|
||||
await hass.async_add_job(acc.char_mute.client_update_value, False)
|
||||
await hass.async_block_till_done()
|
||||
assert call_toggle_mute
|
||||
assert call_toggle_mute[1].data[ATTR_ENTITY_ID] == entity_id
|
||||
assert call_toggle_mute[1].data[ATTR_MEDIA_VOLUME_MUTED] is False
|
||||
assert len(events) == 7
|
||||
assert events[-1].data[ATTR_VALUE] is None
|
||||
|
||||
await hass.async_add_job(acc.char_input_source.client_update_value, 1)
|
||||
await hass.async_block_till_done()
|
||||
assert call_select_source
|
||||
assert call_select_source[0].data[ATTR_ENTITY_ID] == entity_id
|
||||
assert call_select_source[0].data[ATTR_INPUT_SOURCE] == 'HDMI 2'
|
||||
assert len(events) == 8
|
||||
assert events[-1].data[ATTR_VALUE] is None
|
||||
|
||||
await hass.async_add_job(acc.char_volume_selector.client_update_value, 0)
|
||||
await hass.async_block_till_done()
|
||||
assert call_volume_up
|
||||
assert call_volume_up[0].data[ATTR_ENTITY_ID] == entity_id
|
||||
assert len(events) == 9
|
||||
assert events[-1].data[ATTR_VALUE] is None
|
||||
|
||||
await hass.async_add_job(acc.char_volume_selector.client_update_value, 1)
|
||||
await hass.async_block_till_done()
|
||||
assert call_volume_down
|
||||
assert call_volume_down[0].data[ATTR_ENTITY_ID] == entity_id
|
||||
assert len(events) == 10
|
||||
assert events[-1].data[ATTR_VALUE] is None
|
||||
|
||||
await hass.async_add_job(acc.char_volume.client_update_value, 20)
|
||||
await hass.async_block_till_done()
|
||||
assert call_volume_set[0]
|
||||
assert call_volume_set[0].data[ATTR_ENTITY_ID] == entity_id
|
||||
assert call_volume_set[0].data[ATTR_MEDIA_VOLUME_LEVEL] == 20
|
||||
assert len(events) == 11
|
||||
assert events[-1].data[ATTR_VALUE] is None
|
||||
|
||||
|
||||
async def test_media_player_television_basic(hass, hk_driver, events, caplog):
|
||||
"""Test if basic television accessory and HA are updated accordingly."""
|
||||
entity_id = 'media_player.television'
|
||||
|
||||
# Supports turn_on', 'turn_off'
|
||||
hass.states.async_set(entity_id, None, {
|
||||
ATTR_DEVICE_CLASS: DEVICE_CLASS_TV, ATTR_SUPPORTED_FEATURES: 384})
|
||||
await hass.async_block_till_done()
|
||||
acc = TelevisionMediaPlayer(hass, hk_driver, 'MediaPlayer', entity_id, 2,
|
||||
None)
|
||||
await hass.async_add_job(acc.run)
|
||||
|
||||
assert acc.chars_tv == []
|
||||
assert acc.chars_speaker == []
|
||||
assert acc.support_select_source is False
|
||||
|
||||
hass.states.async_set(entity_id, STATE_ON, {ATTR_MEDIA_VOLUME_MUTED: True})
|
||||
await hass.async_block_till_done()
|
||||
assert acc.char_active.value == 1
|
||||
|
||||
hass.states.async_set(entity_id, STATE_OFF)
|
||||
await hass.async_block_till_done()
|
||||
assert acc.char_active.value == 0
|
||||
|
||||
hass.states.async_set(entity_id, STATE_ON, {ATTR_INPUT_SOURCE: 'HDMI 3'})
|
||||
await hass.async_block_till_done()
|
||||
assert acc.char_active.value == 1
|
||||
|
||||
assert 'Error' not in caplog.messages[-1]
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue