mend

packages/mermaid/src/diagrams/er/parser/erDiagram.spec.js

@@ -143,7 +143,7 @@ describe('when parsing ER diagram it...', function () {
       expect(entities[entity].alias).toBe(alias);
     });
 
-    it('can have an alias even if the relationship is defined first', function () {
+    it('can have an alias even if the relationship is defined before class', function () {
       const firstEntity = 'foo';
       const secondEntity = 'bar';
       const alias = 'batman';
@@ -156,6 +156,20 @@ describe('when parsing ER diagram it...', function () {
       expect(entities[firstEntity].alias).toBe(alias);
       expect(entities[secondEntity].alias).toBeUndefined();
     });
+
+    it('can have an alias even if the relationship is defined after class', function () {
+      const firstEntity = 'foo';
+      const secondEntity = 'bar';
+      const alias = 'batman';
+      erDiagram.parser.parse(
+        `erDiagram\nclass ${firstEntity}["${alias}"]\n${firstEntity} ||--o| ${secondEntity} : rel\n`
+      );
+      const entities = erDb.getEntities();
+      expect(entities.hasOwnProperty(firstEntity)).toBe(true);
+      expect(entities.hasOwnProperty(secondEntity)).toBe(true);
+      expect(entities[firstEntity].alias).toBe(alias);
+      expect(entities[secondEntity].alias).toBeUndefined();
+    });
   });
 
   describe('attribute name', () => {