Files
localthings/tests/test_entities.py
T
Marc Billow b67e0e5cf5 Rename ocf/ to registry/, it's a capability registry now
The DTLS/CoAP transport code that made "ocf" an accurate name moved out to
the smartthings-local package. What's left here (capability.py, entities.py,
discovery.py, adapter.py, identity.py, capabilities/, by_type/, plus the
/device/0 batch parser) is entirely the device capability registry, so name
the package for what it does.

Flattened the redundant ocf/registry/ nesting into a single top-level
registry/ package and updated every import across the platform modules and
test suite accordingly. Verified: full test suite (80/80) passes, and the
Docker dev container reconnects to both live appliances and rediscovers
their entities cleanly after the rename.
2026-07-07 22:54:57 -05:00

35 lines
1.3 KiB
Python

from custom_components.localthings.registry.entities import (
SensorDesc, BinarySensorDesc, SelectDesc, SwitchDesc, ButtonDesc,
NumberDesc, PLATFORM_OF,
)
def test_value_fn_defaults_to_identity():
d = SensorDesc(key='power', field='x.com.samsung.da.instantaneousPower')
assert d.value_fn(42) == 42
def test_descriptions_are_frozen():
d = SensorDesc(key='power', field='f')
try:
d.key = 'other'
except Exception as e:
assert 'frozen' in str(type(e)).lower() or 'cannot' in str(e).lower()
else:
raise AssertionError("expected frozen dataclass")
def test_platform_mapping_covers_all_subclasses():
assert PLATFORM_OF[SensorDesc] == 'sensor'
assert PLATFORM_OF[BinarySensorDesc] == 'binary_sensor'
assert PLATFORM_OF[SelectDesc] == 'select'
assert PLATFORM_OF[SwitchDesc] == 'switch'
assert PLATFORM_OF[ButtonDesc] == 'button'
assert PLATFORM_OF[NumberDesc] == 'number'
def test_select_carries_options_and_write_fn():
d = SelectDesc(key='sound', field='mode', options=('voice', 'tone', 'mute'),
write_fn=lambda p, rep: (['settings', 'sound', 'mode', 'vs', '0'], {'mode': p}))
assert d.write_fn('tone', {}) == (['settings', 'sound', 'mode', 'vs', '0'], {'mode': 'tone'})