fix: type regression in vi.mocked() static class methods (#9857)

Co-authored-by: Hiroshi Ogawa <hi.ogawa.zz@gmail.com>

Author: purepear

packages/spy/src/types.ts

@@ -462,12 +462,10 @@ type DeepPartialMock<T extends Procedure | Constructable = Procedure> = Mock<
 export type MaybeMockedConstructor<T> = T extends Constructable
   ? Mock<T>
   : T
-export type MockedFunction<T extends Procedure | Constructable> = Mock<T> & {
-  [K in keyof T]: T[K];
-}
-export type PartiallyMockedFunction<T extends Procedure | Constructable> = PartialMock<T> & {
-  [K in keyof T]: T[K];
-}
+export type MockedFunction<T extends Procedure | Constructable> = Mock<T>
+  & MockedObject<T>
+export type PartiallyMockedFunction<T extends Procedure | Constructable> = PartialMock<T>
+  & MockedObject<T>
 export type MockedFunctionDeep<T extends Procedure | Constructable> = Mock<T>
   & MockedObjectDeep<T>
 export type PartiallyMockedFunctionDeep<T extends Procedure | Constructable> = DeepPartialMock<T>

test/core/test/vi.spec.ts

@@ -151,6 +151,10 @@ describe('testing vi utils', () => {
       public getBar(): string {
         return this.bar
       }
+
+      static getBaz(): string {
+        return 'baz'
+      }
     }
     class FooMock implements Mocked<Foo> {
       readonly barMock: Mock<() => string> = vi.fn()
@@ -168,6 +172,8 @@ describe('testing vi utils', () => {
     if (0) {
       vi.mocked(Foo).mockImplementation(FooMock)
       vi.mocked(Foo).mockImplementation(Foo)
+      vi.mocked(Foo).getBaz.mockImplementation(() => 'baz')
+      vi.mocked(Foo, { partial: true }).getBaz.mockImplementation(() => 'baz')
     }
   })