Fix the build -- use er.RegistryEntryDisabler.INTEGRATION . (#245)

blakeblackshear · Apr 21, 2022 · 65e5ec0 · 65e5ec0
1 parent 4ec0eb3
commit 65e5ec0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_switch.py b/tests/test_switch.py
@@ -186,7 +186,7 @@ async def test_disabled_switch_can_be_enabled(
     entry = entity_registry.async_get(disabled_entity_id)
     assert entry
     assert entry.disabled
-    assert entry.disabled_by == er.DISABLED_INTEGRATION
+    assert entry.disabled_by == er.RegistryEntryDisabler.INTEGRATION
     entity_state = hass.states.get(disabled_entity_id)
     assert not entity_state