test: fix planner typing assertions

This commit is contained in:
Shakker 2026-03-30 21:20:59 +01:00 committed by Shakker
parent 5fb19f296a
commit 72cb2a88f1
1 changed files with 9 additions and 6 deletions

View File

@ -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);
});