fix stuff

hacs · Nov 21, 2023 · f7292ca · f7292ca
1 parent 9f36e60
commit f7292ca
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 9 deletions.
diff --git a/tests/repositories/test_update_repository.py b/tests/repositories/test_update_repository.py
@@ -40,6 +40,9 @@ async def test_update_repository_entity(
     await hass.config_entries.async_reload(hacs.configuration.config_entry.entry_id)
     await hass.async_block_till_done()
 
+    # Get a new HACS instance after reload
+    hacs = get_hacs(hass)
+
     er = async_get_entity_registry(hacs.hass)
 
     entity_id = er.async_get_entity_id("update", DOMAIN, repo.data.id)
@@ -79,9 +82,6 @@ async def test_update_repository_websocket(
     repo.data.installed = True
     repo.data.installed_version = from_version
 
-    await hass.config_entries.async_reload(hacs.configuration.config_entry.entry_id)
-    await hass.async_block_till_done()
-
     response = await ws_client.send_and_receive_json(
         "hacs/repository/download", {"repository": repo.data.id, "version": to_version}
     )

diff --git a/tests/snapshots/hacs-test-org/integration-basic/test_update_repository_entity.json b/tests/snapshots/hacs-test-org/integration-basic/test_update_repository_entity.json
@@ -73,7 +73,7 @@
             "dev": true,
             "experimental": true
         },
-        "stage": null,
+        "stage": "running",
         "status": {
             "active_frontend_endpoint_plugin": false,
             "active_frontend_endpoint_theme": false,

diff --git a/tests/snapshots/hacs-test-org/integration-basic/test_update_repository_websocket.json b/tests/snapshots/hacs-test-org/integration-basic/test_update_repository_websocket.json
@@ -73,12 +73,12 @@
             "dev": true,
             "experimental": true
         },
-        "stage": null,
+        "stage": "running",
         "status": {
             "active_frontend_endpoint_plugin": false,
             "active_frontend_endpoint_theme": false,
             "inital_fetch_done": true,
-            "new": false,
+            "new": true,
             "startup": false
         },
         "system": {

diff --git a/tests/snapshots/hacs-test-org/template-basic/test_update_repository_entity.json b/tests/snapshots/hacs-test-org/template-basic/test_update_repository_entity.json
@@ -71,7 +71,7 @@
             "dev": true,
             "experimental": true
         },
-        "stage": null,
+        "stage": "running",
         "status": {
             "active_frontend_endpoint_plugin": false,
             "active_frontend_endpoint_theme": false,

diff --git a/tests/snapshots/hacs-test-org/template-basic/test_update_repository_websocket.json b/tests/snapshots/hacs-test-org/template-basic/test_update_repository_websocket.json
@@ -34,6 +34,7 @@
                 "category": "template",
                 "default_branch": "main",
                 "description": "This your first repo!",
+                "downloads": 42,
                 "full_name": "hacs-test-org/template-basic",
                 "id": "1296268",
                 "installed": true,
@@ -71,12 +72,12 @@
             "dev": true,
             "experimental": true
         },
-        "stage": null,
+        "stage": "running",
         "status": {
             "active_frontend_endpoint_plugin": false,
             "active_frontend_endpoint_theme": false,
             "inital_fetch_done": true,
-            "new": false,
+            "new": true,
             "startup": false
         },
         "system": {