Update test_shutdown_handler for preheat mode

voila-dashboards · Sep 22, 2021 · acd182f · acd182f
1 parent 74a9b9a
commit acd182f
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-]+)["']"""