From ca2b16119ddb3f11403fc708a1cfc7ac77e3d101 Mon Sep 17 00:00:00 2001
From: khalil <5alil.landolsi@gmail.com>
Date: Sat, 5 Apr 2025 21:49:02 +0100
Subject: [PATCH] fixed unit test

---
 packages/mermaid/src/diagrams/state/stateDiagram-v2.spec.js | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/packages/mermaid/src/diagrams/state/stateDiagram-v2.spec.js b/packages/mermaid/src/diagrams/state/stateDiagram-v2.spec.js
index b31dfd953..970c19a58 100644
--- a/packages/mermaid/src/diagrams/state/stateDiagram-v2.spec.js
+++ b/packages/mermaid/src/diagrams/state/stateDiagram-v2.spec.js
@@ -456,7 +456,8 @@ describe('state diagram V2, ', function () {
         addLink: stateDb.addLink.bind(stateDb),
       };
 
-      parser.parse(diagram);
+      const doc = parser.parse(diagram);
+      stateDb.extract(doc);
 
       const links = stateDb.getLinks();