Ensure entity platform in vacuum tests (#135786)

This commit is contained in:
G Johansson 2025-01-16 23:27:54 +01:00 committed by GitHub
parent 88c3be4ecf
commit e5164496cf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -33,6 +33,7 @@ from .common import async_start
from tests.common import ( from tests.common import (
MockConfigEntry, MockConfigEntry,
MockEntity, MockEntity,
MockEntityPlatform,
MockModule, MockModule,
help_test_all, help_test_all,
import_and_test_deprecated_constant_enum, import_and_test_deprecated_constant_enum,
@ -288,6 +289,8 @@ async def test_supported_features_compat(hass: HomeAssistant) -> None:
_attr_fan_speed_list = ["silent", "normal", "pet hair"] _attr_fan_speed_list = ["silent", "normal", "pet hair"]
entity = _LegacyConstantsStateVacuum() entity = _LegacyConstantsStateVacuum()
entity.hass = hass
entity.platform = MockEntityPlatform(hass)
assert isinstance(entity.supported_features, int) assert isinstance(entity.supported_features, int)
assert entity.supported_features == int(features) assert entity.supported_features == int(features)
assert entity.supported_features_compat is ( assert entity.supported_features_compat is (