gns3-web-ui/src/app/cartography/widgets/links.spec.ts

import { instance, mock } from "ts-mockito";
import { Selection } from "d3-selection";


import { TestSVGCanvas } from "../testing";
import { Layer } from "../models/layer";
import { LinksWidget } from "./links";
import { Node } from "../models/node";
import { Link } from "../../models/link";
import { LinkWidget } from "./link";
import { MultiLinkCalculatorHelper } from "../helpers/multi-link-calculator-helper";


describe('LinksWidget', () => {
  let svg: TestSVGCanvas;
  let widget: LinksWidget;
  let layersEnter: Selection<SVGGElement, Layer, SVGGElement, any>;
  let layer: Layer;
  let linkWidget: LinkWidget;

  beforeEach(() => {
    svg = new TestSVGCanvas();
    linkWidget = instance(mock(LinkWidget));
    widget = new LinksWidget(new MultiLinkCalculatorHelper(), linkWidget);

    const node_1 = new Node();
    node_1.node_id = "1";
    node_1.x = 10;
    node_1.y = 10;

    const node_2 = new Node();
    node_2.node_id = "2";
    node_2.x = 100;
    node_2.y = 100;

    const link_1 = new Link();
    link_1.link_id = "link1";
    link_1.source = node_1;
    link_1.target = node_2;
    link_1.link_type = "ethernet";

    layer = new Layer();
    layer.index = 1;

    layer.links = [link_1];

    const layers = [layer];

    const layersSelection = svg.canvas
        .selectAll<SVGGElement, Layer>('g.layer')
        .data(layers);

    layersEnter = layersSelection
        .enter()
            .append<SVGGElement>('g')
            .attr('class', 'layer');

    layersSelection
        .exit()
            .remove();
  });

  afterEach(() => {
    svg.destroy();
  });

  it('should draw links', () => {
    widget.draw(layersEnter);

    const drew = svg.canvas.selectAll<SVGGElement, Link>('g.link');
    const linkNode = drew.nodes()[0];
    expect(linkNode.getAttribute('link_id')).toEqual('link1');
    expect(linkNode.getAttribute('map-source')).toEqual('1');
    expect(linkNode.getAttribute('map-target')).toEqual('2');
  });

});