Merge pull request #1 from Visual-Regression-Tracker/7-baseline_image_removed

Baseline image is removed together with test

Author: pashidlos

src/test-runs/test-runs.service.ts

@@ -1,5 +1,5 @@
 import { Injectable } from '@nestjs/common';
-import { PNG } from 'pngjs';
+import { PNG, PNGWithMetadata } from 'pngjs';
 import Pixelmatch from 'pixelmatch';
 import { CreateTestRequestDto } from 'src/test/dto/create-test-request.dto';
 import { IgnoreAreaDto } from 'src/test/dto/ignore-area.dto';
@@ -40,6 +40,17 @@ export class TestRunsService {
     testVariation: TestVariation;
   }> {
     const testRun = await this.findOne(id);
+
+    // remove old baseline
+    if (testRun.testVariation.baselineName) {
+      this.staticService.deleteImage(testRun.testVariation.baselineName);
+    }
+
+    // save new baseline
+    const baseline = this.staticService.getImage(testRun.imageName)
+    const imageName = `${Date.now()}.baseline.png`;
+    this.staticService.saveImage(imageName, PNG.sync.write(baseline));
+
     return this.prismaService.testRun.update({
       where: { id },
       include: {
@@ -49,7 +60,7 @@ export class TestRunsService {
         status: TestStatus.ok,
         testVariation: {
           update: {
-            baselineName: testRun.imageName,
+            baselineName: imageName,
           },
         },
       },
@@ -97,10 +108,18 @@ export class TestRunsService {
       status: TestStatus.new,
     };
 
-    // compare with baseline
+    // get baseline image
+    let baseline: PNGWithMetadata
     if (testVariation.baselineName) {
-      const baseline = this.staticService.getImage(testVariation.baselineName);
+      try {
+        baseline = this.staticService.getImage(testVariation.baselineName)
+      } catch (ex) {
+        console.log(`Cannot load baseline image: ${testVariation.baselineName}. ${ex}`)
+      }
+    }
 
+    // compare with baseline
+    if (baseline) {
       const diffImageKey = `${Date.now()}.diff.png`;
       const diff = new PNG({
         width: baseline.width,