From 523286bbcba2129d4ab89da98a3d30ecf5887b69 Mon Sep 17 00:00:00 2001 From: yari-dewalt 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(); });