evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Address late review comments for fan entity component test (#114425)

Browse source 
* Address late review comments for fan entity component test

* Update tests/components/fan/common.py

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>

---------

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
This commit is contained in:
Jan-Philipp Benecke 2024-03-29 10:18:57 +01:00 • committed by GitHub
parent 045dc3f1fb
commit 247ee6e4f0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

13
tests/components/fan/common.py
View file

@ -154,22 +154,11 @@ async def async_set_direction(
class MockFan(MockEntity, FanEntity):
"""Mock Fan class."""
@property
def preset_mode(self) -> str | None:
"""Return preset mode."""
return self._handle("preset_mode")
@property
def preset_modes(self) -> list[str] | None:
"""Return preset mode."""
return self._handle("preset_modes")
@property
def supported_features(self):
"""Return the class of this fan."""
return self._handle("supported_features")
async def async_set_preset_mode(self, preset_mode: str) -> None:
def set_preset_mode(self, preset_mode: str) -> None:
"""Set preset mode."""
self._attr_preset_mode = preset_mode
self.async_write_ha_state()