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

Bump aiowebostv to 0.7.0 (#139145)

Browse Source
This commit is contained in:
Shay Levy 2025-02-24 09:52:53 +02:00 committed by GitHub
parent ea1045d826
commit 8c4b8028cf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
12 changed files with 117 additions and 109 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/components/webostv/config_flow.py
View File

@ -92,13 +92,13 @@ class FlowHandler(ConfigFlow, domain=DOMAIN):
errors["base"] = "cannot_connect" errors["base"] = "cannot_connect"
else: else:
await self.async_set_unique_id( await self.async_set_unique_id(
client.hello_info["deviceUUID"], raise_on_progress=False client.tv_info.hello["deviceUUID"], raise_on_progress=False
) )
self._abort_if_unique_id_configured({CONF_HOST: self._host}) self._abort_if_unique_id_configured({CONF_HOST: self._host})
data = {CONF_HOST: self._host, CONF_CLIENT_SECRET: client.client_key} data = {CONF_HOST: self._host, CONF_CLIENT_SECRET: client.client_key}
if not self._name: if not self._name:
self._name = f"{DEFAULT_NAME} {client.system_info['modelName']}" self._name = f"{DEFAULT_NAME} {client.tv_info.system['modelName']}"
return self.async_create_entry(title=self._name, data=data) return self.async_create_entry(title=self._name, data=data)
return self.async_show_form(step_id="pairing", errors=errors) return self.async_show_form(step_id="pairing", errors=errors)
@ -176,7 +176,7 @@ class FlowHandler(ConfigFlow, domain=DOMAIN):
except WEBOSTV_EXCEPTIONS: except WEBOSTV_EXCEPTIONS:
errors["base"] = "cannot_connect" errors["base"] = "cannot_connect"
else: else:
await self.async_set_unique_id(client.hello_info["deviceUUID"]) await self.async_set_unique_id(client.tv_info.hello["deviceUUID"])
self._abort_if_unique_id_mismatch(reason="wrong_device") self._abort_if_unique_id_mismatch(reason="wrong_device")
data = {CONF_HOST: host, CONF_CLIENT_SECRET: client.client_key} data = {CONF_HOST: host, CONF_CLIENT_SECRET: client.client_key}
return self.async_update_reload_and_abort(reconfigure_entry, data=data) return self.async_update_reload_and_abort(reconfigure_entry, data=data)
@ -214,7 +214,7 @@ class OptionsFlowHandler(OptionsFlow):
sources_list = [] sources_list = []
try: try:
client = await async_control_connect(self.hass, self.host, self.key) client = await async_control_connect(self.hass, self.host, self.key)
sources_list = get_sources(client) sources_list = get_sources(client.tv_state)
except WebOsTvPairError: except WebOsTvPairError:
errors["base"] = "error_pairing" errors["base"] = "error_pairing"
except WEBOSTV_EXCEPTIONS: except WEBOSTV_EXCEPTIONS:

18
homeassistant/components/webostv/diagnostics.py
View File

@ -32,15 +32,15 @@ async def async_get_config_entry_diagnostics(
client_data = { client_data = {
"is_registered": client.is_registered(), "is_registered": client.is_registered(),
"is_connected": client.is_connected(), "is_connected": client.is_connected(),
"current_app_id": client.current_app_id, "current_app_id": client.tv_state.current_app_id,
"current_channel": client.current_channel, "current_channel": client.tv_state.current_channel,
"apps": client.apps, "apps": client.tv_state.apps,
"inputs": client.inputs, "inputs": client.tv_state.inputs,
"system_info": client.system_info, "system_info": client.tv_info.system,
"software_info": client.software_info, "software_info": client.tv_info.software,
"hello_info": client.hello_info, "hello_info": client.tv_info.hello,
"sound_output": client.sound_output, "sound_output": client.tv_state.sound_output,
"is_on": client.is_on, "is_on": client.tv_state.is_on,
} }
return async_redact_data( return async_redact_data(

8
homeassistant/components/webostv/helpers.py
View File

@ -4,7 +4,7 @@ from __future__ import annotations
import logging import logging
from aiowebostv import WebOsClient from aiowebostv import WebOsClient, WebOsTvState
from homeassistant.config_entries import ConfigEntry, ConfigEntryState from homeassistant.config_entries import ConfigEntry, ConfigEntryState
from homeassistant.const import CONF_CLIENT_SECRET, CONF_HOST from homeassistant.const import CONF_CLIENT_SECRET, CONF_HOST
@ -83,16 +83,16 @@ def async_get_client_by_device_entry(
) )
def get_sources(client: WebOsClient) -> list[str]: def get_sources(tv_state: WebOsTvState) -> list[str]:
"""Construct sources list.""" """Construct sources list."""
sources = [] sources = []
found_live_tv = False found_live_tv = False
for app in client.apps.values(): for app in tv_state.apps.values():
sources.append(app["title"]) sources.append(app["title"])
if app["id"] == LIVE_TV_APP_ID: if app["id"] == LIVE_TV_APP_ID:
found_live_tv = True found_live_tv = True
for source in client.inputs.values(): for source in tv_state.inputs.values():
sources.append(source["label"]) sources.append(source["label"])
if source["appId"] == LIVE_TV_APP_ID: if source["appId"] == LIVE_TV_APP_ID:
found_live_tv = True found_live_tv = True

2
homeassistant/components/webostv/manifest.json
View File

@ -6,7 +6,7 @@
"documentation": "https://www.home-assistant.io/integrations/webostv", "documentation": "https://www.home-assistant.io/integrations/webostv",
"iot_class": "local_push", "iot_class": "local_push",
"loggers": ["aiowebostv"], "loggers": ["aiowebostv"],
"requirements": ["aiowebostv==0.6.2"], "requirements": ["aiowebostv==0.7.0"],
"ssdp": [ "ssdp": [
{ {
"st": "urn:lge-com:service:webos-second-screen:1" "st": "urn:lge-com:service:webos-second-screen:1"

67
homeassistant/components/webostv/media_player.py
View File

@ -11,7 +11,7 @@ from http import HTTPStatus
import logging import logging
from typing import Any, Concatenate, cast from typing import Any, Concatenate, cast
from aiowebostv import WebOsClient, WebOsTvPairError from aiowebostv import WebOsTvPairError, WebOsTvState
import voluptuous as vol import voluptuous as vol
from homeassistant import util from homeassistant import util
@ -205,51 +205,52 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
"""Call disconnect on removal.""" """Call disconnect on removal."""
self._client.unregister_state_update_callback(self.async_handle_state_update) self._client.unregister_state_update_callback(self.async_handle_state_update)
async def async_handle_state_update(self, _client: WebOsClient) -> None: async def async_handle_state_update(self, tv_state: WebOsTvState) -> None:
"""Update state from WebOsClient.""" """Update state from WebOsClient."""
self._update_states() self._update_states()
self.async_write_ha_state() self.async_write_ha_state()
def _update_states(self) -> None: def _update_states(self) -> None:
"""Update entity state attributes.""" """Update entity state attributes."""
tv_state = self._client.tv_state
self._update_sources() self._update_sources()
self._attr_state = ( self._attr_state = (
MediaPlayerState.ON if self._client.is_on else MediaPlayerState.OFF MediaPlayerState.ON if tv_state.is_on else MediaPlayerState.OFF
) )
self._attr_is_volume_muted = cast(bool, self._client.muted) self._attr_is_volume_muted = cast(bool, tv_state.muted)
self._attr_volume_level = None self._attr_volume_level = None
if self._client.volume is not None: if tv_state.volume is not None:
self._attr_volume_level = self._client.volume / 100.0 self._attr_volume_level = tv_state.volume / 100.0
self._attr_source = self._current_source self._attr_source = self._current_source
self._attr_source_list = sorted(self._source_list) self._attr_source_list = sorted(self._source_list)
self._attr_media_content_type = None self._attr_media_content_type = None
if self._client.current_app_id == LIVE_TV_APP_ID: if tv_state.current_app_id == LIVE_TV_APP_ID:
self._attr_media_content_type = MediaType.CHANNEL self._attr_media_content_type = MediaType.CHANNEL
self._attr_media_title = None self._attr_media_title = None
if (self._client.current_app_id == LIVE_TV_APP_ID) and ( if (tv_state.current_app_id == LIVE_TV_APP_ID) and (
self._client.current_channel is not None tv_state.current_channel is not None
): ):
self._attr_media_title = cast( self._attr_media_title = cast(
str, self._client.current_channel.get("channelName") str, tv_state.current_channel.get("channelName")
) )
self._attr_media_image_url = None self._attr_media_image_url = None
if self._client.current_app_id in self._client.apps: if tv_state.current_app_id in tv_state.apps:
icon: str = self._client.apps[self._client.current_app_id]["largeIcon"] icon: str = tv_state.apps[tv_state.current_app_id]["largeIcon"]
if not icon.startswith("http"): if not icon.startswith("http"):
icon = self._client.apps[self._client.current_app_id]["icon"] icon = tv_state.apps[tv_state.current_app_id]["icon"]
self._attr_media_image_url = icon self._attr_media_image_url = icon
if self.state != MediaPlayerState.OFF or not self._supported_features: if self.state != MediaPlayerState.OFF or not self._supported_features:
supported = SUPPORT_WEBOSTV supported = SUPPORT_WEBOSTV
if self._client.sound_output == "external_speaker": if tv_state.sound_output == "external_speaker":
supported = supported | SUPPORT_WEBOSTV_VOLUME supported = supported | SUPPORT_WEBOSTV_VOLUME
elif self._client.sound_output != "lineout": elif tv_state.sound_output != "lineout":
supported = ( supported = (
supported supported
| SUPPORT_WEBOSTV_VOLUME | SUPPORT_WEBOSTV_VOLUME
@ -265,9 +266,9 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
) )
self._attr_assumed_state = True self._attr_assumed_state = True
if self._client.is_on and self._client.media_state: if tv_state.is_on and tv_state.media_state:
self._attr_assumed_state = False self._attr_assumed_state = False
for entry in self._client.media_state: for entry in tv_state.media_state:
if entry.get("playState") == "playing": if entry.get("playState") == "playing":
self._attr_state = MediaPlayerState.PLAYING self._attr_state = MediaPlayerState.PLAYING
elif entry.get("playState") == "paused": elif entry.get("playState") == "paused":
@ -275,35 +276,37 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
elif entry.get("playState") == "unloaded": elif entry.get("playState") == "unloaded":
self._attr_state = MediaPlayerState.IDLE self._attr_state = MediaPlayerState.IDLE
tv_info = self._client.tv_info
if self.state != MediaPlayerState.OFF: if self.state != MediaPlayerState.OFF:
maj_v = self._client.software_info.get("major_ver") maj_v = tv_info.software.get("major_ver")
min_v = self._client.software_info.get("minor_ver") min_v = tv_info.software.get("minor_ver")
if maj_v and min_v: if maj_v and min_v:
self._attr_device_info["sw_version"] = f"{maj_v}.{min_v}" self._attr_device_info["sw_version"] = f"{maj_v}.{min_v}"
if model := self._client.system_info.get("modelName"): if model := tv_info.system.get("modelName"):
self._attr_device_info["model"] = model self._attr_device_info["model"] = model
if serial_number := self._client.system_info.get("serialNumber"): if serial_number := tv_info.system.get("serialNumber"):
self._attr_device_info["serial_number"] = serial_number self._attr_device_info["serial_number"] = serial_number
self._attr_extra_state_attributes = {} self._attr_extra_state_attributes = {}
if self._client.sound_output is not None or self.state != MediaPlayerState.OFF: if tv_state.sound_output is not None or self.state != MediaPlayerState.OFF:
self._attr_extra_state_attributes = { self._attr_extra_state_attributes = {
ATTR_SOUND_OUTPUT: self._client.sound_output ATTR_SOUND_OUTPUT: tv_state.sound_output
} }
def _update_sources(self) -> None: def _update_sources(self) -> None:
"""Update list of sources from current source, apps, inputs and configured list.""" """Update list of sources from current source, apps, inputs and configured list."""
tv_state = self._client.tv_state
source_list = self._source_list source_list = self._source_list
self._source_list = {} self._source_list = {}
conf_sources = self._sources conf_sources = self._sources
found_live_tv = False found_live_tv = False
for app in self._client.apps.values(): for app in tv_state.apps.values():
if app["id"] == LIVE_TV_APP_ID: if app["id"] == LIVE_TV_APP_ID:
found_live_tv = True found_live_tv = True
if app["id"] == self._client.current_app_id: if app["id"] == tv_state.current_app_id:
self._current_source = app["title"] self._current_source = app["title"]
self._source_list[app["title"]] = app self._source_list[app["title"]] = app
elif ( elif (
@ -314,10 +317,10 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
): ):
self._source_list[app["title"]] = app self._source_list[app["title"]] = app
for source in self._client.inputs.values(): for source in tv_state.inputs.values():
if source["appId"] == LIVE_TV_APP_ID: if source["appId"] == LIVE_TV_APP_ID:
found_live_tv = True found_live_tv = True
if source["appId"] == self._client.current_app_id: if source["appId"] == tv_state.current_app_id:
self._current_source = source["label"] self._current_source = source["label"]
self._source_list[source["label"]] = source self._source_list[source["label"]] = source
elif ( elif (
@ -334,7 +337,7 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
# not appear in the app or input lists in some cases # not appear in the app or input lists in some cases
elif not found_live_tv: elif not found_live_tv:
app = {"id": LIVE_TV_APP_ID, "title": "Live TV"} app = {"id": LIVE_TV_APP_ID, "title": "Live TV"}
if self._client.current_app_id == LIVE_TV_APP_ID: if tv_state.current_app_id == LIVE_TV_APP_ID:
self._current_source = app["title"] self._current_source = app["title"]
self._source_list["Live TV"] = app self._source_list["Live TV"] = app
elif ( elif (
@ -434,12 +437,12 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
"""Play a piece of media.""" """Play a piece of media."""
_LOGGER.debug("Call play media type <%s>, Id <%s>", media_type, media_id) _LOGGER.debug("Call play media type <%s>, Id <%s>", media_type, media_id)
if media_type == MediaType.CHANNEL and self._client.channels: if media_type == MediaType.CHANNEL and self._client.tv_state.channels:
_LOGGER.debug("Searching channel") _LOGGER.debug("Searching channel")
partial_match_channel_id = None partial_match_channel_id = None
perfect_match_channel_id = None perfect_match_channel_id = None
for channel in self._client.channels: for channel in self._client.tv_state.channels:
if media_id == channel["channelNumber"]: if media_id == channel["channelNumber"]:
perfect_match_channel_id = channel["channelId"] perfect_match_channel_id = channel["channelId"]
continue continue
@ -484,7 +487,7 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
@cmd @cmd
async def async_media_next_track(self) -> None: async def async_media_next_track(self) -> None:
"""Send next track command.""" """Send next track command."""
if self._client.current_app_id == LIVE_TV_APP_ID: if self._client.tv_state.current_app_id == LIVE_TV_APP_ID:
await self._client.channel_up() await self._client.channel_up()
else: else:
await self._client.fast_forward() await self._client.fast_forward()
@ -492,7 +495,7 @@ class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):
@cmd @cmd
async def async_media_previous_track(self) -> None: async def async_media_previous_track(self) -> None:
"""Send the previous track command.""" """Send the previous track command."""
if self._client.current_app_id == LIVE_TV_APP_ID: if self._client.tv_state.current_app_id == LIVE_TV_APP_ID:
await self._client.channel_down() await self._client.channel_down()
else: else:
await self._client.rewind() await self._client.rewind()

2
homeassistant/components/webostv/notify.py
View File

@ -49,7 +49,7 @@ class LgWebOSNotificationService(BaseNotificationService):
data = kwargs[ATTR_DATA] data = kwargs[ATTR_DATA]
icon_path = data.get(ATTR_ICON) if data else None icon_path = data.get(ATTR_ICON) if data else None
if not client.is_on: if not client.tv_state.is_on:
raise HomeAssistantError( raise HomeAssistantError(
translation_domain=DOMAIN, translation_domain=DOMAIN,
translation_key="notify_device_off", translation_key="notify_device_off",

2
requirements_all.txt generated
View File

@ -422,7 +422,7 @@ aiowaqi==3.1.0
aiowatttime==0.1.1 aiowatttime==0.1.1
# homeassistant.components.webostv # homeassistant.components.webostv
aiowebostv==0.6.2 aiowebostv==0.7.0
# homeassistant.components.withings # homeassistant.components.withings
aiowithings==3.1.5 aiowithings==3.1.5

2
requirements_test_all.txt generated
View File

@ -404,7 +404,7 @@ aiowaqi==3.1.0
aiowatttime==0.1.1 aiowatttime==0.1.1
# homeassistant.components.webostv # homeassistant.components.webostv
aiowebostv==0.6.2 aiowebostv==0.7.0
# homeassistant.components.withings # homeassistant.components.withings
aiowithings==3.1.5 aiowithings==3.1.5

33
tests/components/webostv/conftest.py
View File

@ -3,6 +3,7 @@
from collections.abc import Generator from collections.abc import Generator
from unittest.mock import AsyncMock, Mock, patch from unittest.mock import AsyncMock, Mock, patch
from aiowebostv import WebOsTvInfo, WebOsTvState
import pytest import pytest
from homeassistant.components.webostv.const import LIVE_TV_APP_ID from homeassistant.components.webostv.const import LIVE_TV_APP_ID
@ -40,26 +41,30 @@ def client_fixture():
), ),
): ):
client = mock_client_class.return_value client = mock_client_class.return_value
client.hello_info = {"deviceUUID": FAKE_UUID} client.tv_info = WebOsTvInfo(
client.software_info = {"major_ver": "major", "minor_ver": "minor"} hello={"deviceUUID": FAKE_UUID},
client.system_info = {"modelName": TV_MODEL, "serialNumber": "1234567890"} system={"modelName": TV_MODEL, "serialNumber": "1234567890"},
software={"major_ver": "major", "minor_ver": "minor"},
)
client.client_key = CLIENT_KEY client.client_key = CLIENT_KEY
client.apps = MOCK_APPS client.tv_state = WebOsTvState(
client.inputs = MOCK_INPUTS apps=MOCK_APPS,
client.current_app_id = LIVE_TV_APP_ID inputs=MOCK_INPUTS,
current_app_id=LIVE_TV_APP_ID,
channels=[CHANNEL_1, CHANNEL_2],
current_channel=CHANNEL_1,
volume=37,
sound_output="speaker",
muted=False,
is_on=True,
media_state=[{"playState": ""}],
)
client.channels = [CHANNEL_1, CHANNEL_2]
client.current_channel = CHANNEL_1
client.volume = 37
client.sound_output = "speaker"
client.muted = False
client.is_on = True
client.is_registered = Mock(return_value=True) client.is_registered = Mock(return_value=True)
client.is_connected = Mock(return_value=True) client.is_connected = Mock(return_value=True)
async def mock_state_update_callback(): async def mock_state_update_callback():
await client.register_state_update_callback.call_args[0][0](client) await client.register_state_update_callback.call_args[0][0](client.tv_state)
client.mock_state_update = AsyncMock(side_effect=mock_state_update_callback) client.mock_state_update = AsyncMock(side_effect=mock_state_update_callback)

6
tests/components/webostv/test_config_flow.py
View File

@ -84,8 +84,8 @@ async def test_options_flow_live_tv_in_apps(
hass: HomeAssistant, client, apps, inputs hass: HomeAssistant, client, apps, inputs
) -> None: ) -> None:
"""Test options config flow Live TV found in apps.""" """Test options config flow Live TV found in apps."""
client.apps = apps client.tv_state.apps = apps
client.inputs = inputs client.tv_state.inputs = inputs
entry = await setup_webostv(hass) entry = await setup_webostv(hass)
result = await hass.config_entries.options.async_init(entry.entry_id) result = await hass.config_entries.options.async_init(entry.entry_id)
@ -411,7 +411,7 @@ async def test_reconfigure_wrong_device(hass: HomeAssistant, client) -> None:
assert result["type"] is FlowResultType.FORM assert result["type"] is FlowResultType.FORM
assert result["step_id"] == "reconfigure" assert result["step_id"] == "reconfigure"
client.hello_info = {"deviceUUID": "wrong_uuid"} client.tv_info.hello = {"deviceUUID": "wrong_uuid"}
result = await hass.config_entries.flow.async_configure( result = await hass.config_entries.flow.async_configure(
result["flow_id"], result["flow_id"],
user_input={CONF_HOST: "new_host"}, user_input={CONF_HOST: "new_host"},

76
tests/components/webostv/test_media_player.py
View File

@ -156,7 +156,7 @@ async def test_media_next_previous_track(
getattr(client, client_call[1]).assert_called_once() getattr(client, client_call[1]).assert_called_once()
# check next/previous for not Live TV channels # check next/previous for not Live TV channels
client.current_app_id = "in1" client.tv_state.current_app_id = "in1"
data = {ATTR_ENTITY_ID: ENTITY_ID} data = {ATTR_ENTITY_ID: ENTITY_ID}
await hass.services.async_call(MP_DOMAIN, service, data, True) await hass.services.async_call(MP_DOMAIN, service, data, True)
@ -303,8 +303,8 @@ async def test_device_info_startup_off(
hass: HomeAssistant, client, device_registry: dr.DeviceRegistry hass: HomeAssistant, client, device_registry: dr.DeviceRegistry
) -> None: ) -> None:
"""Test device info when device is off at startup.""" """Test device info when device is off at startup."""
client.system_info = None client.tv_info.system = {}
client.is_on = False client.tv_state.is_on = False
entry = await setup_webostv(hass) entry = await setup_webostv(hass)
await client.mock_state_update() await client.mock_state_update()
@ -335,14 +335,14 @@ async def test_entity_attributes(
assert state == snapshot(exclude=props("entity_picture")) assert state == snapshot(exclude=props("entity_picture"))
# Volume level not available # Volume level not available
client.volume = None client.tv_state.volume = None
await client.mock_state_update() await client.mock_state_update()
attrs = hass.states.get(ENTITY_ID).attributes attrs = hass.states.get(ENTITY_ID).attributes
assert attrs.get(ATTR_MEDIA_VOLUME_LEVEL) is None assert attrs.get(ATTR_MEDIA_VOLUME_LEVEL) is None
# Channel change # Channel change
client.current_channel = CHANNEL_2 client.tv_state.current_channel = CHANNEL_2
await client.mock_state_update() await client.mock_state_update()
attrs = hass.states.get(ENTITY_ID).attributes attrs = hass.states.get(ENTITY_ID).attributes
@ -353,8 +353,8 @@ async def test_entity_attributes(
assert device == snapshot assert device == snapshot
# Sound output when off # Sound output when off
client.sound_output = None client.tv_state.sound_output = None
client.is_on = False client.tv_state.is_on = False
await client.mock_state_update() await client.mock_state_update()
state = hass.states.get(ENTITY_ID) state = hass.states.get(ENTITY_ID)
@ -410,13 +410,13 @@ async def test_update_sources_live_tv_find(hass: HomeAssistant, client) -> None:
assert len(sources) == 3 assert len(sources) == 3
# Live TV is current app # Live TV is current app
client.apps = { client.tv_state.apps = {
LIVE_TV_APP_ID: { LIVE_TV_APP_ID: {
"title": "Live TV", "title": "Live TV",
"id": "some_id", "id": "some_id",
}, },
} }
client.current_app_id = "some_id" client.tv_state.current_app_id = "some_id"
await client.mock_state_update() await client.mock_state_update()
sources = hass.states.get(ENTITY_ID).attributes[ATTR_INPUT_SOURCE_LIST] sources = hass.states.get(ENTITY_ID).attributes[ATTR_INPUT_SOURCE_LIST]
@ -424,7 +424,7 @@ async def test_update_sources_live_tv_find(hass: HomeAssistant, client) -> None:
assert len(sources) == 3 assert len(sources) == 3
# Live TV is is in inputs # Live TV is is in inputs
client.inputs = { client.tv_state.inputs = {
LIVE_TV_APP_ID: { LIVE_TV_APP_ID: {
"label": "Live TV", "label": "Live TV",
"id": "some_id", "id": "some_id",
@ -438,7 +438,7 @@ async def test_update_sources_live_tv_find(hass: HomeAssistant, client) -> None:
assert len(sources) == 1 assert len(sources) == 1
# Live TV is current input # Live TV is current input
client.inputs = { client.tv_state.inputs = {
LIVE_TV_APP_ID: { LIVE_TV_APP_ID: {
"label": "Live TV", "label": "Live TV",
"id": "some_id", "id": "some_id",
@ -452,7 +452,7 @@ async def test_update_sources_live_tv_find(hass: HomeAssistant, client) -> None:
assert len(sources) == 1 assert len(sources) == 1
# Live TV not found # Live TV not found
client.current_app_id = "other_id" client.tv_state.current_app_id = "other_id"
await client.mock_state_update() await client.mock_state_update()
sources = hass.states.get(ENTITY_ID).attributes[ATTR_INPUT_SOURCE_LIST] sources = hass.states.get(ENTITY_ID).attributes[ATTR_INPUT_SOURCE_LIST]
@ -460,8 +460,8 @@ async def test_update_sources_live_tv_find(hass: HomeAssistant, client) -> None:
assert len(sources) == 1 assert len(sources) == 1
# Live TV not found in sources/apps but is current app # Live TV not found in sources/apps but is current app
client.apps = {} client.tv_state.apps = {}
client.current_app_id = LIVE_TV_APP_ID client.tv_state.current_app_id = LIVE_TV_APP_ID
await client.mock_state_update() await client.mock_state_update()
sources = hass.states.get(ENTITY_ID).attributes[ATTR_INPUT_SOURCE_LIST] sources = hass.states.get(ENTITY_ID).attributes[ATTR_INPUT_SOURCE_LIST]
@ -469,7 +469,7 @@ async def test_update_sources_live_tv_find(hass: HomeAssistant, client) -> None:
assert len(sources) == 1 assert len(sources) == 1
# Bad update, keep old update # Bad update, keep old update
client.inputs = {} client.tv_state.inputs = {}
await client.mock_state_update() await client.mock_state_update()
sources = hass.states.get(ENTITY_ID).attributes[ATTR_INPUT_SOURCE_LIST] sources = hass.states.get(ENTITY_ID).attributes[ATTR_INPUT_SOURCE_LIST]
@ -543,7 +543,7 @@ async def test_control_error_handling(
"""Test control errors handling.""" """Test control errors handling."""
await setup_webostv(hass) await setup_webostv(hass)
client.play.side_effect = exception client.play.side_effect = exception
client.is_on = is_on client.tv_state.is_on = is_on
await client.mock_state_update() await client.mock_state_update()
data = {ATTR_ENTITY_ID: ENTITY_ID} data = {ATTR_ENTITY_ID: ENTITY_ID}
@ -566,7 +566,7 @@ async def test_turn_off_when_device_is_off(hass: HomeAssistant, client) -> None:
async def test_supported_features(hass: HomeAssistant, client) -> None: async def test_supported_features(hass: HomeAssistant, client) -> None:
"""Test test supported features.""" """Test test supported features."""
client.sound_output = "lineout" client.tv_state.sound_output = "lineout"
await setup_webostv(hass) await setup_webostv(hass)
await client.mock_state_update() await client.mock_state_update()
@ -577,7 +577,7 @@ async def test_supported_features(hass: HomeAssistant, client) -> None:
assert attrs[ATTR_SUPPORTED_FEATURES] == supported assert attrs[ATTR_SUPPORTED_FEATURES] == supported
# Support volume mute, step # Support volume mute, step
client.sound_output = "external_speaker" client.tv_state.sound_output = "external_speaker"
await client.mock_state_update() await client.mock_state_update()
supported = supported | SUPPORT_WEBOSTV_VOLUME supported = supported | SUPPORT_WEBOSTV_VOLUME
attrs = hass.states.get(ENTITY_ID).attributes attrs = hass.states.get(ENTITY_ID).attributes
@ -585,7 +585,7 @@ async def test_supported_features(hass: HomeAssistant, client) -> None:
assert attrs[ATTR_SUPPORTED_FEATURES] == supported assert attrs[ATTR_SUPPORTED_FEATURES] == supported
# Support volume mute, step, set # Support volume mute, step, set
client.sound_output = "speaker" client.tv_state.sound_output = "speaker"
await client.mock_state_update() await client.mock_state_update()
supported = supported | SUPPORT_WEBOSTV_VOLUME | MediaPlayerEntityFeature.VOLUME_SET supported = supported | SUPPORT_WEBOSTV_VOLUME | MediaPlayerEntityFeature.VOLUME_SET
attrs = hass.states.get(ENTITY_ID).attributes attrs = hass.states.get(ENTITY_ID).attributes
@ -623,8 +623,8 @@ async def test_supported_features(hass: HomeAssistant, client) -> None:
async def test_cached_supported_features(hass: HomeAssistant, client) -> None: async def test_cached_supported_features(hass: HomeAssistant, client) -> None:
"""Test test supported features.""" """Test test supported features."""
client.is_on = False client.tv_state.is_on = False
client.sound_output = None client.tv_state.sound_output = None
supported = ( supported = (
SUPPORT_WEBOSTV | SUPPORT_WEBOSTV_VOLUME | MediaPlayerEntityFeature.TURN_ON SUPPORT_WEBOSTV | SUPPORT_WEBOSTV_VOLUME | MediaPlayerEntityFeature.TURN_ON
) )
@ -652,8 +652,8 @@ async def test_cached_supported_features(hass: HomeAssistant, client) -> None:
) )
# TV on, support volume mute, step # TV on, support volume mute, step
client.is_on = True client.tv_state.is_on = True
client.sound_output = "external_speaker" client.tv_state.sound_output = "external_speaker"
await client.mock_state_update() await client.mock_state_update()
supported = SUPPORT_WEBOSTV | SUPPORT_WEBOSTV_VOLUME supported = SUPPORT_WEBOSTV | SUPPORT_WEBOSTV_VOLUME
@ -662,8 +662,8 @@ async def test_cached_supported_features(hass: HomeAssistant, client) -> None:
assert attrs[ATTR_SUPPORTED_FEATURES] == supported assert attrs[ATTR_SUPPORTED_FEATURES] == supported
# TV off, support volume mute, step # TV off, support volume mute, step
client.is_on = False client.tv_state.is_on = False
client.sound_output = None client.tv_state.sound_output = None
await client.mock_state_update() await client.mock_state_update()
supported = SUPPORT_WEBOSTV | SUPPORT_WEBOSTV_VOLUME supported = SUPPORT_WEBOSTV | SUPPORT_WEBOSTV_VOLUME
@ -672,8 +672,8 @@ async def test_cached_supported_features(hass: HomeAssistant, client) -> None:
assert attrs[ATTR_SUPPORTED_FEATURES] == supported assert attrs[ATTR_SUPPORTED_FEATURES] == supported
# TV on, support volume mute, step, set # TV on, support volume mute, step, set
client.is_on = True client.tv_state.is_on = True
client.sound_output = "speaker" client.tv_state.sound_output = "speaker"
await client.mock_state_update() await client.mock_state_update()
supported = ( supported = (
@ -684,8 +684,8 @@ async def test_cached_supported_features(hass: HomeAssistant, client) -> None:
assert attrs[ATTR_SUPPORTED_FEATURES] == supported assert attrs[ATTR_SUPPORTED_FEATURES] == supported
# TV off, support volume mute, step, set # TV off, support volume mute, step, set
client.is_on = False client.tv_state.is_on = False
client.sound_output = None client.tv_state.sound_output = None
await client.mock_state_update() await client.mock_state_update()
supported = ( supported = (
@ -728,8 +728,8 @@ async def test_cached_supported_features(hass: HomeAssistant, client) -> None:
async def test_supported_features_no_cache(hass: HomeAssistant, client) -> None: async def test_supported_features_no_cache(hass: HomeAssistant, client) -> None:
"""Test supported features if device is off and no cache.""" """Test supported features if device is off and no cache."""
client.is_on = False client.tv_state.is_on = False
client.sound_output = None client.tv_state.sound_output = None
await setup_webostv(hass) await setup_webostv(hass)
supported = ( supported = (
@ -772,7 +772,7 @@ async def test_get_image_http(
) -> None: ) -> None:
"""Test get image via http.""" """Test get image via http."""
url = "http://something/valid_icon" url = "http://something/valid_icon"
client.apps[LIVE_TV_APP_ID]["icon"] = url client.tv_state.apps[LIVE_TV_APP_ID]["icon"] = url
await setup_webostv(hass) await setup_webostv(hass)
await client.mock_state_update() await client.mock_state_update()
@ -797,7 +797,7 @@ async def test_get_image_http_error(
) -> None: ) -> None:
"""Test get image via http error.""" """Test get image via http error."""
url = "http://something/icon_error" url = "http://something/icon_error"
client.apps[LIVE_TV_APP_ID]["icon"] = url client.tv_state.apps[LIVE_TV_APP_ID]["icon"] = url
await setup_webostv(hass) await setup_webostv(hass)
await client.mock_state_update() await client.mock_state_update()
@ -823,7 +823,7 @@ async def test_get_image_https(
) -> None: ) -> None:
"""Test get image via http.""" """Test get image via http."""
url = "https://something/valid_icon_https" url = "https://something/valid_icon_https"
client.apps[LIVE_TV_APP_ID]["icon"] = url client.tv_state.apps[LIVE_TV_APP_ID]["icon"] = url
await setup_webostv(hass) await setup_webostv(hass)
await client.mock_state_update() await client.mock_state_update()
@ -871,18 +871,18 @@ async def test_update_media_state(hass: HomeAssistant, client) -> None:
"""Test updating media state.""" """Test updating media state."""
await setup_webostv(hass) await setup_webostv(hass)
client.media_state = [{"playState": "playing"}] client.tv_state.media_state = [{"playState": "playing"}]
await client.mock_state_update() await client.mock_state_update()
assert hass.states.get(ENTITY_ID).state == MediaPlayerState.PLAYING assert hass.states.get(ENTITY_ID).state == MediaPlayerState.PLAYING
client.media_state = [{"playState": "paused"}] client.tv_state.media_state = [{"playState": "paused"}]
await client.mock_state_update() await client.mock_state_update()
assert hass.states.get(ENTITY_ID).state == MediaPlayerState.PAUSED assert hass.states.get(ENTITY_ID).state == MediaPlayerState.PAUSED
client.media_state = [{"playState": "unloaded"}] client.tv_state.media_state = [{"playState": "unloaded"}]
await client.mock_state_update() await client.mock_state_update()
assert hass.states.get(ENTITY_ID).state == MediaPlayerState.IDLE assert hass.states.get(ENTITY_ID).state == MediaPlayerState.IDLE
client.is_on = False client.tv_state.is_on = False
await client.mock_state_update() await client.mock_state_update()
assert hass.states.get(ENTITY_ID).state == STATE_OFF assert hass.states.get(ENTITY_ID).state == STATE_OFF

2
tests/components/webostv/test_notify.py
View File

@ -104,7 +104,7 @@ async def test_errors(
) -> None: ) -> None:
"""Test error scenarios.""" """Test error scenarios."""
await setup_webostv(hass) await setup_webostv(hass)
client.is_on = is_on client.tv_state.is_on = is_on
assert hass.services.has_service("notify", SERVICE_NAME) assert hass.services.has_service("notify", SERVICE_NAME)