diff --git a/api/src/experiments/experiments.resolver.spec.ts b/api/src/experiments/experiments.resolver.spec.ts index ef3cdd599b33bfeac24849b3f99a6370db5c46d7..0ff1cb091669713156af236d4b504706a29074a1 100644 --- a/api/src/experiments/experiments.resolver.spec.ts +++ b/api/src/experiments/experiments.resolver.spec.ts @@ -1,5 +1,6 @@ import { Test, TestingModule } from '@nestjs/testing'; -import { User } from 'src/users/models/user.model'; +import { ExperimentStatus } from '../engine/models/experiment/experiment.model'; +import { User } from '../users/models/user.model'; import { ENGINE_SERVICE } from '../engine/engine.constants'; import { IEngineService } from '../engine/engine.interfaces'; import { ExperimentsResolver } from './experiments.resolver'; @@ -164,11 +165,17 @@ describe('ExperimentsResolver', () => { engineService.createExperiment = undefined; engineService.runExperiment.mockResolvedValue([]); experimentsService.create.mockReturnValue({ id: 'test' }); - await resolver.createExperiment(request, user, data, true); + const result = await resolver.createExperiment( + request, + user, + data, + true, + ); expect(engineService.runExperiment.mock.calls.length).toBeGreaterThan( 0, ); + expect(result.status).toBe(ExperimentStatus.SUCCESS); }); }); });