test(mothership): assert the claim admits a first execution or an interrupted re-run

Author: waleedlatif1

apps/sim/lib/mothership/async-runs/repository.test.ts

@@ -5,7 +5,13 @@ import {
   copilotRequestStops,
   copilotRuns,
 } from '@sim/db/schema'
-import { dbChainMockFns, hasMockCondition, queueTableRows, resetDbChainMock } from '@sim/testing'
+import {
+  dbChainMockFns,
+  flattenMockConditions,
+  hasMockCondition,
+  queueTableRows,
+  resetDbChainMock,
+} from '@sim/testing'
 import { beforeEach, describe, expect, it } from 'vitest'
 import {
   areStreamToolExecutionsSettled,
@@ -240,14 +246,26 @@ describe('durable Sim tool ownership', () => {
         (condition) => condition.type === 'eq' && condition.right === input.userId
       )
     ).toBe(true)
+    /** A first execution, or a re-run of one whose owner died before it recorded a result. */
+    const claimable = flattenMockConditions(dbChainMockFns.where.mock.calls[1]?.[0]).find(
+      (condition) => condition.type === 'or'
+    )
+    const [firstExecution, interruptedExecution] = (claimable?.conditions ?? []) as unknown[]
     expect(
       hasMockCondition(
-        dbChainMockFns.where.mock.calls[1]?.[0],
+        firstExecution,
         (condition) =>
           condition.type === 'isNull' &&
           condition.column === copilotAsyncToolCalls.executionStartedAt
       )
     ).toBe(true)
+    expect(
+      hasMockCondition(
+        interruptedExecution,
+        (condition) =>
+          condition.type === 'lt' && condition.left === copilotAsyncToolCalls.executionAttempts
+      )
+    ).toBe(true)
     expect(dbChainMockFns.set).toHaveBeenCalledWith(
       expect.objectContaining({ executionStartedAt: expect.any(Date), status: 'running' })
     )