fix: class instance is fine since not own prop

vitest-dev · Dec 6, 2024 · 164cb70 · 164cb70
1 parent d2cda83
commit 164cb70
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/packages/mocker/src/automocker.ts b/packages/mocker/src/automocker.ts
@@ -115,7 +115,7 @@ export function mockObject(
               const type = getType(value)
               const isFunction
                 = type.includes('Function') && typeof value === 'function'
-              if (isFunction && !value._isMockFunction) {
+              if (isFunction) {
                 // mock and delegate calls to original prototype method, which should be also mocked already
                 const original = this[key]
                 const mock = spyOn(this, key as string)

diff --git a/packages/spy/src/index.ts b/packages/spy/src/index.ts
@@ -467,16 +467,11 @@ export function spyOn<T, K extends keyof T>(
 
   const currentStub = getSpy(obj, method, accessType)
   if (currentStub) {
-    // TODO: should we reset?
     return currentStub
   }
 
   const stub = tinyspy.internalSpyOn(obj, objMethod as any)
 
-  // if (vitestSpy in stub) {
-  //   return stub as any as MockInstance
-  // }
-
   return enhanceSpy(stub) as MockInstance
 }