Skip to content

Commit 2f618bd

Browse files
committed
test(run-engine): count updateMany batch writes in the routing store
The counting store only overrode updateBatchTaskRun, so guarding batch completion with updateManyBatchTaskRun dropped its tally to zero and the store-routing assertions failed. Both write paths now count.
1 parent 86807cb commit 2f618bd

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

internal-packages/run-engine/src/engine/systems/batchSystem.test.ts

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,14 @@ class CountingPostgresRunStore extends PostgresRunStore {
8888
return super.updateBatchTaskRun(args, tx);
8989
}
9090

91+
override async updateManyBatchTaskRun(
92+
args: Prisma.BatchTaskRunUpdateManyArgs,
93+
tx?: any
94+
): Promise<Prisma.BatchPayload> {
95+
this.batchUpdates++;
96+
return super.updateManyBatchTaskRun(args, tx);
97+
}
98+
9199
override async findWaitpoint<T extends Prisma.WaitpointFindFirstArgs>(
92100
args: any,
93101
client?: any

0 commit comments

Comments
 (0)