test: make notification test a bit more robust by clearing any previo…

…us notifications queued
klarna-incubator · Oct 17, 2024 · 4fa3855 · 4fa3855
1 parent 232991a
commit 4fa3855
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/core/src/data/threats/ThreatDataService.spec.ts b/core/src/data/threats/ThreatDataService.spec.ts
@@ -1,5 +1,4 @@
 import { randomUUID } from "crypto";
-import pg from "pg";
 import Control from "../controls/Control.js";
 import { DataAccessLayer } from "../dal.js";
 import Mitigation from "../mitigations/Mitigation.js";

diff --git a/core/src/notifications/sender.spec.ts b/core/src/notifications/sender.spec.ts
@@ -38,6 +38,10 @@ describe("notification sender", () => {
     notificationService = new NotificationDataService(dal);
   });
 
+  beforeEach(async () => {
+    await notificationService._truncate();
+  });
+
   afterAll(async () => {
     await dal.pool.end();
   });
@@ -101,6 +105,7 @@ describe("notification sender", () => {
       const mockFn = (mockedSend as Mock).mockImplementation(async () => true);
 
       await notificationService.queue(sampleNotification);
+
       for (let i = 0; i < 3; i++) {
         await notificationSender(
           notificationService,