@Test public void testRemoveEdge() { Diagram diagram = new Diagram(); String edgeId = "e1"; String from = "n1"; String to = "n2"; diagram.addEdge(edgeId, from, to, Point.of(1, 2), Point.of(3, 4), Point.of(5, 6)); String edgeId2 = "e2"; String from2 = "n3"; String to2 = "n4"; diagram.addEdge(edgeId2, from2, to2, Point.of(1, 2), Point.of(3, 4), Point.of(5, 6)); assertNotNull(diagram.canvas); assertEquals(2, diagram.edges.size()); diagram.removeEdge(edgeId2); assertEquals(1, diagram.edges.size()); assertEquals(edgeId, diagram.edges.get(0).transitionId); }