@Test public void getNodes() { 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()); }
@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 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 expandExpr() { Nodes.addNode(new Node("nn", Node.Type.NAMENODE)); Nodes.addNode(new Node("dn0", Node.Type.DATANODE)); Nodes.addNode(new Node("dn1", Node.Type.DATANODE)); // id list assertEquals(Arrays.asList("nn", "dn2"), Nodes.expandExpr("nn,dn2")); // wildcard assertEquals(Arrays.asList("dn0", "dn1"), Nodes.expandExpr("dn*")); assertEquals(Arrays.asList("nn", "dn0", "dn1"), Nodes.expandExpr("*")); // range assertEquals(Arrays.asList("1", "2", "3"), Nodes.expandExpr("1..3")); assertEquals(Arrays.asList("dn1", "dn2", "dn3"), Nodes.expandExpr("dn1..3")); }
@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")); } }
@Test public void getNode() { assertNull(Nodes.getNode("n0")); Node n0 = Nodes.addNode(new Node("n0")); assertSame(n0, Nodes.getNode("n0")); }