fix test

Author: THardy98

packages/test/src/helpers-integration.ts

@@ -321,9 +321,10 @@ export async function setActivityState(
       //  • the activity vanished (already completed)
       return activityInfo ? !activityInfo.paused : true;
     } else {
-      // If we are resetting, success when heartbeat
-      // details are reset.
-      return activityInfo?.heartbeatDetails === null;
+      // If we are resetting, success when either
+      //  • heartbeat details have been reset OR
+      //  • the activity vanished (completed / retried)
+      return activityInfo ? activityInfo.heartbeatDetails === null : true;
     }
   }, 15000);
 }

packages/test/src/test-integration-workflows.ts

@@ -1513,7 +1513,7 @@ test('Activity reset returns expected cancellation details', async (t) => {
 
   await worker.runUntil(async () => {
     const testActivityId = randomUUID();
-    const handle = await startWorkflow(heartbeatPauseWorkflow, { args: [testActivityId, true, 2] });
+    const handle = await startWorkflow(heartbeatPauseWorkflow, { args: [testActivityId, true, 1] });
 
     // Wait for it to exist and heartbeat
     await waitUntil(async () => {