From 5f376c2a27be69f9eb42c9a58718b43c101845a6 Mon Sep 17 00:00:00 2001 From: Pascal Vizeli Date: Fri, 18 Jun 2021 22:59:11 +0200 Subject: [PATCH] Using images data from version file (#2969) * Using images data from version file * fix tests --- supervisor/updater.py | 14 +++++++------- tests/test_updater.py | 14 +++++++------- 2 files changed, 14 insertions(+), 14 deletions(-) diff --git a/supervisor/updater.py b/supervisor/updater.py index 0f7d2c33a..e5257c538 100644 --- a/supervisor/updater.py +++ b/supervisor/updater.py @@ -256,13 +256,13 @@ class Updater(FileConfiguration, CoreSysAttributes): self._data[ATTR_MULTICAST] = AwesomeVersion(data["multicast"]) # Update images for that versions - self._data[ATTR_IMAGE][ATTR_HOMEASSISTANT] = data["image"]["core"] - self._data[ATTR_IMAGE][ATTR_SUPERVISOR] = data["image"]["supervisor"] - self._data[ATTR_IMAGE][ATTR_AUDIO] = data["image"]["audio"] - self._data[ATTR_IMAGE][ATTR_CLI] = data["image"]["cli"] - self._data[ATTR_IMAGE][ATTR_DNS] = data["image"]["dns"] - self._data[ATTR_IMAGE][ATTR_OBSERVER] = data["image"]["observer"] - self._data[ATTR_IMAGE][ATTR_MULTICAST] = data["image"]["multicast"] + self._data[ATTR_IMAGE][ATTR_HOMEASSISTANT] = data["images"]["core"] + self._data[ATTR_IMAGE][ATTR_SUPERVISOR] = data["images"]["supervisor"] + self._data[ATTR_IMAGE][ATTR_AUDIO] = data["images"]["audio"] + self._data[ATTR_IMAGE][ATTR_CLI] = data["images"]["cli"] + self._data[ATTR_IMAGE][ATTR_DNS] = data["images"]["dns"] + self._data[ATTR_IMAGE][ATTR_OBSERVER] = data["images"]["observer"] + self._data[ATTR_IMAGE][ATTR_MULTICAST] = data["images"]["multicast"] except KeyError as err: raise UpdaterError( diff --git a/tests/test_updater.py b/tests/test_updater.py index 4559afffd..607b96d85 100644 --- a/tests/test_updater.py +++ b/tests/test_updater.py @@ -26,25 +26,25 @@ async def test_fetch_versions(coresys: CoreSys) -> None: assert coresys.updater.version_multicast == data["multicast"] assert coresys.updater.version_observer == data["observer"] - assert coresys.updater.image_homeassistant == data["image"]["core"].format( + assert coresys.updater.image_homeassistant == data["images"]["core"].format( machine=coresys.machine ) - assert coresys.updater.image_supervisor == data["image"]["supervisor"].format( + assert coresys.updater.image_supervisor == data["images"]["supervisor"].format( arch=coresys.arch.supervisor ) - assert coresys.updater.image_cli == data["image"]["cli"].format( + assert coresys.updater.image_cli == data["images"]["cli"].format( arch=coresys.arch.supervisor ) - assert coresys.updater.image_audio == data["image"]["audio"].format( + assert coresys.updater.image_audio == data["images"]["audio"].format( arch=coresys.arch.supervisor ) - assert coresys.updater.image_dns == data["image"]["dns"].format( + assert coresys.updater.image_dns == data["images"]["dns"].format( arch=coresys.arch.supervisor ) - assert coresys.updater.image_observer == data["image"]["observer"].format( + assert coresys.updater.image_observer == data["images"]["observer"].format( arch=coresys.arch.supervisor ) - assert coresys.updater.image_multicast == data["image"]["multicast"].format( + assert coresys.updater.image_multicast == data["images"]["multicast"].format( arch=coresys.arch.supervisor )