From 977149af30902c0f59c8a50621aceaf44bada1ab Mon Sep 17 00:00:00 2001
From: David East <davideast@users.noreply.github.com>
Date: Tue, 2 Sep 2025 20:19:31 -0400
Subject: [PATCH] fix(a2a-server): Fix flaky test on Windows by awaiting server
 close (#7616)

Co-authored-by: cornmander <shikhman@google.com>
---
 packages/a2a-server/src/endpoints.test.ts | 25 +++++++++++++----------
 1 file changed, 14 insertions(+), 11 deletions(-)

diff --git a/packages/a2a-server/src/endpoints.test.ts b/packages/a2a-server/src/endpoints.test.ts
index 77a1e59ac0..6cabb1f840 100644
--- a/packages/a2a-server/src/endpoints.test.ts
+++ b/packages/a2a-server/src/endpoints.test.ts
@@ -88,21 +88,24 @@ describe('Agent Server Endpoints', () => {
     });
   });
 
-  afterAll(
-    () =>
-      new Promise<void>((resolve, reject) => {
+  afterAll(async () => {
+    if (server) {
+      await new Promise<void>((resolve, reject) => {
         server.close((err) => {
           if (err) return reject(err);
-
-          try {
-            fs.rmSync(testWorkspace, { recursive: true, force: true });
-          } catch (e) {
-            console.warn(`Could not remove temp dir '${testWorkspace}':`, e);
-          }
           resolve();
         });
-      }),
-  );
+      });
+    }
+
+    if (testWorkspace) {
+      try {
+        fs.rmSync(testWorkspace, { recursive: true, force: true });
+      } catch (e) {
+        console.warn(`Could not remove temp dir '${testWorkspace}':`, e);
+      }
+    }
+  });
 
   it('should create a new task via POST /tasks', async () => {
     const response = await createTask('test-context');