fix(mocker): clear automocked modules on unmock (#2353)

Author: mcous

packages/vitest/src/runtime/mocker.ts

@@ -291,7 +291,7 @@ export class VitestMocker {
     const id = this.normalizePath(path)
 
     const mock = this.mockMap.get(suitefile)
-    if (mock?.[id])
+    if (mock && id in mock)
       delete mock[id]
 
     const mockId = this.getMockPath(id)

test/core/test/fixtures/mocked-dependency.ts

@@ -0,0 +1,3 @@
+export function helloWorld(): void {
+  throw new Error('not implemented')
+}

test/core/test/unmock-import.test.ts

@@ -8,6 +8,7 @@ beforeEach(() => {
     },
   }))
 })
+
 afterEach(() => {
   vi.doUnmock('/data')
 })
@@ -24,3 +25,13 @@ test('second import should had been re-mock', async () => {
   const { data } = await import('/data')
   expect(data.state).toBe('STARTED')
 })
+
+test('unmock should clear modules replaced with imitation', async () => {
+  vi.doMock('./fixtures/mocked-dependency')
+  const { helloWorld } = await import('./fixtures/mocked-dependency')
+  expect(vi.isMockFunction(helloWorld)).toBe(true)
+
+  vi.doUnmock('./fixtures/mocked-dependency')
+  const { helloWorld: unmocked } = await import('./fixtures/mocked-dependency')
+  expect(vi.isMockFunction(unmocked)).toBe(false)
+})