test: usecaseDb

packages/mermaid/src/diagrams/usecase/usecase.spec.ts

@@ -0,0 +1,21 @@
+import { UsecaseDB } from './usecaseDB.js';
+
+describe('Usecase diagram', function () {
+  const db = new UsecaseDB();
+  beforeEach(function () {
+    db.clear();
+  });
+  it('should add node when adding participants', function () {
+    db.addParticipants({ service: 'Authz' });
+    expect(db.getServices()).toStrictEqual(['Authz']);
+  });
+  it('should add actor when adding relationship', function () {
+    db.addRelationship('Student', '(Enrol)', '-->');
+    expect(db.getActors()).toStrictEqual(['Student']);
+  });
+  it('should add use case and service when adding relationship', function () {
+    db.addRelationship('(Enrol)', 'Enrolment System', '-->');
+    expect(db.getServices()).toStrictEqual(['Enrolment System']);
+    expect(db.getUseCases()).toStrictEqual(['(Enrol)']);
+  });
+});

packages/mermaid/src/diagrams/usecase/usecaseDB.ts

@@ -58,13 +58,19 @@ export class UsecaseDB {
     }
   }
 
-  addRelationship(source: string, target: string, token: string): void {
+  /**
+   * Adds an edge to the graph
+   * @param source - id of node
+   * @param target - id of node
+   * @param arrowString - '-->' or '->' or '-- some text -->'
+   */
+  addRelationship(source: string, target: string, arrowString: string): void {
     source = common.sanitizeText(source, getConfig());
     target = common.sanitizeText(target, getConfig());
     const sourceNode = this.getNode(source, isUseCaseLabel(source) ? 'usecase' : 'actor');
     const targetNode = this.getNode(target, isUseCaseLabel(target) ? 'usecase' : 'service');
-    const label = (/--(.+?)(-->|->)/.exec(token)?.[1] ?? '').trim();
-    const arrow = token.includes('-->') ? '-->' : '->';
+    const label = (/--(.+?)(-->|->)/.exec(arrowString)?.[1] ?? '').trim();
+    const arrow = arrowString.includes('-->') ? '-->' : '->';
     this.links.push(new UsecaseLink(sourceNode, targetNode, arrow, label));
   }
 
@@ -181,6 +187,16 @@ export class UsecaseDB {
     return this.systemBoundaries;
   }
 
+  getUseCases(): string[] {
+    const useCases = [];
+    for (const node of this.nodes) {
+      if (node.nodeType === 'usecase') {
+        useCases.push(node.id);
+      }
+    }
+    return useCases;
+  }
+
   getAccDescription = getAccDescription;
   getAccTitle = getAccTitle;
   getDiagramTitle = getDiagramTitle;
@@ -241,7 +257,9 @@ export default {
   getData: db.getData.bind(db),
   getRelationships: db.getRelationships.bind(db),
   getDiagramTitle: db.getDiagramTitle.bind(db),
+  getServices: db.getServices.bind(db),
   getSystemBoundaries: db.getSystemBoundaries.bind(db),
+  getUseCases: db.getUseCases.bind(db),
   setAccDescription,
   setAccTitle,
   setDiagramTitle: db.setDiagramTitle.bind(db),