@Test public void testNodeMaxActions() throws Exception { conf.setInt(Constants.MAX_PER_NODE_NUM_ACTIONS, 3); conf.setInt(Constants.MAX_PER_NODE_LOG_LENGTH, 100); NodeService nodeService = injector.getInstance(NodeService.class); Node node = new Node("1", "1", ImmutableSet.<Service>of(), TestHelper.EMPTY_NODE_PROPERTIES); Assert.assertTrue(node.getActions().isEmpty()); for (int i = 0; i < 3; ++i) { Assert.assertEquals(i, node.getActions().size()); nodeService.startAction(node, "taskId", "service" + i, "action" + i); nodeService.completeAction(node); Assert.assertEquals(i + 1, node.getActions().size()); } nodeService.startAction(node, "taskId", "service3", "action3"); nodeService.completeAction(node); Assert.assertEquals(3, node.getActions().size()); Assert.assertEquals(Lists.newArrayList("service1", "service2", "service3"), Lists.newArrayList(Iterables.transform(node.getActions(), new Function<Node.Action, String>() { @Override public String apply(Node.Action input) { return input.getService(); } }))); }
node.getProperties().setHostname(hostname); nodeService.completeAction(node); } else { nodeService.failAction(node, finish.getStdout(), finish.getStderr());