@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)); }
@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")); } }