57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
"""The tests for the LG webOS media player platform."""
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
from homeassistant.components.webostv import media_player as webostv
|
|
|
|
|
|
class FakeLgWebOSDevice(webostv.LgWebOSDevice):
|
|
"""A fake device without the client setup required for the real one."""
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
"""Initialise parameters needed for tests with fake values."""
|
|
self._source_list = {}
|
|
self._client = mock.MagicMock()
|
|
self._name = "fake_device"
|
|
self._current_source = None
|
|
|
|
|
|
class TestLgWebOSDevice(unittest.TestCase):
|
|
"""Test the LgWebOSDevice class."""
|
|
|
|
def setUp(self):
|
|
"""Configure a fake device for each test."""
|
|
self.device = FakeLgWebOSDevice()
|
|
|
|
def test_select_source_with_empty_source_list(self):
|
|
"""Ensure we don't call client methods when we don't have sources."""
|
|
self.device.select_source("nonexistent")
|
|
assert 0 == self.device._client.launch_app.call_count
|
|
assert 0 == self.device._client.set_input.call_count
|
|
|
|
def test_select_source_with_titled_entry(self):
|
|
"""Test that a titled source is treated as an app."""
|
|
self.device._source_list = {
|
|
"existent": {"id": "existent_id", "title": "existent_title"}
|
|
}
|
|
|
|
self.device.select_source("existent")
|
|
|
|
assert "existent_title" == self.device._current_source
|
|
assert [mock.call("existent_id")] == (
|
|
self.device._client.launch_app.call_args_list
|
|
)
|
|
|
|
def test_select_source_with_labelled_entry(self):
|
|
"""Test that a labelled source is treated as an input source."""
|
|
self.device._source_list = {
|
|
"existent": {"id": "existent_id", "label": "existent_label"}
|
|
}
|
|
|
|
self.device.select_source("existent")
|
|
|
|
assert "existent_label" == self.device._current_source
|
|
assert [mock.call("existent_id")] == (
|
|
self.device._client.set_input.call_args_list
|
|
)
|