Update test_shutdown_handler for preheat mode

voila-dashboards · Oct 4, 2021 · d94fca1 · d94fca1
1 parent 71807b5
commit d94fca1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/app/shutdown_kernel_test.py b/tests/app/shutdown_kernel_test.py
@@ -1,7 +1,8 @@
 import re
 
 
-async def test_shutdown_handler(http_server_client, base_url):
+async def test_shutdown_handler(http_server_client, base_url, wait_for_kernel):
+    await wait_for_kernel()
     response = await http_server_client.fetch(base_url)
     html_text = response.body.decode('utf-8')
     pattern = r"""kernelId": ["']([0-9a-zA-Z-]+)["']"""