diff --git a/src/tests/services/SyncService.spec.ts b/src/tests/services/SyncService.spec.ts index 2707f1fd0f5..d76df39243a 100644 --- a/src/tests/services/SyncService.spec.ts +++ b/src/tests/services/SyncService.spec.ts @@ -8,6 +8,8 @@ import * as connect from '../../apis/connect.ts' import { provideConnection } from '../../composables/useConnection.js' import { SyncService } from '../../services/SyncService.js' +vi.mock('../../apis/connect') + const connection = { documentId: 123, sessionId: 345, @@ -53,7 +55,6 @@ describe('Sync service', () => { getBaseVersionEtag, setBaseVersionEtag, ) - vi.mock('../../apis/connect') vi.mocked(connect.open).mockResolvedValue(openResult) const openHandler = vi.fn() const service = new SyncService({ connection, openConnection }) diff --git a/src/tests/services/WebsocketPolyfill.spec.ts b/src/tests/services/WebsocketPolyfill.spec.ts index 1cecd225ce6..1f61490130b 100644 --- a/src/tests/services/WebsocketPolyfill.spec.ts +++ b/src/tests/services/WebsocketPolyfill.spec.ts @@ -8,14 +8,15 @@ import { shallowRef } from 'vue' import { SyncService } from '../../services/SyncService.ts' import initWebSocketPolyfill from '../../services/WebSocketPolyfill.js' +vi.mock(import('../../services/SyncService.ts'), () => { + const SyncService = vi.fn() + SyncService.prototype.bus = { on: vi.fn() } + SyncService.prototype.open = vi.fn().mockImplementation(async () => ({})) + SyncService.prototype.hasActiveConnection = vi.fn() + return { SyncService } +}) + describe('Init function', () => { - vi.mock(import('../../services/SyncService.ts'), () => { - const SyncService = vi.fn() - SyncService.prototype.bus = { on: vi.fn() } - SyncService.prototype.open = vi.fn().mockImplementation(async () => ({})) - SyncService.prototype.hasActiveConnection = vi.fn() - return { SyncService } - }) const mockSyncService = () => new SyncService({ connection: shallowRef(undefined), openConnection: vi.fn(),