8d6d70d6b5
* Use `setup_test_component_platform` helper for select entity component tests instead of `hass.components` * Use _values instead of _attr_current_option * Clean up * Set default current_option for second mock entity --------- Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
23 lines
672 B
Python
23 lines
672 B
Python
"""Common helpers for select entity component tests."""
|
|
|
|
from homeassistant.components.select import SelectEntity
|
|
|
|
from tests.common import MockEntity
|
|
|
|
|
|
class MockSelectEntity(MockEntity, SelectEntity):
|
|
"""Mock Select class."""
|
|
|
|
@property
|
|
def current_option(self):
|
|
"""Return the current option of this select."""
|
|
return self._handle("current_option")
|
|
|
|
@property
|
|
def options(self) -> list:
|
|
"""Return the list of available options of this select."""
|
|
return self._handle("options")
|
|
|
|
def select_option(self, option: str) -> None:
|
|
"""Change the selected option."""
|
|
self._values["current_option"] = option
|