From c0255d93811ecf5246d3f89d4f3b377dfb4aeecb Mon Sep 17 00:00:00 2001 From: Tom PERRILLAT-COLLOMB Date: Thu, 24 Aug 2023 16:15:20 +0200 Subject: [PATCH] mend --- .../src/diagrams/er/parser/erDiagram.spec.js | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/packages/mermaid/src/diagrams/er/parser/erDiagram.spec.js b/packages/mermaid/src/diagrams/er/parser/erDiagram.spec.js index 98954edb3a7..6b0865a2d47 100644 --- a/packages/mermaid/src/diagrams/er/parser/erDiagram.spec.js +++ b/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', () => {