From a80dd711a7309de3dc7624f66625d15055ae46a9 Mon Sep 17 00:00:00 2001
From: Sidharth Vinod <sidharthv96@gmail.com>
Date: Sun, 7 Jul 2024 16:28:57 +0530
Subject: [PATCH] fix(class): #5624 Reset direction to default in classDiagram

---
 packages/mermaid/src/diagrams/class/classDb.ts        |  1 +
 .../mermaid/src/diagrams/class/classDiagram.spec.ts   | 11 +++++++++++
 2 files changed, 12 insertions(+)

diff --git a/packages/mermaid/src/diagrams/class/classDb.ts b/packages/mermaid/src/diagrams/class/classDb.ts
index cb8b90af4c..a59c1eada1 100644
--- a/packages/mermaid/src/diagrams/class/classDb.ts
+++ b/packages/mermaid/src/diagrams/class/classDb.ts
@@ -113,6 +113,7 @@ export const clear = function () {
   functions.push(setupToolTips);
   namespaces = new Map();
   namespaceCounter = 0;
+  direction = 'TB';
   commonClear();
 };
 
diff --git a/packages/mermaid/src/diagrams/class/classDiagram.spec.ts b/packages/mermaid/src/diagrams/class/classDiagram.spec.ts
index 8412c6fbf2..9804b325ef 100644
--- a/packages/mermaid/src/diagrams/class/classDiagram.spec.ts
+++ b/packages/mermaid/src/diagrams/class/classDiagram.spec.ts
@@ -444,6 +444,17 @@ class C13["With Città foreign language"]
         ]
       `);
     });
+
+    it('should revert direction to default once direction is removed', () => {
+      parser.parse(`classDiagram
+          direction RL
+          class A`);
+      expect(classDb.getDirection()).toBe('RL');
+      classDb.clear();
+      parser.parse(`classDiagram
+          class B`);
+      expect(classDb.getDirection()).toBe('TB');
+    });
   });
 
   describe('when parsing class defined in brackets', function () {