Quellcode durchsuchen

maint: allow buttons to be either config or diagnostic

Issue #4020
Jason Rumney vor 2 Monaten
Ursprung
Commit
00463ff5e7
1 geänderte Dateien mit 6 neuen und 0 gelöschten Zeilen
  1. 6 0
      tests/devices/base_device_tests.py

+ 6 - 0
tests/devices/base_device_tests.py

@@ -124,6 +124,12 @@ class TuyaDeviceTestCase(IsolatedAsyncioTestCase):
                         EntityCategory.DIAGNOSTIC,
                         msg=f"{k} is {e.entity_category.value}, expected diagnostic",
                     )
+                elif type(e) is TuyaLocalButton:
+                    self.assertIn(
+                        e.entity_category,
+                        [EntityCategory.CONFIG, EntityCategory.DIAGNOSTIC],
+                        msg=f"{k} is unsupported {e.entity_category.value}",
+                    )
                 else:
                     self.assertEqual(
                         e.entity_category,