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()