@Test
public void getNodes_by_type() {
Node nn = Nodes.addNode(new Node("nn", Node.Type.NAMENODE));
Node dn0 = Nodes.addNode(new Node("dn0", Node.Type.DATANODE));
Node dn1 = Nodes.addNode(new Node("dn1", Node.Type.DATANODE));
assertEquals(Arrays.asList(nn), Nodes.getNodes(Node.Type.NAMENODE));
assertEquals(Arrays.asList(dn0, dn1), Nodes.getNodes(Node.Type.DATANODE));
}
@Test
public void removeNode() {
Node n0 = Nodes.addNode(new Node("n0", Node.Type.NAMENODE));
Node n1 = Nodes.addNode(new Node("n1", Node.Type.DATANODE));
Node n2 = Nodes.addNode(new Node("n2", Node.Type.DATANODE));
assertEquals(Arrays.asList(n0, n1, n2), Nodes.getNodes());
Nodes.removeNode(n1);
assertEquals(Arrays.asList(n0, n2), Nodes.getNodes());
Nodes.removeNode(n1);
}
@Test
public void getNodes_by_state() {
Node n0 = Nodes.addNode(new Node("n0"));
Node n1 = Nodes.addNode(new Node("n1", Node.Type.DATANODE));
Node n2 = Nodes.addNode(new Node("n2", Node.Type.DATANODE));
n1.state = Node.State.RUNNING;
n2.state = Node.State.RUNNING;
assertEquals(Arrays.asList(n0), Nodes.getNodes(Node.State.IDLE));
assertEquals(Arrays.asList(n1, n2), Nodes.getNodes(Node.State.RUNNING));
}
public static Nodes rndNodes(int max) {
int nnodes = rndInt(0, 5);
Nodes ns = new Nodes();
for (int i = nnodes; i >= 0; i--) {
Node n = rndNode();
ns.getNodes().add(n);
}
return ns;
}
@Test
public void toJson_fromJson() {
Nodes.frameworkId = "id";
Node n0 = Nodes.addNode(new Node("n0"));
Node n1 = Nodes.addNode(new Node("n1", Node.Type.DATANODE));
JSONObject json = Nodes.toJson();
Nodes.fromJson(json);
assertEquals("id", Nodes.frameworkId);
assertEquals(Arrays.asList(n0, n1), Nodes.getNodes());
}
@Test
public void addNode() {
Node nn = Nodes.addNode(new Node("nn", Node.Type.NAMENODE));
assertEquals(Arrays.asList(nn), Nodes.getNodes());
// duplicate id
try {
Nodes.addNode(new Node("nn", Node.Type.DATANODE));
fail();
} catch (IllegalArgumentException e) {
assertTrue(e.getMessage(), e.getMessage().contains("duplicate"));
}
// second namenode
try {
Nodes.addNode(new Node("nn1", Node.Type.NAMENODE));
fail();
} catch (IllegalArgumentException e) {
assertTrue(e.getMessage(), e.getMessage().contains("second"));
}
}
