fix: fix retry partial

packages/snapshot/src/port/state.ts

@@ -120,21 +120,20 @@ export default class SnapshotState {
   }
 
   clearTest(testName: string): void {
-    // TODO: key by test.id
+    // TODO: key by test.id (handle multiple tests with same title)
     // TODO: reset this.added, matched, etc..
 
     this._inlineSnapshots = this._inlineSnapshots.filter(s => s.testName !== testName)
     this._inlineSnapshotStacks = this._inlineSnapshotStacks.filter(s => s.testName !== testName)
     if (this._counters.has(testName)) {
-      let counter = this._counters.get(testName)!
-      for (const key in this._snapshotData) {
-        if (keyToTestName(key) === testName) {
-          counter--
+      const counter = this._counters.get(testName)!
+      for (let i = 1; i <= counter; i++) {
+        const key = testNameToKey(testName, counter)
+        if (key in this._snapshotData || key in this._initialData) {
           this._snapshotData[key] = this._initialData[key]
-          this._uncheckedKeys.add(key)
         }
       }
-      this._counters.set(testName, counter)
+      this._counters.delete(testName)
     }
   }
 

test/core/test/__snapshots__/repro.test.ts.snap

@@ -11,3 +11,7 @@ exports[`file retry 1`] = `"foo"`;
 exports[`file retry many 1`] = `"foo"`;
 
 exports[`file retry many 2`] = `"bar"`;
+
+exports[`file retry partial 1`] = `"foo"`;
+
+exports[`file retry partial 2`] = `"bar"`;

test/core/test/repro.test.ts

@@ -39,8 +39,7 @@ test('file retry many', { retry: 1 }, (ctx) => {
   }
 })
 
-// TODO
-test.skip('file retry partial', { retry: 1 }, (ctx) => {
+test('file retry partial', { retry: 1 }, (ctx) => {
   expect('foo').toMatchSnapshot()
   if (ctx.task.result?.retryCount === 0) {
     throw new Error('boom')