From 108e2ec1ba54e8ffb311d96c41aba28c39b397c5 Mon Sep 17 00:00:00 2001
From: Chris Talkington <chris@talkingtontech.com>
Date: Sun, 30 Aug 2020 07:42:31 -0500
Subject: [PATCH] Update roku to use CoordinatorEntity (#39405)

---
 homeassistant/components/roku/__init__.py | 31 +++++------------------
 1 file changed, 7 insertions(+), 24 deletions(-)

diff --git a/homeassistant/components/roku/__init__.py b/homeassistant/components/roku/__init__.py
index b02ad6430e9..663823d71b4 100644
--- a/homeassistant/components/roku/__init__.py
+++ b/homeassistant/components/roku/__init__.py
@@ -15,9 +15,12 @@ from homeassistant.const import ATTR_NAME, CONF_HOST
 from homeassistant.exceptions import ConfigEntryNotReady
 from homeassistant.helpers import config_validation as cv
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from homeassistant.helpers.entity import Entity
 from homeassistant.helpers.typing import HomeAssistantType
-from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+from homeassistant.helpers.update_coordinator import (
+    CoordinatorEntity,
+    DataUpdateCoordinator,
+    UpdateFailed,
+)
 from homeassistant.util.dt import utcnow
 
 from .const import (
@@ -149,42 +152,22 @@ class RokuDataUpdateCoordinator(DataUpdateCoordinator[Device]):
             raise UpdateFailed(f"Invalid response from API: {error}") from error
 
 
-class RokuEntity(Entity):
+class RokuEntity(CoordinatorEntity):
     """Defines a base Roku entity."""
 
     def __init__(
         self, *, device_id: str, name: str, coordinator: RokuDataUpdateCoordinator
     ) -> None:
         """Initialize the Roku entity."""
+        super().__init__(coordinator)
         self._device_id = device_id
         self._name = name
-        self.coordinator = coordinator
-
-    @property
-    def available(self) -> bool:
-        """Return True if entity is available."""
-        return self.coordinator.last_update_success
 
     @property
     def name(self) -> str:
         """Return the name of the entity."""
         return self._name
 
-    @property
-    def should_poll(self) -> bool:
-        """Return the polling requirement of the entity."""
-        return False
-
-    async def async_added_to_hass(self) -> None:
-        """Connect to dispatcher listening for entity data notifications."""
-        self.async_on_remove(
-            self.coordinator.async_add_listener(self.async_write_ha_state)
-        )
-
-    async def async_update(self) -> None:
-        """Update an Roku entity."""
-        await self.coordinator.async_request_refresh()
-
     @property
     def device_info(self) -> Dict[str, Any]:
         """Return device information about this Roku device."""