From 247ee6e4f01580cf90aeb2e834b1b4b226aa9a9f Mon Sep 17 00:00:00 2001 From: Jan-Philipp Benecke Date: Fri, 29 Mar 2024 10:18:57 +0100 Subject: [PATCH] Address late review comments for fan entity component test (#114425) * Address late review comments for fan entity component test * Update tests/components/fan/common.py Co-authored-by: Martin Hjelmare --------- Co-authored-by: Martin Hjelmare --- tests/components/fan/common.py | 13 +------------ 1 file changed, 1 insertion(+), 12 deletions(-) diff --git a/tests/components/fan/common.py b/tests/components/fan/common.py index e9919c865c8..a48e66c08f4 100644 --- a/tests/components/fan/common.py +++ b/tests/components/fan/common.py @@ -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()