TestVariationsService. findOrCreate covered

Author: pashidlos

src/test-variations/test-variations.service.spec.ts

@@ -1,22 +1,116 @@
 import { Test, TestingModule } from '@nestjs/testing';
 import { TestVariationsService } from './test-variations.service';
 import { PrismaService } from '../prisma/prisma.service';
+import { CreateTestRequestDto } from '../test/dto/create-test-request.dto';
+import { StaticService } from '../shared/static/static.service';
+
+const initModule = async ({
+  findManyMock = jest.fn().mockReturnValue([]),
+  createMock = jest.fn(),
+}) => {
+  const module: TestingModule = await Test.createTestingModule({
+    providers: [
+      TestVariationsService,
+      { provide: StaticService, useValue: {} },
+      {
+        provide: PrismaService, useValue: {
+          testVariation: {
+            findMany: findManyMock,
+            create: createMock
+          }
+        }
+      },
+    ],
+  }).compile();
+
+  return module.get<TestVariationsService>(TestVariationsService);
+}
+
+const dataRequiredFields: CreateTestRequestDto = {
+  buildId: 'buildId',
+  projectId: 'projectId',
+  name: 'Test name',
+  imageBase64: 'Image'
+}
+
+const dataAllFields: CreateTestRequestDto = {
+  buildId: 'buildId',
+  projectId: 'projectId',
+  name: 'Test name',
+  imageBase64: 'Image',
+  os: 'OS',
+  browser: 'browser',
+  viewport: 'viewport',
+  device: 'device',
+}
 
 describe('TestVariationsService', () => {
   let service: TestVariationsService;
 
-  beforeEach(async () => {
-    const module: TestingModule = await Test.createTestingModule({
-      providers: [
-        TestVariationsService,
-        { provide: PrismaService, useValue: {} },
-      ],
-    }).compile();
+  describe('findOrCreate', () => {
+
+    it('can find by required fields', async () => {
+      const data = dataRequiredFields
+      const findManyMock = jest.fn()
+      service = await initModule({ findManyMock: findManyMock.mockResolvedValueOnce([data]) })
+
+      const result = await service.findOrCreate(data)
+
+      expect(findManyMock).toHaveBeenCalledWith({
+        where: {
+          name: data.name,
+          projectId: data.projectId,
+          os: null,
+          browser: null,
+          viewport: null,
+          device: null,
+        },
+      })
+      expect(result).toBe(data)
+    })
+
+    it('can find by all fields', async () => {
+      const data = dataAllFields
+      const findManyMock = jest.fn()
+      service = await initModule({ findManyMock: findManyMock.mockResolvedValueOnce([data]) })
+
+      const result = await service.findOrCreate(data)
+
+      expect(findManyMock).toHaveBeenCalledWith({
+        where: {
+          name: data.name,
+          projectId: data.projectId,
+          os: data.os,
+          browser: data.browser,
+          viewport: data.viewport,
+          device: data.device,
+        },
+      })
+      expect(result).toBe(data)
+    })
+
+    it('can create if not found', async () => {
+      const data = dataAllFields
+      const createMock = jest.fn()
+      service = await initModule({ createMock: createMock.mockResolvedValueOnce(data) })
 
-    service = module.get<TestVariationsService>(TestVariationsService);
-  });
+      const result = await service.findOrCreate(data)
 
-  it('should be defined', () => {
-    expect(service).toBeDefined();
-  });
+      expect(createMock).toHaveBeenCalledWith({
+        data: {
+          name: data.name,
+          os: data.os,
+          browser: data.browser,
+          viewport: data.viewport,
+          device: data.device,
+          project: {
+            connect: {
+              id: data.projectId,
+            }
+          }
+        },
+      })
+      expect(result).toBe(data)
+    })
+  })
 });

src/test-variations/test-variations.service.ts

@@ -3,7 +3,7 @@ import { CreateTestRequestDto } from '../test/dto/create-test-request.dto';
 import { IgnoreAreaDto } from '../test/dto/ignore-area.dto';
 import { PrismaService } from '../prisma/prisma.service';
 import { TestVariation, Baseline } from '@prisma/client';
-import { StaticService } from 'src/shared/static/static.service';
+import { StaticService } from '../shared/static/static.service';
 
 @Injectable()
 export class TestVariationsService {

src/test/dto/create-test-request.dto.ts

@@ -13,22 +13,22 @@ export class CreateTestRequestDto {
   @ApiProperty()
   @IsOptional()
   @IsString()
-  os: string;
+  os?: string;
 
   @ApiProperty()
   @IsOptional()
   @IsString()
-  browser: string;
+  browser?: string;
 
   @ApiProperty()
   @IsOptional()
   @IsString()
-  viewport: string;
+  viewport?: string;
 
   @ApiProperty()
   @IsOptional()
   @IsString()
-  device: string;
+  device?: string;
 
   @ApiProperty()
   @IsUUID()
@@ -41,5 +41,5 @@ export class CreateTestRequestDto {
   @ApiProperty()
   @IsOptional()
   @IsNumber()
-  diffTollerancePercent: number;
+  diffTollerancePercent?: number;
 }