270 lines
9.5 KiB
Python
270 lines
9.5 KiB
Python
"""The tests for the Demo Media player platform."""
|
|
import unittest
|
|
from unittest.mock import patch
|
|
import asyncio
|
|
|
|
import pytest
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.setup import setup_component, async_setup_component
|
|
import homeassistant.components.media_player as mp
|
|
from homeassistant.helpers.aiohttp_client import DATA_CLIENTSESSION
|
|
|
|
from tests.common import get_test_home_assistant
|
|
from tests.components.media_player import common
|
|
|
|
entity_id = "media_player.walkman"
|
|
|
|
|
|
class TestDemoMediaPlayer(unittest.TestCase):
|
|
"""Test the media_player module."""
|
|
|
|
def setUp(self): # pylint: disable=invalid-name
|
|
"""Set up things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
|
|
def tearDown(self):
|
|
"""Shut down test instance."""
|
|
self.hass.stop()
|
|
|
|
def test_source_select(self):
|
|
"""Test the input source service."""
|
|
entity_id = "media_player.lounge_room"
|
|
|
|
assert setup_component(
|
|
self.hass, mp.DOMAIN, {"media_player": {"platform": "demo"}}
|
|
)
|
|
state = self.hass.states.get(entity_id)
|
|
assert "dvd" == state.attributes.get("source")
|
|
|
|
with pytest.raises(vol.Invalid):
|
|
common.select_source(self.hass, None, entity_id)
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get(entity_id)
|
|
assert "dvd" == state.attributes.get("source")
|
|
|
|
common.select_source(self.hass, "xbox", entity_id)
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get(entity_id)
|
|
assert "xbox" == state.attributes.get("source")
|
|
|
|
def test_clear_playlist(self):
|
|
"""Test clear playlist."""
|
|
assert setup_component(
|
|
self.hass, mp.DOMAIN, {"media_player": {"platform": "demo"}}
|
|
)
|
|
assert self.hass.states.is_state(entity_id, "playing")
|
|
|
|
common.clear_playlist(self.hass, entity_id)
|
|
self.hass.block_till_done()
|
|
assert self.hass.states.is_state(entity_id, "off")
|
|
|
|
def test_volume_services(self):
|
|
"""Test the volume service."""
|
|
assert setup_component(
|
|
self.hass, mp.DOMAIN, {"media_player": {"platform": "demo"}}
|
|
)
|
|
state = self.hass.states.get(entity_id)
|
|
assert 1.0 == state.attributes.get("volume_level")
|
|
|
|
with pytest.raises(vol.Invalid):
|
|
common.set_volume_level(self.hass, None, entity_id)
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get(entity_id)
|
|
assert 1.0 == state.attributes.get("volume_level")
|
|
|
|
common.set_volume_level(self.hass, 0.5, entity_id)
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get(entity_id)
|
|
assert 0.5 == state.attributes.get("volume_level")
|
|
|
|
common.volume_down(self.hass, entity_id)
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get(entity_id)
|
|
assert 0.4 == state.attributes.get("volume_level")
|
|
|
|
common.volume_up(self.hass, entity_id)
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get(entity_id)
|
|
assert 0.5 == state.attributes.get("volume_level")
|
|
|
|
assert False is state.attributes.get("is_volume_muted")
|
|
|
|
with pytest.raises(vol.Invalid):
|
|
common.mute_volume(self.hass, None, entity_id)
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get(entity_id)
|
|
assert False is state.attributes.get("is_volume_muted")
|
|
|
|
common.mute_volume(self.hass, True, entity_id)
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get(entity_id)
|
|
assert True is state.attributes.get("is_volume_muted")
|
|
|
|
def test_turning_off_and_on(self):
|
|
"""Test turn_on and turn_off."""
|
|
assert setup_component(
|
|
self.hass, mp.DOMAIN, {"media_player": {"platform": "demo"}}
|
|
)
|
|
assert self.hass.states.is_state(entity_id, "playing")
|
|
|
|
common.turn_off(self.hass, entity_id)
|
|
self.hass.block_till_done()
|
|
assert self.hass.states.is_state(entity_id, "off")
|
|
assert not mp.is_on(self.hass, entity_id)
|
|
|
|
common.turn_on(self.hass, entity_id)
|
|
self.hass.block_till_done()
|
|
assert self.hass.states.is_state(entity_id, "playing")
|
|
|
|
common.toggle(self.hass, entity_id)
|
|
self.hass.block_till_done()
|
|
assert self.hass.states.is_state(entity_id, "off")
|
|
assert not mp.is_on(self.hass, entity_id)
|
|
|
|
def test_playing_pausing(self):
|
|
"""Test media_pause."""
|
|
assert setup_component(
|
|
self.hass, mp.DOMAIN, {"media_player": {"platform": "demo"}}
|
|
)
|
|
assert self.hass.states.is_state(entity_id, "playing")
|
|
|
|
common.media_pause(self.hass, entity_id)
|
|
self.hass.block_till_done()
|
|
assert self.hass.states.is_state(entity_id, "paused")
|
|
|
|
common.media_play_pause(self.hass, entity_id)
|
|
self.hass.block_till_done()
|
|
assert self.hass.states.is_state(entity_id, "playing")
|
|
|
|
common.media_play_pause(self.hass, entity_id)
|
|
self.hass.block_till_done()
|
|
assert self.hass.states.is_state(entity_id, "paused")
|
|
|
|
common.media_play(self.hass, entity_id)
|
|
self.hass.block_till_done()
|
|
assert self.hass.states.is_state(entity_id, "playing")
|
|
|
|
def test_prev_next_track(self):
|
|
"""Test media_next_track and media_previous_track ."""
|
|
assert setup_component(
|
|
self.hass, mp.DOMAIN, {"media_player": {"platform": "demo"}}
|
|
)
|
|
state = self.hass.states.get(entity_id)
|
|
assert 1 == state.attributes.get("media_track")
|
|
|
|
common.media_next_track(self.hass, entity_id)
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get(entity_id)
|
|
assert 2 == state.attributes.get("media_track")
|
|
|
|
common.media_next_track(self.hass, entity_id)
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get(entity_id)
|
|
assert 3 == state.attributes.get("media_track")
|
|
|
|
common.media_previous_track(self.hass, entity_id)
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get(entity_id)
|
|
assert 2 == state.attributes.get("media_track")
|
|
|
|
assert setup_component(
|
|
self.hass, mp.DOMAIN, {"media_player": {"platform": "demo"}}
|
|
)
|
|
ent_id = "media_player.lounge_room"
|
|
state = self.hass.states.get(ent_id)
|
|
assert 1 == state.attributes.get("media_episode")
|
|
|
|
common.media_next_track(self.hass, ent_id)
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get(ent_id)
|
|
assert 2 == state.attributes.get("media_episode")
|
|
|
|
common.media_previous_track(self.hass, ent_id)
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get(ent_id)
|
|
assert 1 == state.attributes.get("media_episode")
|
|
|
|
def test_play_media(self):
|
|
"""Test play_media ."""
|
|
assert setup_component(
|
|
self.hass, mp.DOMAIN, {"media_player": {"platform": "demo"}}
|
|
)
|
|
ent_id = "media_player.living_room"
|
|
state = self.hass.states.get(ent_id)
|
|
assert 0 < (mp.SUPPORT_PLAY_MEDIA & state.attributes.get("supported_features"))
|
|
assert state.attributes.get("media_content_id") is not None
|
|
|
|
with pytest.raises(vol.Invalid):
|
|
common.play_media(self.hass, None, "some_id", ent_id)
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get(ent_id)
|
|
assert 0 < (mp.SUPPORT_PLAY_MEDIA & state.attributes.get("supported_features"))
|
|
assert not "some_id" == state.attributes.get("media_content_id")
|
|
|
|
common.play_media(self.hass, "youtube", "some_id", ent_id)
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get(ent_id)
|
|
assert 0 < (mp.SUPPORT_PLAY_MEDIA & state.attributes.get("supported_features"))
|
|
assert "some_id" == state.attributes.get("media_content_id")
|
|
|
|
@patch(
|
|
"homeassistant.components.demo.media_player.DemoYoutubePlayer." "media_seek",
|
|
autospec=True,
|
|
)
|
|
def test_seek(self, mock_seek):
|
|
"""Test seek."""
|
|
assert setup_component(
|
|
self.hass, mp.DOMAIN, {"media_player": {"platform": "demo"}}
|
|
)
|
|
ent_id = "media_player.living_room"
|
|
state = self.hass.states.get(ent_id)
|
|
assert state.attributes["supported_features"] & mp.SUPPORT_SEEK
|
|
assert not mock_seek.called
|
|
with pytest.raises(vol.Invalid):
|
|
common.media_seek(self.hass, None, ent_id)
|
|
self.hass.block_till_done()
|
|
assert not mock_seek.called
|
|
common.media_seek(self.hass, 100, ent_id)
|
|
self.hass.block_till_done()
|
|
assert mock_seek.called
|
|
|
|
|
|
async def test_media_image_proxy(hass, hass_client):
|
|
"""Test the media server image proxy server ."""
|
|
assert await async_setup_component(
|
|
hass, mp.DOMAIN, {"media_player": {"platform": "demo"}}
|
|
)
|
|
|
|
fake_picture_data = "test.test"
|
|
|
|
class MockResponse:
|
|
def __init__(self):
|
|
self.status = 200
|
|
self.headers = {"Content-Type": "sometype"}
|
|
|
|
@asyncio.coroutine
|
|
def read(self):
|
|
return fake_picture_data.encode("ascii")
|
|
|
|
@asyncio.coroutine
|
|
def release(self):
|
|
pass
|
|
|
|
class MockWebsession:
|
|
@asyncio.coroutine
|
|
def get(self, url):
|
|
return MockResponse()
|
|
|
|
def detach(self):
|
|
pass
|
|
|
|
hass.data[DATA_CLIENTSESSION] = MockWebsession()
|
|
|
|
assert hass.states.is_state(entity_id, "playing")
|
|
state = hass.states.get(entity_id)
|
|
client = await hass_client()
|
|
req = await client.get(state.attributes.get("entity_picture"))
|
|
assert req.status == 200
|
|
assert await req.text() == fake_picture_data
|