@Test
public void testRemove() {
String node1 = "node-A";
nodes.add(node1);
String node2 = "node-B";
nodes.add(node2);
int totalNodes = nodes.getTotalNodes();
assertEquals(4, totalNodes);
nodes.remove(node1);
totalNodes = nodes.getTotalNodes();
assertEquals(2, totalNodes);
}
@Test
public void testAdd() {
String node1 = "node-A";
nodes.add(node1);
int totalNodes = nodes.getTotalNodes();
assertEquals(2, totalNodes);
}
@Test
public void testGet() {
String node1 = "node-C";
nodes.add(node1);
String node2 = "node-D";
nodes.add(node2);
int totalNodes = nodes.getTotalNodes();
assertEquals(4, totalNodes);
String node = nodes.get("abcd");
assertEquals(node1, node);
node = nodes.get(node2);
assertEquals(node2, node);
}
