mermaid/src/diagrams/classDiagram/classDiagram.spec.js

/**
 * Created by knut on 14-11-18.
 */
describe('class diagram, ', function () {
    describe('when parsing an info graph it', function () {
        var cd, cDDb;
        beforeEach(function () {
            cd = require('./parser/classDiagram').parser;
            cDDb = require('./classDb');
            cd.yy = cDDb;
        });

        it('should handle relation definitions', function () {
            var str = `classDiagram
Class01 <|-- Class02
Class03 *-- Class04
Class05 o-- Class06
Class07 .. Class08
Class09 -- Class1`;

            cd.parse(str);
        });
        it('should handle relation definition of different types and directions', function () {
            var str = `classDiagram
Class11 <|.. Class12
Class13 --> Class14
Class15 ..> Class16
Class17 ..|> Class18
Class19 <--* Class20`;

            cd.parse(str);
        });

        it('should handle cardinality and labels', function () {
            var str = `classDiagram
Class01 "1" *-- "many" Class02 : contains
Class03 o-- Class04 : aggregation
Class05 --> "1" Class06`;

            cd.parse(str);
        });
        it('should handle class definitions', function () {
            var str = `classDiagram
class Car
Driver -- Car : drives >
Car *-- Wheel : have 4 >
Car -- Person : < owns`;

            cd.parse(str);
        });

        it('should handle method statements', function () {
            var str = `classDiagram
Object <|-- ArrayList
Object : equals()
ArrayList : Object[] elementData
ArrayList : size()`;

            cd.parse(str);
        });
        it('should handle parsing of method statements  grouped by brackets', function () {
            var str = `classDiagram
class Dummy {
String data
  void methods()
}

class Flight {
   flightNumber : Integer
   departureTime : Date
}`;

            cd.parse(str);
        });

        it('should handle parsing of separators', function () {
            var str = `classDiagram
class Foo1 {
  You can use
  several lines
  ..
  as you want
  and group
  ==
  things together.
  __
  You can have as many groups
  as you want
  --
  End of class
}

class User {
  .. Simple Getter ..
  + getName()
  + getAddress()
  .. Some setter ..
  + setName()
  __ private data __
  int age
  -- encrypted --
  String password
}`;

            cd.parse(str);
        });

    });

    describe('when fetching data from an classDiagram graph it', function () {
        var cd, cDDb;
        beforeEach(function () {
            cd = require('./parser/classDiagram').parser;
            cDDb = require('./classDb');
            cd.yy = cDDb;
            cd.yy.clear();
        });
        it('should handle relation definitions EXTENSION', function () {
            var str = `classDiagram
                       Class01 <|-- Class02`;

            cd.parse(str);

            var relations = cd.yy.getRelations();

            expect(cd.yy.getClass('Class01').id).toBe('Class01');
            expect(cd.yy.getClass('Class02').id).toBe('Class02');
            expect(relations[0].relation.type1).toBe(cDDb.relationType.EXTENSION);
            expect(relations[0].relation.type2).toBe('none');
            expect(relations[0].relation.lineType).toBe(cDDb.lineType.LINE);
        });
        it('should handle relation definitions AGGREGATION and dotted line', function () {
            var str = `classDiagram
                       Class01 o.. Class02`;

            cd.parse(str);

            var relations = cd.yy.getRelations();

            expect(cd.yy.getClass('Class01').id).toBe('Class01');
            expect(cd.yy.getClass('Class02').id).toBe('Class02');
            expect(relations[0].relation.type1).toBe(cDDb.relationType.AGGREGATION);
            expect(relations[0].relation.type2).toBe('none');
            expect(relations[0].relation.lineType).toBe(cDDb.lineType.DOTTED_LINE);
        });
        it('should handle relation definitions COMPOSITION on both sides', function () {
            var str = `classDiagram
                       Class01 *--* Class02`;

            cd.parse(str);

            var relations = cd.yy.getRelations();

            expect(cd.yy.getClass('Class01').id).toBe('Class01');
            expect(cd.yy.getClass('Class02').id).toBe('Class02');
            expect(relations[0].relation.type1).toBe(cDDb.relationType.COMPOSITION);
            expect(relations[0].relation.type2).toBe(cDDb.relationType.COMPOSITION);
            expect(relations[0].relation.lineType).toBe(cDDb.lineType.LINE);
        });
        it('should handle relation definitions no types', function () {
            var str = `classDiagram
                       Class01 -- Class02`;

            cd.parse(str);

            var relations = cd.yy.getRelations();

            expect(cd.yy.getClass('Class01').id).toBe('Class01');
            expect(cd.yy.getClass('Class02').id).toBe('Class02');
            expect(relations[0].relation.type1).toBe('none');
            expect(relations[0].relation.type2).toBe('none');
            expect(relations[0].relation.lineType).toBe(cDDb.lineType.LINE);
        });
        it('should handle relation definitions with type only on right side', function () {
            var str = `classDiagram
                       Class01 --|> Class02`;

            cd.parse(str);

            var relations = cd.yy.getRelations();

            expect(cd.yy.getClass('Class01').id).toBe('Class01');
            expect(cd.yy.getClass('Class02').id).toBe('Class02');
            expect(relations[0].relation.type1).toBe('none');
            expect(relations[0].relation.type2).toBe(cDDb.relationType.EXTENSION);
            expect(relations[0].relation.lineType).toBe(cDDb.lineType.LINE);
        });

        it('should handle multiple classes and relation definitions', function () {
            var str = `classDiagram
                        Class01 <|-- Class02
                        Class03 *-- Class04
                        Class05 o-- Class06
                        Class07 .. Class08
                        Class09 -- Class10`;

            cd.parse(str);

            var relations = cd.yy.getRelations();

            expect(cd.yy.getClass('Class01').id).toBe('Class01');
            expect(cd.yy.getClass('Class10').id).toBe('Class10');

            expect(relations.length).toBe(5);

            expect(relations[0].relation.type1).toBe(cDDb.relationType.EXTENSION);
            expect(relations[0].relation.type2).toBe('none');
            expect(relations[0].relation.lineType).toBe(cDDb.lineType.LINE);
            expect(relations[3].relation.type1).toBe('none');
            expect(relations[3].relation.type2).toBe('none');
            expect(relations[3].relation.lineType).toBe(cDDb.lineType.DOTTED_LINE);
        });
    });
});