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'], ['cylinder', 'cylinder'], ['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); }); }); [ 'Multi
Line', 'Multi
Line', 'Multi
Line', 'MultiLine' ].forEach(function(labelText) { it('should handle multiline texts with different line breaks', function() { const addedNodes = []; const mockG = { setNode: function(id, object) { addedNodes.push([id, object]); } }; addVertices( { v1: { type: 'rect', id: 'my-node-id', classes: [], styles: [], text: 'Multi
Line' } }, 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].label).toBeDefined(); expect(addedNodes[0][1].label).toBeDefined(); // node expect(addedNodes[0][1].label.firstChild.innerHTML).toEqual('Multi'); // node, line 1 expect(addedNodes[0][1].label.lastChild.innerHTML).toEqual('Line'); // node, line 2 }); }); [ [['fill:#fff'], 'fill:#fff;', ''], [['color:#ccc'], '', 'color:#ccc;'], [['fill:#fff', 'color:#ccc'], 'fill:#fff;', 'color:#ccc;'], [['fill:#fff', 'color:#ccc', 'text-align:center'], 'fill:#fff;', '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' }, { text: 'MultiLine' }, { 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' }, { style: ['stroke:DarkGray', 'stroke-width:2px'], text: 'MultiLine' } ], mockG, 'svg-id' ); addedEdges.forEach(function(edge) { expect(edge).toHaveProperty('label', 'Multi\nLine'); expect(edge).toHaveProperty('labelpos', 'c'); }); }); [ [['stroke:DarkGray'], 'stroke:DarkGray;', ''], [['color:red'], '', 'fill:red;'], [['stroke:DarkGray', 'color:red'], 'stroke:DarkGray;', 'fill:red;'], [['stroke:DarkGray', 'color:red', 'stroke-width:2px'], 'stroke:DarkGray;stroke-width:2px;', 'fill:red;'] ].forEach(function([style, expectedStyle, expectedLabelStyle]) { it(`should add the styles to style and/or labelStyle for style ${style}`, function() { const addedEdges = []; const mockG = { setEdge: function(s, e, data, c) { addedEdges.push(data); } }; addEdges( [ { style: style, text: 'styling' } ], mockG, 'svg-id' ); expect(addedEdges).toHaveLength(1); expect(addedEdges[0]).toHaveProperty('style', expectedStyle); expect(addedEdges[0]).toHaveProperty('labelStyle', expectedLabelStyle); }); }); }); });