From 50109bb4a198a9c89edd8553395fbf2cd4506579 Mon Sep 17 00:00:00 2001 From: Oleg Ivaniv Date: Tue, 5 May 2026 11:06:31 +0200 Subject: [PATCH] test(instance-ai): mock service storage locally --- .../instance-ai/__tests__/credit-counting.test.ts | 15 +++++++-------- .../__tests__/instance-ai.service.test.ts | 15 +++++++-------- 2 files changed, 14 insertions(+), 16 deletions(-) diff --git a/packages/cli/src/modules/instance-ai/__tests__/credit-counting.test.ts b/packages/cli/src/modules/instance-ai/__tests__/credit-counting.test.ts index a3ed4651f9c..6b85d71896a 100644 --- a/packages/cli/src/modules/instance-ai/__tests__/credit-counting.test.ts +++ b/packages/cli/src/modules/instance-ai/__tests__/credit-counting.test.ts @@ -21,16 +21,15 @@ jest.mock('@n8n/instance-ai', () => { mapMastraChunkToEvent: jest.fn(), }; }); -jest.mock('@mastra/core/agent', () => ({})); -jest.mock('@mastra/core/storage', () => ({ - MemoryStorage: class {}, - MastraCompositeStore: class {}, - WorkflowsStorage: class {}, +jest.mock('../storage/typeorm-composite-store', () => ({ + TypeORMCompositeStore: class TypeORMCompositeStore {}, })); -jest.mock('@mastra/memory', () => ({ - Memory: class {}, +jest.mock('../storage/typeorm-memory-storage', () => ({ + TypeORMMemoryStorage: class TypeORMMemoryStorage {}, +})); +jest.mock('../storage/typeorm-workflows-storage', () => ({ + TypeORMWorkflowsStorage: class TypeORMWorkflowsStorage {}, })); -jest.mock('@mastra/core/workflows', () => ({})); import type { User } from '@n8n/db'; diff --git a/packages/cli/src/modules/instance-ai/__tests__/instance-ai.service.test.ts b/packages/cli/src/modules/instance-ai/__tests__/instance-ai.service.test.ts index b7dc35f199e..4efaef58afd 100644 --- a/packages/cli/src/modules/instance-ai/__tests__/instance-ai.service.test.ts +++ b/packages/cli/src/modules/instance-ai/__tests__/instance-ai.service.test.ts @@ -21,16 +21,15 @@ jest.mock('@n8n/instance-ai', () => { mapMastraChunkToEvent: jest.fn(), }; }); -jest.mock('@mastra/core/agent', () => ({})); -jest.mock('@mastra/core/storage', () => ({ - MemoryStorage: class {}, - MastraCompositeStore: class {}, - WorkflowsStorage: class {}, +jest.mock('../storage/typeorm-composite-store', () => ({ + TypeORMCompositeStore: class TypeORMCompositeStore {}, })); -jest.mock('@mastra/memory', () => ({ - Memory: class {}, +jest.mock('../storage/typeorm-memory-storage', () => ({ + TypeORMMemoryStorage: class TypeORMMemoryStorage {}, +})); +jest.mock('../storage/typeorm-workflows-storage', () => ({ + TypeORMWorkflowsStorage: class TypeORMWorkflowsStorage {}, })); -jest.mock('@mastra/core/workflows', () => ({})); import type { User } from '@n8n/db';