fix: Let HA know the integration is , so that polling can be disabled…

… via the integration system options

CHANGELOG.md

@@ -1,3 +1,9 @@
+# v2.0.1
+
+## Fixes
+
+- Let HA know the integration is `local_polling`, so that polling can be disabled via the integration system options
+
 # v2.0.0
 
 ## New Features

custom_components/cololight/__init__.py

@@ -3,9 +3,10 @@
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.helpers.typing import HomeAssistantType
 from homeassistant.core import callback
-from homeassistant.const import CONF_NAME
+from homeassistant.const import CONF_NAME, Platform
 
 DOMAIN = "cololight"
+PLATFORMS = [Platform.LIGHT]
 
 
 async def async_setup(hass, config):
@@ -16,22 +17,25 @@ async def async_setup(hass, config):
 async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry):
     """Load the saved entities."""
 
-    entry.add_update_listener(update_listener)
+    entry.async_on_unload(entry.add_update_listener(update_listener))
+
+    await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
 
-    hass.async_create_task(
-        hass.config_entries.async_forward_entry_setup(entry, LIGHT_DOMAIN)
-    )
     return True
 
 
 async def async_unload_entry(hass, entry):
     """Unload a config entry."""
-    return await hass.config_entries.async_forward_entry_unload(entry, LIGHT_DOMAIN)
+
+    unload_ok = await hass.config_entries.async_forward_entry_unload(
+        entry, LIGHT_DOMAIN
+    )
+
+    if unload_ok:
+        hass.data[DOMAIN].pop(entry.entry_id)
+    return unload_ok
 
 
 async def update_listener(hass, entry):
     """Handle options update."""
-    await hass.config_entries.async_forward_entry_unload(entry, LIGHT_DOMAIN)
-    hass.async_create_task(
-        hass.config_entries.async_forward_entry_setup(entry, LIGHT_DOMAIN)
-    )
+    await hass.config_entries.async_reload(entry.entry_id)

custom_components/cololight/manifest.json

@@ -4,8 +4,9 @@
   "config_flow": true,
   "documentation": "https://github.com/BazaJayGee66/homeassistant_cololight",
   "issue_tracker": "https://github.com/BazaJayGee66/homeassistant_cololight/issues",
-  "version": "v2.0.0",
+  "version": "v2.0.1",
   "dependencies": [],
+  "iot_class": "local_polling",
   "codeowners": ["@BazaJayGee66"],
   "requirements": ["pycololight==2.0.0"]
 }

custom_components/tests/test_light.py

@@ -238,7 +238,6 @@ async def test_turn_off(mock_send, hass):
 
 
 async def test_light_has_effects(hass):
-
     state = hass.states.get(ENTITY_1_LIGHT)
 
     expected_efects_list = [
@@ -251,7 +250,6 @@ async def test_light_has_effects(hass):
 
 
 async def test_light_adds_custom_effect(hass):
-
     state = hass.states.get(ENTITY_2_LIGHT)
 
     expected_efects_list = [
@@ -265,7 +263,6 @@ async def test_light_adds_custom_effect(hass):
 
 
 async def test_light_handles_incorrect_custom_effect(hass):
-
     state = hass.states.get(ENTITY_3_LIGHT)
 
     expected_efects_list = [
@@ -278,7 +275,6 @@ async def test_light_handles_incorrect_custom_effect(hass):
 
 
 async def test_light_has_only_specified_default_effects(hass):
-
     state = hass.states.get(ENTITY_4_LIGHT)
 
     expected_efects_list = ["80s Club", "Custom Effect"]
@@ -287,7 +283,6 @@ async def test_light_has_only_specified_default_effects(hass):
 
 
 async def test_icon(hass):
-
     state_hexagon = hass.states.get(ENTITY_1_LIGHT)
     state_strip = hass.states.get(ENTITY_6_LIGHT)