|
18 | 18 | from zigpy.quirks.v2 import CustomDeviceV2, QuirkBuilder
|
19 | 19 | import zigpy.types as t
|
20 | 20 | from zigpy.zcl.clusters import closures, general
|
21 |
| -from zigpy.zcl.clusters.general import OnOff |
22 | 21 | from zigpy.zcl.clusters.manufacturer_specific import ManufacturerSpecificCluster
|
23 | 22 | import zigpy.zcl.foundation as zcl_f
|
24 | 23 |
|
@@ -851,52 +850,3 @@ async def test_cover_inversion_switch_not_created(zha_gateway: Gateway) -> None:
|
851 | 850 | # entity should not be created when mode or config status aren't present
|
852 | 851 | with pytest.raises(KeyError):
|
853 | 852 | get_entity(zha_device, platform=Platform.SWITCH)
|
854 |
| - |
855 |
| - |
856 |
| -async def test_quirks_switch_attr_converter(zha_gateway: Gateway) -> None: |
857 |
| - """Test ZHA quirks v2 switch with attribute_converter.""" |
858 |
| - |
859 |
| - registry = DeviceRegistry() |
860 |
| - zigpy_dev = create_mock_zigpy_device( |
861 |
| - zha_gateway, |
862 |
| - { |
863 |
| - 1: { |
864 |
| - SIG_EP_INPUT: [ |
865 |
| - general.Basic.cluster_id, |
866 |
| - general.OnOff.cluster_id, |
867 |
| - ], |
868 |
| - SIG_EP_OUTPUT: [], |
869 |
| - SIG_EP_TYPE: zha.DeviceType.SIMPLE_SENSOR, |
870 |
| - } |
871 |
| - }, |
872 |
| - manufacturer="manufacturer", |
873 |
| - model="model", |
874 |
| - ) |
875 |
| - |
876 |
| - ( |
877 |
| - QuirkBuilder(zigpy_dev.manufacturer, zigpy_dev.model, registry=registry) |
878 |
| - .switch( |
879 |
| - OnOff.AttributeDefs.on_off.name, |
880 |
| - OnOff.cluster_id, |
881 |
| - translation_key="on_off_quirks", |
882 |
| - fallback_name="On/off", |
883 |
| - attribute_converter=lambda x: not bool(x), # invert value with lambda |
884 |
| - ) |
885 |
| - .add_to_registry() |
886 |
| - ) |
887 |
| - |
888 |
| - zigpy_device_ = registry.get_device(zigpy_dev) |
889 |
| - |
890 |
| - assert isinstance(zigpy_device_, CustomDeviceV2) |
891 |
| - cluster = zigpy_device_.endpoints[1].on_off |
892 |
| - |
893 |
| - zha_device = await join_zigpy_device(zha_gateway, zigpy_device_) |
894 |
| - entity = get_entity(zha_device, platform=Platform.SWITCH, qualifier="on_off") |
895 |
| - assert entity.translation_key == "on_off_quirks" |
896 |
| - |
897 |
| - # send updated value, check if the value is inverted |
898 |
| - await send_attributes_report(zha_gateway, cluster, {"on_off": 1}) |
899 |
| - assert entity.state["state"] is False |
900 |
| - |
901 |
| - await send_attributes_report(zha_gateway, cluster, {"on_off": 0}) |
902 |
| - assert entity.state["state"] is True |
0 commit comments