From acd182fe9924048476afb106a55709d20d3011ee Mon Sep 17 00:00:00 2001 From: Duc Trung LE Date: Thu, 23 Sep 2021 00:05:02 +0200 Subject: [PATCH] Update `test_shutdown_handler` for preheat mode --- tests/app/shutdown_kernel_test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/app/shutdown_kernel_test.py b/tests/app/shutdown_kernel_test.py index 1e97e1769..063fbf9d1 100644 --- 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-]+)["']"""