From 779e020dc439e2cfbc9c3b02bc11043f23ba3500 Mon Sep 17 00:00:00 2001
From: Brett Adams <Bre77@users.noreply.github.com>
Date: Mon, 29 Aug 2022 14:41:37 +1000
Subject: [PATCH] Add update platform to Advantage Air (#75391)

Co-authored-by: J. Nick Koston <nick@koston.org>
---
 .../components/advantage_air/__init__.py      |  1 +
 .../components/advantage_air/update.py        | 52 +++++++++++++++++++
 .../advantage_air/fixtures/getSystemData.json |  3 +-
 .../advantage_air/fixtures/needsUpdate.json   | 49 +++++++++++++++++
 tests/components/advantage_air/test_update.py | 28 ++++++++++
 5 files changed, 132 insertions(+), 1 deletion(-)
 create mode 100644 homeassistant/components/advantage_air/update.py
 create mode 100644 tests/components/advantage_air/fixtures/needsUpdate.json
 create mode 100644 tests/components/advantage_air/test_update.py

diff --git a/homeassistant/components/advantage_air/__init__.py b/homeassistant/components/advantage_air/__init__.py
index b5e6e0be024..3e07a8fbcef 100644
--- a/homeassistant/components/advantage_air/__init__.py
+++ b/homeassistant/components/advantage_air/__init__.py
@@ -20,6 +20,7 @@ PLATFORMS = [
     Platform.SELECT,
     Platform.SENSOR,
     Platform.SWITCH,
+    Platform.UPDATE,
     Platform.LIGHT,
 ]
 
diff --git a/homeassistant/components/advantage_air/update.py b/homeassistant/components/advantage_air/update.py
new file mode 100644
index 00000000000..9294ecab238
--- /dev/null
+++ b/homeassistant/components/advantage_air/update.py
@@ -0,0 +1,52 @@
+"""Advantage Air Update platform."""
+from homeassistant.components.update import UpdateEntity
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity import DeviceInfo
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+
+from .const import DOMAIN as ADVANTAGE_AIR_DOMAIN
+from .entity import AdvantageAirEntity
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    config_entry: ConfigEntry,
+    async_add_entities: AddEntitiesCallback,
+) -> None:
+    """Set up AdvantageAir update platform."""
+
+    instance = hass.data[ADVANTAGE_AIR_DOMAIN][config_entry.entry_id]
+
+    async_add_entities([AdvantageAirApp(instance)])
+
+
+class AdvantageAirApp(AdvantageAirEntity, UpdateEntity):
+    """Representation of Advantage Air App."""
+
+    _attr_name = "App"
+
+    def __init__(self, instance):
+        """Initialize the Advantage Air App."""
+        super().__init__(instance)
+        self._attr_device_info = DeviceInfo(
+            identifiers={
+                (ADVANTAGE_AIR_DOMAIN, self.coordinator.data["system"]["rid"])
+            },
+            manufacturer="Advantage Air",
+            model=self.coordinator.data["system"]["sysType"],
+            name=self.coordinator.data["system"]["name"],
+            sw_version=self.coordinator.data["system"]["myAppRev"],
+        )
+
+    @property
+    def installed_version(self):
+        """Return the current app version."""
+        return self.coordinator.data["system"]["myAppRev"]
+
+    @property
+    def latest_version(self):
+        """Return if there is an update."""
+        if self.coordinator.data["system"]["needsUpdate"]:
+            return "Needs Update"
+        return self.installed_version
diff --git a/tests/components/advantage_air/fixtures/getSystemData.json b/tests/components/advantage_air/fixtures/getSystemData.json
index 00ce2b1f095..327ee2d53d5 100644
--- a/tests/components/advantage_air/fixtures/getSystemData.json
+++ b/tests/components/advantage_air/fixtures/getSystemData.json
@@ -167,9 +167,10 @@
     "hasThings": false,
     "hasThingsBOG": false,
     "hasThingsLight": false,
+    "needsUpdate": false,
     "name": "testname",
     "rid": "uniqueid",
     "sysType": "e-zone",
-    "myAppRev": "testversion"
+    "myAppRev": "1.234"
   }
 }
diff --git a/tests/components/advantage_air/fixtures/needsUpdate.json b/tests/components/advantage_air/fixtures/needsUpdate.json
new file mode 100644
index 00000000000..584aa50cb77
--- /dev/null
+++ b/tests/components/advantage_air/fixtures/needsUpdate.json
@@ -0,0 +1,49 @@
+{
+  "aircons": {
+    "ac1": {
+      "info": {
+        "climateControlModeIsRunning": false,
+        "countDownToOff": 10,
+        "countDownToOn": 0,
+        "fan": "high",
+        "filterCleanStatus": 0,
+        "freshAirStatus": "off",
+        "mode": "vent",
+        "myZone": 1,
+        "name": "AC One",
+        "setTemp": 24,
+        "state": "on"
+      },
+      "zones": {
+        "z01": {
+          "error": 0,
+          "maxDamper": 100,
+          "measuredTemp": 25,
+          "minDamper": 0,
+          "motion": 20,
+          "motionConfig": 2,
+          "name": "Zone open with Sensor",
+          "number": 1,
+          "rssi": 40,
+          "setTemp": 24,
+          "state": "open",
+          "type": 1,
+          "value": 100
+        }
+      }
+    }
+  },
+  "system": {
+    "hasAircons": false,
+    "hasLights": false,
+    "hasSensors": false,
+    "hasThings": false,
+    "hasThingsBOG": false,
+    "hasThingsLight": false,
+    "needsUpdate": true,
+    "name": "testname",
+    "rid": "uniqueid",
+    "sysType": "e-zone",
+    "myAppRev": "1.234"
+  }
+}
diff --git a/tests/components/advantage_air/test_update.py b/tests/components/advantage_air/test_update.py
new file mode 100644
index 00000000000..2fef887997f
--- /dev/null
+++ b/tests/components/advantage_air/test_update.py
@@ -0,0 +1,28 @@
+"""Test the Advantage Air Update Platform."""
+
+from homeassistant.const import STATE_ON
+from homeassistant.helpers import entity_registry as er
+
+from tests.common import load_fixture
+from tests.components.advantage_air import TEST_SYSTEM_URL, add_mock_config
+
+
+async def test_update_platform(hass, aioclient_mock):
+    """Test update platform."""
+
+    aioclient_mock.get(
+        TEST_SYSTEM_URL,
+        text=load_fixture("advantage_air/needsUpdate.json"),
+    )
+    await add_mock_config(hass)
+
+    registry = er.async_get(hass)
+
+    entity_id = "update.testname_app"
+    state = hass.states.get(entity_id)
+    assert state
+    assert state.state == STATE_ON
+
+    entry = registry.async_get(entity_id)
+    assert entry
+    assert entry.unique_id == "uniqueid"