update entity name in tests

home-assistant · Jun 28, 2023 · 8abbc18 · 8abbc18
1 parent bf8a84f
commit 8abbc18
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 16 deletions.
diff --git a/homeassistant/components/openhome/update.py b/homeassistant/components/openhome/update.py
@@ -78,20 +78,16 @@ async def async_update(self) -> None:
             self._attr_release_url = None
             return
 
-        self._attr_installed_version = software_status["current_software"][
-            "version"
-        ]
+        self._attr_installed_version = software_status["current_software"]["version"]
 
         if software_status["status"] == "update_available":
-            self._attr_latest_version = software_status["update_info"]["updates"][
-                0
-            ]["version"]
-            self._attr_release_summary = software_status["update_info"]["updates"][
-                0
-            ]["description"]
-            self._attr_release_url = software_status["update_info"][
-                "releasenotesuri"
+            self._attr_latest_version = software_status["update_info"]["updates"][0][
+                "version"
             ]
+            self._attr_release_summary = software_status["update_info"]["updates"][0][
+                "description"
+            ]
+            self._attr_release_url = software_status["update_info"]["releasenotesuri"]
 
     async def async_install(
         self, version: str | None, backup: bool, **kwargs: Any

diff --git a/tests/components/openhome/test_update.py b/tests/components/openhome/test_update.py
@@ -93,7 +93,7 @@ async def test_not_supported(hass: HomeAssistant):
     update_firmware = AsyncMock()
     await setup_integration(hass, None, update_firmware)
 
-    state = hass.states.get("update.friendly_name_firmware_update")
+    state = hass.states.get("update.friendly_name")
 
     assert state
     assert state.state == STATE_UNKNOWN
@@ -111,7 +111,7 @@ async def test_on_latest_firmware(hass: HomeAssistant):
     update_firmware = AsyncMock()
     await setup_integration(hass, LATEST_FIRMWARE_INSTALLED, update_firmware)
 
-    state = hass.states.get("update.friendly_name_firmware_update")
+    state = hass.states.get("update.friendly_name")
 
     assert state
     assert state.state == STATE_UNKNOWN
@@ -129,7 +129,7 @@ async def test_update_available(hass: HomeAssistant):
     update_firmware = AsyncMock()
     await setup_integration(hass, FIRMWARE_UPDATE_AVAILABLE, update_firmware)
 
-    state = hass.states.get("update.friendly_name_firmware_update")
+    state = hass.states.get("update.friendly_name")
 
     assert state
     assert state.state == STATE_ON
@@ -148,7 +148,7 @@ async def test_update_available(hass: HomeAssistant):
     await hass.services.async_call(
         PLATFORM_DOMAIN,
         SERVICE_INSTALL,
-        {ATTR_ENTITY_ID: "update.friendly_name_firmware_update"},
+        {ATTR_ENTITY_ID: "update.friendly_name"},
         blocking=True,
     )
     await hass.async_block_till_done()
@@ -166,7 +166,7 @@ async def test_firmware_update_not_required(hass: HomeAssistant):
         await hass.services.async_call(
             PLATFORM_DOMAIN,
             SERVICE_INSTALL,
-            {ATTR_ENTITY_ID: "update.friendly_name_firmware_update"},
+            {ATTR_ENTITY_ID: "update.friendly_name"},
             blocking=True,
         )
     update_firmware.assert_not_called()