From 523286bbcba2129d4ab89da98a3d30ecf5887b69 Mon Sep 17 00:00:00 2001
From: yari-dewalt <yari.dewalt@gmail.com>
Date: Mon, 27 Jan 2025 09:29:51 -0800
Subject: [PATCH] Update test file to work with new Db

---
 .../mermaid/src/diagrams/requirement/requirementDb.spec.ts | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/packages/mermaid/src/diagrams/requirement/requirementDb.spec.ts b/packages/mermaid/src/diagrams/requirement/requirementDb.spec.ts
index da6684ca1..715d4d053 100644
--- a/packages/mermaid/src/diagrams/requirement/requirementDb.spec.ts
+++ b/packages/mermaid/src/diagrams/requirement/requirementDb.spec.ts
@@ -1,8 +1,9 @@
-import requirementDb from './requirementDb.js';
+import { RequirementDB } from './requirementDb.js';
 import { describe, it, expect } from 'vitest';
-import type { RelationshipType } from './types.js';
+import type { Relation, RelationshipType } from './types.js';
 
 describe('requirementDb', () => {
+  const requirementDb = new RequirementDB();
   beforeEach(() => {
     requirementDb.clear();
   });
@@ -23,7 +24,7 @@ describe('requirementDb', () => {
     requirementDb.addRelationship('contains' as RelationshipType, 'src', 'dst');
     const relationships = requirementDb.getRelationships();
     const relationship = relationships.find(
-      (r) => r.type === 'contains' && r.src === 'src' && r.dst === 'dst'
+      (r: Relation) => r.type === 'contains' && r.src === 'src' && r.dst === 'dst'
     );
     expect(relationship).toBeDefined();
   });