From 72cb2a88f1bd83d182b2a9bcb0fb47799a4a132f Mon Sep 17 00:00:00 2001
From: Shakker <shakkerdroid@gmail.com>
Date: Mon, 30 Mar 2026 21:20:59 +0100
Subject: [PATCH] test: fix planner typing assertions

---
 test/scripts/test-planner.test.ts | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/test/scripts/test-planner.test.ts b/test/scripts/test-planner.test.ts
index 9a1de5d54f5..08cafc2630f 100644
--- a/test/scripts/test-planner.test.ts
+++ b/test/scripts/test-planner.test.ts
@@ -410,9 +410,12 @@ describe("test planner", () => {
 
     expect(targetedUnit).toBeTruthy();
     expect(defaultUnitWithSameId).toBeTruthy();
-    expect(defaultUnitWithSameId).not.toBe(targetedUnit);
-    expect(plan.topLevelSingleShardAssignments.get(targetedUnit)).toBeUndefined();
-    expect(plan.topLevelSingleShardAssignments.get(defaultUnitWithSameId)).toBeDefined();
+    const targetedUnitRecord = targetedUnit!;
+    const defaultUnitRecord = defaultUnitWithSameId!;
+
+    expect(defaultUnitRecord).not.toBe(targetedUnitRecord);
+    expect(plan.topLevelSingleShardAssignments.get(targetedUnitRecord)).toBeUndefined();
+    expect(plan.topLevelSingleShardAssignments.get(defaultUnitRecord)).toBeDefined();
 
     artifacts.cleanupTempArtifacts();
   });
@@ -490,11 +493,11 @@ describe("test planner", () => {
     artifacts.cleanupTempArtifacts();
 
     await expect(
-      new Promise((resolve, reject) => {
+      new Promise<void>((resolve, reject) => {
         stream.on("error", reject);
-        stream.end("after cleanup\n", resolve);
+        stream.end("after cleanup\n", () => resolve());
       }),
-    ).resolves.toBeNull();
+    ).resolves.toBeUndefined();
     expect(fs.existsSync(artifactDir)).toBe(false);
   });