chore: Fix unit tests

mermaid-js · Sep 5, 2023 · fd7406b · fd7406b
1 parent c38478f
commit fd7406b
Showing 1 changed file with 8 additions and 10 deletions.
diff --git a/packages/mermaid/src/diagrams/class/classDiagram.spec.ts b/packages/mermaid/src/diagrams/class/classDiagram.spec.ts
@@ -814,19 +814,17 @@ describe('given a class diagram with members and methods ', function () {
     });
 
     it('should handle direct member declaration', function () {
-      const str = 'classDiagram\n' + 'Car : wheels';
-
-      parser.parse(str);
-      expect(classDb.getClasses()).toHaveProperty('Car');
-      expect(classDb.getClasses()['Car']['members']).toContain('wheels');
+      parser.parse('classDiagram\n' + 'Car : wheels');
+      const car = classDb.getClass('Car');
+      expect(car.members.length).toBe(1);
+      expect(car.members[0].id).toBe('wheels');
     });
 
     it('should handle direct member declaration with type', function () {
-      const str = 'classDiagram\n' + 'Car : int wheels';
-
-      parser.parse(str);
-      expect(classDb.getClasses()).toHaveProperty('Car');
-      expect(classDb.getClasses()['Car']['members']).toContain('int wheels');
+      parser.parse('classDiagram\n' + 'Car : int wheels');
+      const car = classDb.getClass('Car');
+      expect(car.members.length).toBe(1);
+      expect(car.members[0].id).toBe('int wheels');
     });
 
     it('should handle simple member declaration with type', function () {