+6

packages/spy/src/index.ts

Reviewed

··· 433 433 // are only used by the automocker, so this doesn't matter 434 434 for (const prop of prototypeMembers) { 435 435 const prototypeMock = returnValue[prop] 436 436 + // the method was overidden because of inheritence, ignore it 437 437 + // eslint-disable-next-line ts/no-use-before-define 438 438 + if (prototypeMock !== mock.prototype[prop]) { 439 439 + continue 440 440 + } 441 441 + 436 442 const isMock = isMockFunction(prototypeMock) 437 443 const prototypeState = isMock ? prototypeMock.mock : undefined 438 444 const prototypeConfig = isMock ? MOCK_CONFIGS.get(prototypeMock) : undefined

+11

test/core/src/class-inheritence/bar.ts

Reviewed

··· 1 1 + import { Foo } from './foo.js' 2 2 + 3 3 + export class Bar extends Foo { 4 4 + override doSomething(): boolean { 5 5 + return true 6 6 + } 7 7 + 8 8 + doSomethingElse(): boolean { 9 9 + return true 10 10 + } 11 11 + }

+9

test/core/src/class-inheritence/foo.ts

Reviewed

··· 1 1 + export class Foo { 2 2 + doSomething(): boolean { 3 3 + return false 4 4 + } 5 5 + 6 6 + unused() { 7 7 + // 8 8 + } 9 9 + }

+57

test/core/test/mocking/automocking-class-inheritence.test.ts

Reviewed

··· 1 1 + // bar.spec.ts 2 2 + import { describe, expect, it, vi } from 'vitest' 3 3 + import { Bar } from '../../src/class-inheritence/bar' 4 4 + 5 5 + vi.mock(import('./../../src/class-inheritence/foo')) 6 6 + 7 7 + describe('not mocking class when parent is mocked', () => { 8 8 + describe('doSomething', () => { 9 9 + it('returns true', () => { 10 10 + const bar = new Bar() 11 11 + expect(bar.doSomething()).toBe(true) 12 12 + }) 13 13 + 14 14 + it('should match the prototype', () => { 15 15 + const bar = new Bar() 16 16 + expect(bar.doSomething).toBe(Bar.prototype.doSomething) 17 17 + }) 18 18 + 19 19 + it('should not be mocked', () => { 20 20 + const bar = new Bar() 21 21 + expect(bar.doSomething).not.toHaveProperty('mock') 22 22 + }) 23 23 + }) 24 24 + 25 25 + describe('doSomethingElse', () => { 26 26 + it('returns true', () => { 27 27 + const bar = new Bar() 28 28 + expect(bar.doSomethingElse()).toBe(true) 29 29 + }) 30 30 + 31 31 + it('should match the prototype', () => { 32 32 + const bar = new Bar() 33 33 + expect(bar.doSomethingElse).toBe(Bar.prototype.doSomethingElse) 34 34 + }) 35 35 + 36 36 + it('should not be mocked', () => { 37 37 + const bar = new Bar() 38 38 + expect(bar.doSomethingElse).not.toHaveProperty('mock') 39 39 + }) 40 40 + }) 41 41 + }) 42 42 + 43 43 + describe('mocking class when parent is not mocked', () => { 44 44 + it('mocks correctly', () => { 45 45 + class Bar { 46 46 + doSomething() {} 47 47 + } 48 48 + 49 49 + const Zoo = vi.mockObject(class Zoo extends Bar { 50 50 + ownMethod() {} 51 51 + }) 52 52 + 53 53 + const zoo = new Zoo() 54 54 + expect(vi.isMockFunction(zoo.doSomething)).toBe(true) 55 55 + expect(vi.isMockFunction(zoo.ownMethod)).toBe(true) 56 56 + }) 57 57 + })