mirror of
https://github.com/n8n-io/n8n.git
synced 2024-12-27 05:29:42 -08:00
25 lines
797 B
TypeScript
25 lines
797 B
TypeScript
import { Container } from 'typedi';
|
|
import { mock } from 'jest-mock-extended';
|
|
import type { DeepPartial } from 'ts-essentials';
|
|
import { DataSource, EntityManager, type EntityMetadata } from '@n8n/typeorm';
|
|
import type { Class } from 'n8n-core';
|
|
|
|
export const mockInstance = <T>(
|
|
serviceClass: Class<T>,
|
|
data: DeepPartial<T> | undefined = undefined,
|
|
) => {
|
|
const instance = mock<T>(data);
|
|
Container.set(serviceClass, instance);
|
|
return instance;
|
|
};
|
|
|
|
export const mockEntityManager = (entityClass: Class) => {
|
|
const entityManager = mockInstance(EntityManager);
|
|
const dataSource = mockInstance(DataSource, {
|
|
manager: entityManager,
|
|
getMetadata: () => mock<EntityMetadata>({ target: entityClass }),
|
|
});
|
|
Object.assign(entityManager, { connection: dataSource });
|
|
return entityManager;
|
|
};
|