From 2f0223d6033d1d5bcca50d206dcb479d209979bd Mon Sep 17 00:00:00 2001
From: Bas Nijholt <bas@nijho.lt>
Date: Thu, 10 Aug 2023 14:12:32 -0700
Subject: [PATCH] await in test again

---
 .github/workflows/pytest.yaml | 38 +++++++++++++++--------------------
 tests/test_switch.py          |  1 +
 2 files changed, 17 insertions(+), 22 deletions(-)

diff --git a/.github/workflows/pytest.yaml b/.github/workflows/pytest.yaml
index 92412924..93f19e48 100644
--- a/.github/workflows/pytest.yaml
+++ b/.github/workflows/pytest.yaml
@@ -14,32 +14,26 @@ jobs:
       fail-fast: false
       matrix:
         include:
-          # - python-version: "3.10"
-          #   core-version: "2022.11.5"
-          # - python-version: "3.10"
-          #   core-version: "2022.12.9"
-          # - python-version: "3.10"
-          #   core-version: "2023.1.7"
-          # - python-version: "3.10"
-          #   core-version: "2023.2.5"
-          # - python-version: "3.10"
-          #   core-version: "2023.3.6"
-          # - python-version: "3.10"
-          #   core-version: "2023.4.6"
-          # - python-version: "3.10"
-          #   core-version: "2023.5.2"
+          - python-version: "3.10"
+            core-version: "2022.11.5"
+          - python-version: "3.10"
+            core-version: "2022.12.9"
+          - python-version: "3.10"
+            core-version: "2023.1.7"
+          - python-version: "3.10"
+            core-version: "2023.2.5"
+          - python-version: "3.10"
+            core-version: "2023.3.6"
+          - python-version: "3.10"
+            core-version: "2023.4.6"
+          - python-version: "3.10"
+            core-version: "2023.5.4"
           - python-version: "3.11"
-            core-version: "2023.6.1"
+            core-version: "2023.6.3"
           - python-version: "3.11"
             core-version: "2023.7.3"
           - python-version: "3.11"
-            n: "0"
-            core-version: "3a06659120aa629e0db290ec9e83e2ad129baaf3"
-          - python-version: "3.11"
-            n: "4"
-            core-version: "c76fac0633749e8f463f9b6c22a4d47bba5ba1e7"
-          - python-version: "3.11"
-            core-version: "2023.8.0"
+            core-version: "2023.8.1"
           - python-version: "3.11"
             core-version: "dev"
     steps:
diff --git a/tests/test_switch.py b/tests/test_switch.py
index 4744f878..484bdb83 100644
--- a/tests/test_switch.py
+++ b/tests/test_switch.py
@@ -766,6 +766,7 @@ async def test_auto_reset_manual_control(hass):
 
     async def update():
         await switch._update_attrs_and_maybe_adapt_lights(context=context, transition=0)
+        await asyncio.gather(*switch.manager.adaptation_tasks)
         await hass.async_block_till_done()
 
     async def turn_light(state, **kwargs):