import { addVertices, addEdges } from './flowRenderer'; import { setConfig } from '../../config'; setConfig({ flowchart: { htmlLabels: false } }); describe('the flowchart renderer', function() { describe('when adding vertices to a graph', function() { [ ['round', 'rect', 5], ['square', 'rect'], ['diamond', 'question'], ['hexagon', 'hexagon'], ['odd', 'rect_left_inv_arrow'], ['lean_right', 'lean_right'], ['lean_left', 'lean_left'], ['trapezoid', 'trapezoid'], ['inv_trapezoid', 'inv_trapezoid'], ['odd_right', 'rect_left_inv_arrow'], ['circle', 'circle'], ['ellipse', 'ellipse'], ['stadium', 'stadium'], ['group', 'rect'] ].forEach(function([type, expectedShape, expectedRadios = 0]) { it(`should add the correct shaped node to the graph for vertex type ${type}`, function() { const addedNodes = []; const mockG = { setNode: function(id, object) { addedNodes.push([id, object]); } }; addVertices( { v1: { type, id: 'my-node-id', classes: [], styles: [], text: 'my vertex text' } }, mockG, 'svg-id' ); expect(addedNodes).toHaveLength(1); expect(addedNodes[0][0]).toEqual('my-node-id'); expect(addedNodes[0][1]).toHaveProperty('id', 'my-node-id'); expect(addedNodes[0][1]).toHaveProperty('labelType', 'svg'); expect(addedNodes[0][1]).toHaveProperty('shape', expectedShape); expect(addedNodes[0][1]).toHaveProperty('rx', expectedRadios); expect(addedNodes[0][1]).toHaveProperty('ry', expectedRadios); }); }); }); [ [['fill:#fff'], 'fill:#fff;', ''], [['color:#ccc'], 'color:#ccc;', 'color:#ccc;'], [['fill:#fff', 'color:#ccc'], 'fill:#fff;color:#ccc;', 'color:#ccc;'], [ ['fill:#fff', 'color:#ccc', 'text-align:center'], 'fill:#fff;color:#ccc;text-align:center;', 'color:#ccc;text-align:center;' ] ].forEach(function([style, expectedStyle, expectedLabelStyle]) { it(`should add the styles to style and/or labelStyle for style ${style}`, function() { const addedNodes = []; const mockG = { setNode: function(id, object) { addedNodes.push([id, object]); } }; addVertices( { v1: { type: 'rect', id: 'my-node-id', classes: [], styles: style, text: 'my vertex text' } }, mockG, 'svg-id' ); expect(addedNodes).toHaveLength(1); expect(addedNodes[0][0]).toEqual('my-node-id'); expect(addedNodes[0][1]).toHaveProperty('id', 'my-node-id'); expect(addedNodes[0][1]).toHaveProperty('labelType', 'svg'); expect(addedNodes[0][1]).toHaveProperty('style', expectedStyle); expect(addedNodes[0][1]).toHaveProperty('labelStyle', expectedLabelStyle); }); }); describe('when adding edges to a graph', function() { it('should handle multiline texts and set centered label position', function() { const addedEdges = []; const mockG = { setEdge: function(s, e, data, c) { addedEdges.push(data); } }; addEdges( [ { text: 'Multi
Line' }, { text: 'Multi
Line' }, { text: 'Multi
Line' }, { style: ['stroke:DarkGray', 'stroke-width:2px'], text: 'Multi
Line' }, { style: ['stroke:DarkGray', 'stroke-width:2px'], text: 'Multi
Line' }, { style: ['stroke:DarkGray', 'stroke-width:2px'], text: 'Multi
Line' } ], mockG, 'svg-id' ); addedEdges.forEach(function(edge) { expect(edge).toHaveProperty('label', 'Multi\nLine'); expect(edge).toHaveProperty('labelpos', 'c'); }); }); }); });