84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
"""Constants and mock for the twkinly component tests."""
|
|
|
|
from uuid import uuid4
|
|
|
|
from aiohttp.client_exceptions import ClientConnectionError
|
|
|
|
from homeassistant.components.twinkly.const import DEV_NAME
|
|
|
|
TEST_HOST = "test.twinkly.com"
|
|
TEST_ID = "twinkly_test_device_id"
|
|
TEST_NAME = "twinkly_test_device_name"
|
|
TEST_NAME_ORIGINAL = "twinkly_test_original_device_name" # the original (deprecated) name stored in the conf
|
|
TEST_MODEL = "twinkly_test_device_model"
|
|
|
|
|
|
class ClientMock:
|
|
"""A mock of the ttls.client.Twinkly."""
|
|
|
|
def __init__(self) -> None:
|
|
"""Create a mocked client."""
|
|
self.is_offline = False
|
|
self.state = True
|
|
self.brightness = {"mode": "enabled", "value": 10}
|
|
self.color = None
|
|
|
|
self.id = str(uuid4())
|
|
self.device_info = {
|
|
"uuid": self.id,
|
|
"device_name": self.id, # we make sure that entity id is different for each test
|
|
"product_code": TEST_MODEL,
|
|
}
|
|
|
|
@property
|
|
def host(self) -> str:
|
|
"""Get the mocked host."""
|
|
return TEST_HOST
|
|
|
|
async def get_details(self):
|
|
"""Get the mocked device info."""
|
|
if self.is_offline:
|
|
raise ClientConnectionError()
|
|
return self.device_info
|
|
|
|
async def is_on(self) -> bool:
|
|
"""Get the mocked on/off state."""
|
|
if self.is_offline:
|
|
raise ClientConnectionError()
|
|
return self.state
|
|
|
|
async def turn_on(self) -> None:
|
|
"""Set the mocked on state."""
|
|
if self.is_offline:
|
|
raise ClientConnectionError()
|
|
self.state = True
|
|
|
|
async def turn_off(self) -> None:
|
|
"""Set the mocked off state."""
|
|
if self.is_offline:
|
|
raise ClientConnectionError()
|
|
self.state = False
|
|
|
|
async def get_brightness(self) -> int:
|
|
"""Get the mocked brightness."""
|
|
if self.is_offline:
|
|
raise ClientConnectionError()
|
|
return self.brightness
|
|
|
|
async def set_brightness(self, brightness: int) -> None:
|
|
"""Set the mocked brightness."""
|
|
if self.is_offline:
|
|
raise ClientConnectionError()
|
|
self.brightness = {"mode": "enabled", "value": brightness}
|
|
|
|
def change_name(self, new_name: str) -> None:
|
|
"""Change the name of this virtual device."""
|
|
self.device_info[DEV_NAME] = new_name
|
|
|
|
async def set_static_colour(self, colour) -> None:
|
|
"""Set static color."""
|
|
self.color = colour
|
|
|
|
async def interview(self) -> None:
|
|
"""Interview."""
|