add to test

dmulcahey · Oct 4, 2024 · 8b06ddf · 8b06ddf
1 parent 9947942
commit 8b06ddf
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/test_quirks_v2.py b/tests/test_quirks_v2.py
@@ -230,10 +230,17 @@ async def test_quirks_v2_quirk_builder_cloning(device_mock):
     cloned = base.clone()
     base.add_to_registry()
 
-    cloned.applies_to(device_mock.manufacturer, device_mock.model).add_to_registry()
+    (
+        cloned.adds(PowerConfiguration.cluster_id)
+        .applies_to(device_mock.manufacturer, device_mock.model)
+        .add_to_registry()
+    )
 
     quirked = registry.get_device(device_mock)
     assert isinstance(quirked, CustomDeviceV2)
+    assert (
+        quirked.endpoints[1].in_clusters.get(PowerConfiguration.cluster_id) is not None
+    )
 
 
 async def test_quirks_v2_signature_match(device_mock):