From 12677b11a5051b9592516a2477cba5330a168db4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=AB=98=E7=84=B6?= Date: Fri, 20 Mar 2026 18:37:20 +0800 Subject: [PATCH] fix(server): fix failure unit test case caused by #489 --- server/tests/k8s/test_kubernetes_service.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/server/tests/k8s/test_kubernetes_service.py b/server/tests/k8s/test_kubernetes_service.py index 438b53c57..8a50871a7 100644 --- a/server/tests/k8s/test_kubernetes_service.py +++ b/server/tests/k8s/test_kubernetes_service.py @@ -227,7 +227,8 @@ async def test_create_sandbox_with_no_timeout_calls_provider_with_expires_at_non assert kwargs["expires_at"] is None assert kwargs["labels"].get(SANDBOX_MANUAL_CLEANUP_LABEL) == "true" - def test_create_sandbox_with_network_policy_passes_egress_token_and_annotations( + @pytest.mark.asyncio + async def test_create_sandbox_with_network_policy_passes_egress_token_and_annotations( self, k8s_service, create_sandbox_request ): create_sandbox_request.network_policy = NetworkPolicy(default_action="deny", egress=[]) @@ -245,14 +246,15 @@ def test_create_sandbox_with_network_policy_passes_egress_token_and_annotations( "src.services.k8s.kubernetes_service.generate_egress_token", return_value="egress-token", ): - k8s_service.create_sandbox(create_sandbox_request) + await k8s_service.create_sandbox(create_sandbox_request) _, kwargs = k8s_service.workload_provider.create_workload.call_args assert kwargs["egress_auth_token"] == "egress-token" assert kwargs["egress_mode"] == EGRESS_MODE_DNS assert kwargs["annotations"][SANDBOX_EGRESS_AUTH_TOKEN_METADATA_KEY] == "egress-token" - def test_create_sandbox_with_network_policy_passes_egress_mode_dns_nft_from_config( + @pytest.mark.asyncio + async def test_create_sandbox_with_network_policy_passes_egress_mode_dns_nft_from_config( self, k8s_service, create_sandbox_request ): create_sandbox_request.network_policy = NetworkPolicy(default_action="deny", egress=[]) @@ -273,7 +275,7 @@ def test_create_sandbox_with_network_policy_passes_egress_mode_dns_nft_from_conf "src.services.k8s.kubernetes_service.generate_egress_token", return_value="egress-token", ): - k8s_service.create_sandbox(create_sandbox_request) + await k8s_service.create_sandbox(create_sandbox_request) _, kwargs = k8s_service.workload_provider.create_workload.call_args assert kwargs["egress_mode"] == EGRESS_MODE_DNS_NFT