@Test
public void testGraph() throws Exception {
Graph graph = new PeopleGraphImpl().graphFor(1);
graph.add(new PersonImpl(1, "Zequinha", 10.0));
graph.add(new PersonImpl(2, "Barbosa", 20.0));
List<Person> closest = graph.getClosestFriends(5);
assertEquals(2, closest.size());
}
@Test
public void graphWithTwoNodesAndTwoNeighbours() throws Exception {
Node node = new Node("1");
node.addNeighbour(new Node("3"));
Node node2 = new Node("2");
node2.addNeighbour(new Node("4"));
graph.add(node);
graph.add(node2);
assertEquals("1 3\n2 4\n", serializeGraph());
}
@Test
public void graphWithOneNodeAndOneNeighbourReturnsNodeAndNeighbour() throws Exception {
Node node = new Node("1");
node.addNeighbour(new Node("2"));
graph.add(node);
assertEquals("1 2\n", serializeGraph());
}
private void addNode(String nodeName) {
graph.add(new Node(nodeName));
}
