test variation delete added

Author: pashidlos

src/test-variations/test-variations.controller.ts

@@ -1,4 +1,4 @@
-import { Controller, ParseUUIDPipe, Get, UseGuards, Param, Query, Put, Body } from '@nestjs/common';
+import { Controller, ParseUUIDPipe, Get, UseGuards, Param, Query, Put, Body, Delete } from '@nestjs/common';
 import { ApiTags, ApiParam, ApiBearerAuth, ApiQuery, ApiOkResponse } from '@nestjs/swagger';
 import { TestVariationsService } from './test-variations.service';
 import { TestVariation, Baseline } from '@prisma/client';
@@ -56,7 +56,18 @@ export class TestVariationsController {
   @ApiOkResponse({ type: BuildDto })
   @ApiBearerAuth()
   @UseGuards(JwtAuthGuard)
-  merge(@Query('projectId') projectId: string, @Query('branchName') branchName: string): Promise<BuildDto> {
+  merge(
+    @Query('projectId', new ParseUUIDPipe()) projectId: string,
+    @Query('branchName') branchName: string
+  ): Promise<BuildDto> {
     return this.testVariations.merge(projectId, branchName);
   }
+
+  @Delete(':id')
+  @ApiParam({ name: 'id', required: true })
+  @ApiBearerAuth()
+  @UseGuards(JwtAuthGuard)
+  delete(@Param('id', new ParseUUIDPipe()) id: string): Promise<TestVariation> {
+    return this.testVariations.delete(id);
+  }
 }

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

@@ -1,7 +1,7 @@
 import { Injectable, Inject, forwardRef } from '@nestjs/common';
 import { IgnoreAreaDto } from '../test-runs/dto/ignore-area.dto';
 import { PrismaService } from '../prisma/prisma.service';
-import { TestVariation, Baseline, Project, Build } from '@prisma/client';
+import { TestVariation, Baseline, Project, Build, TestRun } from '@prisma/client';
 import { StaticService } from '../shared/static/static.service';
 import { CommentDto } from '../shared/dto/comment.dto';
 import { BaselineDataDto, convertBaselineDataToQuery } from '../shared/dto/baseline-data.dto';
@@ -165,4 +165,31 @@ export class TestVariationsService {
 
     return build;
   }
+
+  async delete(id: string): Promise<TestVariation> {
+    const [testVariation, testRuns] = await Promise.all([
+      this.getDetails(id),
+      this.prismaService.testRun.findMany({
+        where: { testVariationId: id },
+      })
+    ])
+
+    // delete testRun
+    await Promise.all(testRuns.map(item => this.testRunsService.delete(item.id)));
+    
+    // delete baseline
+    await Promise.all(testVariation.baselines.map(baseline => this.deleteBaseline(baseline)));
+
+    // delete testVariation
+    return this.prismaService.testVariation.delete({
+      where: { id },
+    });
+  }
+
+  private async deleteBaseline(baseline: Baseline): Promise<Baseline> {
+    this.staticService.deleteImage(baseline.baselineName);
+    return this.prismaService.baseline.delete({
+      where: { id: baseline.id },
+    });
+  }
 }