diff --git a/demos/classchart.html b/demos/classchart.html
index 98162697e..3ad8fb100 100644
--- a/demos/classchart.html
+++ b/demos/classchart.html
@@ -38,12 +38,14 @@
+quack()
}
class Fish{
- -int sizeInFeet
+ -Listint sizeInFeet
-canEat()
}
class Zebra{
+bool is_wild
+run(List~T~, List~OT~)
+ %% +run-composite(List~T, K~)
+ +run-nested(List~List~OT~~)
}
@@ -154,7 +156,7 @@
~InternalProperty : string
~AnotherInternalProperty : List~List~string~~
}
- class People~List~Person~~
+ class People List~List~Person~~
diff --git a/packages/mermaid/src/diagrams/class/classDiagram.spec.ts b/packages/mermaid/src/diagrams/class/classDiagram.spec.ts
index f3568e7b8..b775521a4 100644
--- a/packages/mermaid/src/diagrams/class/classDiagram.spec.ts
+++ b/packages/mermaid/src/diagrams/class/classDiagram.spec.ts
@@ -1439,11 +1439,19 @@ class Class2
const testClasses = parser.yy.getClasses();
const testRelations = parser.yy.getRelations();
expect(Object.keys(testNamespaceA.classes).length).toBe(2);
- expect(testNamespaceA.classes['A1'].members[0]).toBe('+foo : string');
- expect(testNamespaceA.classes['A2'].members[0]).toBe('+bar : int');
+ expect(testNamespaceA.classes['A1'].members[0].getDisplayDetails().displayText).toBe(
+ '+foo : string'
+ );
+ expect(testNamespaceA.classes['A2'].members[0].getDisplayDetails().displayText).toBe(
+ '+bar : int'
+ );
expect(Object.keys(testNamespaceB.classes).length).toBe(2);
- expect(testNamespaceB.classes['B1'].members[0]).toBe('+foo : bool');
- expect(testNamespaceB.classes['B2'].members[0]).toBe('+bar : float');
+ expect(testNamespaceB.classes['B1'].members[0].getDisplayDetails().displayText).toBe(
+ '+foo : bool'
+ );
+ expect(testNamespaceB.classes['B2'].members[0].getDisplayDetails().displayText).toBe(
+ '+bar : float'
+ );
expect(Object.keys(testClasses).length).toBe(4);
expect(testClasses['A1'].parent).toBe('A');
expect(testClasses['A2'].parent).toBe('A');
diff --git a/packages/mermaid/src/diagrams/class/classTypes.spec.ts b/packages/mermaid/src/diagrams/class/classTypes.spec.ts
index 4a3493e0a..56a865fe3 100644
--- a/packages/mermaid/src/diagrams/class/classTypes.spec.ts
+++ b/packages/mermaid/src/diagrams/class/classTypes.spec.ts
@@ -382,42 +382,42 @@ describe('given text representing a method, ', function () {
const str = `getTimes(List~T~, List~OT~)`;
const classMember = new ClassMember(str, 'method');
- expect(classMember.getDisplayDetails().displayText).toBe('getTimes(List, List');
+ expect(classMember.getDisplayDetails().displayText).toBe('getTimes(List, List)');
});
it('should handle public visibility', function () {
const str = `+getTimes(List~T~, List~OT~)`;
const classMember = new ClassMember(str, 'method');
- expect(classMember.getDisplayDetails().displayText).toBe('+getTimes(List, List');
+ expect(classMember.getDisplayDetails().displayText).toBe('+getTimes(List, List)');
});
it('should handle private visibility', function () {
const str = `-getTimes(List~T~, List~OT~)`;
const classMember = new ClassMember(str, 'method');
- expect(classMember.getDisplayDetails().displayText).toBe('-getTimes(List, List');
+ expect(classMember.getDisplayDetails().displayText).toBe('-getTimes(List, List)');
});
it('should handle protected visibility', function () {
const str = `#getTimes(List~T~, List~OT~)`;
const classMember = new ClassMember(str, 'method');
- expect(classMember.getDisplayDetails().displayText).toBe('#getTimes(List, List');
+ expect(classMember.getDisplayDetails().displayText).toBe('#getTimes(List, List)');
});
it('should handle internal visibility', function () {
const str = `~getTimes(List~T~, List~OT~)`;
const classMember = new ClassMember(str, 'method');
- expect(classMember.getDisplayDetails().displayText).toBe('~getTimes(List, List');
+ expect(classMember.getDisplayDetails().displayText).toBe('~getTimes(List, List)');
});
it('should return correct css for static classifier', function () {
const str = `getTimes(List~T~, List~OT~)$`;
const classMember = new ClassMember(str, 'method');
- expect(classMember.getDisplayDetails().displayText).toBe('getTimes(List, List');
+ expect(classMember.getDisplayDetails().displayText).toBe('getTimes(List, List)');
expect(classMember.getDisplayDetails().cssStyle).toBe(staticCssStyle);
});