mirror of
https://github.com/home-assistant/core.git
synced 2025-05-05 14:39:16 +00:00
d21b8166f0
* Add subscription tier to Twitch integration.
* Add test for Twitch tiers. Tests do not currently pass, so this is only theoretical.
* Fix variable type
* Show tier levels as 1,2,3 instead of the raw API values of 1000,2000,3000.
* Make Twitch subscription tier fixtures strings.
* Use proper assertion value for subscription tier test.
Edited on a bus on my phone. 😎
* Update homeassistant/components/twitch/coordinator.py
* Update tests/components/twitch/test_sensor.py
---------
Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com>
98 lines
3.5 KiB
Python
98 lines
3.5 KiB
Python
"""The tests for an update of the Twitch component."""
|
|
|
|
from datetime import datetime
|
|
from unittest.mock import AsyncMock
|
|
|
|
from dateutil.tz import tzutc
|
|
from twitchAPI.object.api import FollowedChannel, Stream, UserSubscription
|
|
from twitchAPI.type import TwitchResourceNotFound
|
|
|
|
from homeassistant.components.twitch import DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from . import TwitchIterObject, get_generator_from_data, setup_integration
|
|
|
|
from tests.common import MockConfigEntry, load_json_object_fixture
|
|
|
|
ENTITY_ID = "sensor.channel123"
|
|
|
|
|
|
async def test_offline(
|
|
hass: HomeAssistant, twitch_mock: AsyncMock, config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test offline state."""
|
|
twitch_mock.return_value.get_streams.return_value = get_generator_from_data(
|
|
[], Stream
|
|
)
|
|
await setup_integration(hass, config_entry)
|
|
|
|
sensor_state = hass.states.get(ENTITY_ID)
|
|
assert sensor_state.state == "offline"
|
|
assert sensor_state.attributes["entity_picture"] == "logo.png"
|
|
|
|
|
|
async def test_streaming(
|
|
hass: HomeAssistant, twitch_mock: AsyncMock, config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test streaming state."""
|
|
await setup_integration(hass, config_entry)
|
|
|
|
sensor_state = hass.states.get(ENTITY_ID)
|
|
assert sensor_state.state == "streaming"
|
|
assert sensor_state.attributes["entity_picture"] == "stream-medium.png"
|
|
assert sensor_state.attributes["game"] == "Good game"
|
|
assert sensor_state.attributes["title"] == "Title"
|
|
assert sensor_state.attributes["started_at"] == datetime(
|
|
year=2021, month=3, day=10, hour=3, minute=18, second=11, tzinfo=tzutc()
|
|
)
|
|
assert sensor_state.attributes["viewers"] == 42
|
|
|
|
|
|
async def test_oauth_without_sub_and_follow(
|
|
hass: HomeAssistant, twitch_mock: AsyncMock, config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test state with oauth."""
|
|
twitch_mock.return_value.get_followed_channels.return_value = TwitchIterObject(
|
|
"empty_response.json", FollowedChannel
|
|
)
|
|
twitch_mock.return_value.check_user_subscription.side_effect = (
|
|
TwitchResourceNotFound
|
|
)
|
|
await setup_integration(hass, config_entry)
|
|
|
|
sensor_state = hass.states.get(ENTITY_ID)
|
|
assert sensor_state.attributes["subscribed"] is False
|
|
assert sensor_state.attributes["following"] is False
|
|
|
|
|
|
async def test_oauth_with_sub(
|
|
hass: HomeAssistant, twitch_mock: AsyncMock, config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test state with oauth and sub."""
|
|
twitch_mock.return_value.get_followed_channels.return_value = TwitchIterObject(
|
|
"empty_response.json", FollowedChannel
|
|
)
|
|
twitch_mock.return_value.check_user_subscription.return_value = UserSubscription(
|
|
**load_json_object_fixture("check_user_subscription_2.json", DOMAIN)
|
|
)
|
|
await setup_integration(hass, config_entry)
|
|
|
|
sensor_state = hass.states.get(ENTITY_ID)
|
|
assert sensor_state.attributes["subscribed"] is True
|
|
assert sensor_state.attributes["subscription_is_gifted"] is False
|
|
assert sensor_state.attributes["subscription_tier"] == 1
|
|
assert sensor_state.attributes["following"] is False
|
|
|
|
|
|
async def test_oauth_with_follow(
|
|
hass: HomeAssistant, twitch_mock: AsyncMock, config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test state with oauth and follow."""
|
|
await setup_integration(hass, config_entry)
|
|
|
|
sensor_state = hass.states.get(ENTITY_ID)
|
|
assert sensor_state.attributes["following"] is True
|
|
assert sensor_state.attributes["following_since"] == datetime(
|
|
year=2023, month=8, day=1
|
|
)
|