refactor: repositories (#16038)

This commit is contained in:
Jason Rasmussen
2025-02-11 15:12:31 -05:00
committed by GitHub
parent 9d85272c2b
commit 5f3a42a132
26 changed files with 216 additions and 242 deletions
@@ -1,12 +1,17 @@
import { IEventRepository } from 'src/interfaces/event.interface';
import { EventRepository } from 'src/repositories/event.repository';
import { RepositoryInterface } from 'src/types';
import { Mocked, vitest } from 'vitest';
export const newEventRepositoryMock = (): Mocked<IEventRepository> => {
export const newEventRepositoryMock = (): Mocked<RepositoryInterface<EventRepository>> => {
return {
setup: vitest.fn(),
emit: vitest.fn() as any,
clientSend: vitest.fn() as any,
clientBroadcast: vitest.fn() as any,
serverSend: vitest.fn(),
afterInit: vitest.fn(),
handleConnection: vitest.fn(),
handleDisconnect: vitest.fn(),
setAuthFn: vitest.fn(),
};
};
@@ -1,7 +1,8 @@
import { IMachineLearningRepository } from 'src/interfaces/machine-learning.interface';
import { MachineLearningRepository } from 'src/repositories/machine-learning.repository';
import { RepositoryInterface } from 'src/types';
import { Mocked, vitest } from 'vitest';
export const newMachineLearningRepositoryMock = (): Mocked<IMachineLearningRepository> => {
export const newMachineLearningRepositoryMock = (): Mocked<RepositoryInterface<MachineLearningRepository>> => {
return {
encodeImage: vitest.fn(),
encodeText: vitest.fn(),