return agentStatus.getShortAgentId(); case uuid: return agentStatus.getAgentId(); case machine: return agentStatus.getInstanceId(); case internalHost: return agentStatus.getInternalHost(); case internalIp: return agentStatus.getInternalIp(); case externalHost: return agentStatus.getExternalHost(); case status: return agentStatus.getState(); case location: return agentStatus.getLocation(); case shortLocation: return agentStatus.getShortLocation(); case instanceType: return agentStatus.getInstanceType(); case internalUri: return agentStatus.getSelf(); case externalUri: return agentStatus.getExternalUri(); default: return null;
@Test public void testJsonDecode() throws Exception { String json = Resources.toString(Resources.getResource("agent-status.json"), Charsets.UTF_8); AgentStatusRepresentation actual = codec.fromJson(json); assertEquals(actual, expected); assertEquals(actual.getAgentId(), expected.getAgentId()); assertEquals(actual.getShortAgentId(), expected.getShortAgentId()); assertEquals(actual.getInstanceId(), expected.getInstanceId()); assertEquals(actual.getSelf(), expected.getSelf()); assertEquals(actual.getState(), expected.getState()); assertEquals(actual.getInstanceType(), expected.getInstanceType()); assertEquals(actual.getResources(), expected.getResources()); assertEquals(actual.getLocation(), expected.getLocation()); assertEquals(actual.getShortLocation(), expected.getShortLocation()); assertEquals(actual.getSlots(), expected.getSlots()); assertEquals(actual.getVersion(), expected.getVersion()); } }
public static Function<AgentStatus, AgentStatusRepresentation> fromAgentStatus(List<AgentStatus> agentStatuses, Repository repository) { return fromAgentStatus(new AgentStatusRepresentationFactory(agentStatuses, repository)); }
@Override public void onSuccess(AgentStatusRepresentation agent) { future.set(new Instance(agent.getInstanceId(), firstNonNull(agent.getInstanceType(), "unknown"), agent.getLocation(), agent.getSelf(), agent.getExternalUri())); }
AgentStatusRepresentation actual = agents.iterator().next(); assertEquals(actual.getAgentId(), agentId); assertEquals(actual.getState(), AgentLifecycleState.ONLINE); assertEquals(actual.getInstanceId(), instanceId); assertEquals(actual.getLocation(), location); assertEquals(actual.getInstanceType(), instanceType); assertEquals(actual.getSelf(), internalUri); assertEquals(actual.getExternalUri(), externalUri); assertEquals(actual.getResources(), resources);
@Test public void testGetAllSlotsStatus() { Response response = agentResource.getAllSlotsStatus(); AgentStatusRepresentation actual = (AgentStatusRepresentation) response.getEntity(); assertEquals(actual.getAgentId(), agent.getAgentId()); assertEquals(actual.getState(), ONLINE); assertEquals(actual.getInstanceType(), null); assertEquals(actual.getResources(), TEST_RESOURCES); assertEquals(actual.getLocation(), agent.getLocation()); } }
assertEquals(slots.get(BANANA_ASSIGNMENT_EXACT).size(), 0); execute("show", "-h", agent.getInternalHost()); slots = slotsByAssignment(); assertEquals(slots.get(APPLE_ASSIGNMENT).size(), 2); assertEquals(slots.get(BANANA_ASSIGNMENT_EXACT).size(), 2); execute("show", "-h", agent.getInternalIp()); slots = slotsByAssignment(); assertEquals(slots.get(APPLE_ASSIGNMENT).size(), 2); assertEquals(slots.get(BANANA_ASSIGNMENT_EXACT).size(), 2); execute("show", "-h", agent.getExternalHost()); slots = slotsByAssignment(); assertEquals(slots.get(APPLE_ASSIGNMENT).size(), 2); assertEquals(slots.get(BANANA_ASSIGNMENT_EXACT).size(), 2); execute("show", "-m", agent.getInstanceId()); slots = slotsByAssignment(); assertEquals(slots.get(APPLE_ASSIGNMENT).size(), 1);
private List<AgentStatusRepresentation> waitForAgentsToStart(List<String> instanceIds) { for (int loop = 0; true; loop++) { try { URI uri = uriBuilderFrom(coordinatorUri).replacePath("v1/admin/agent").build(); Request request = Request.Builder.prepareGet() .setUri(uri) .build(); List<AgentStatusRepresentation> agents = client.execute(request, createJsonResponseHandler(AGENTS_CODEC)); Map<String, AgentStatusRepresentation> runningAgents = newHashMap(); for (AgentStatusRepresentation agent : agents) { if (agent.getState() == AgentLifecycleState.ONLINE) { runningAgents.put(agent.getInstanceId(), agent); } } if (runningAgents.keySet().containsAll(instanceIds)) { WaitUtils.clearWaitMessage(); runningAgents.keySet().retainAll(instanceIds); return ImmutableList.copyOf(runningAgents.values()); } } catch (Exception ignored) { } WaitUtils.wait(loop); } }
List<String> instanceIds = newArrayList(); for (AgentStatusRepresentation agent : agents) { instanceIds.add(agent.getInstanceId());
uuidFilter.uuid.add(agent.getAgentId());
public String getInternalIp() { String host = getInternalHost(); if (host == null) { return null; } if ("localhost".equalsIgnoreCase(host)) { return "127.0.0.1"; } try { return InetAddress.getByName(host).getHostAddress(); } catch (UnknownHostException e) { return "unknown"; } }
AgentStatusRepresentation actual = agents.iterator().next(); assertEquals(actual.getAgentId(), agentId); assertEquals(actual.getState(), AgentLifecycleState.ONLINE); assertEquals(actual.getInstanceId(), instanceId); assertEquals(actual.getLocation(), location); assertEquals(actual.getInstanceType(), instanceType); assertEquals(actual.getSelf(), internalUri); assertEquals(actual.getExternalUri(), externalUri); assertEquals(actual.getResources(), resources);
private void assertSlotStatus(SlotStatusRepresentation slot, UUID expectedSlotId, Assignment expectedAssignment, SlotLifecycleState expectedState, AgentStatusRepresentation expectedAgent) { assertEquals(slot.getId(), expectedSlotId); if (expectedState != SlotLifecycleState.TERMINATED) { assertEquals(slot.getBinary(), expectedAssignment.getBinary()); assertEquals(slot.getConfig(), expectedAssignment.getConfig()); assertTrue(slot.getInstallPath().startsWith(tempDir.getAbsolutePath())); } else { assertNull(slot.getBinary()); assertNull(slot.getConfig()); assertNull(slot.getInstallPath()); } assertEquals(slot.getStatus(), expectedState.toString()); assertEquals(slot.getInstanceId(), expectedAgent.getInstanceId()); assertTrue(slot.getLocation().startsWith(expectedAgent.getLocation())); assertTrue(slot.getLocation().endsWith(slot.getShortLocation())); assertTrue(slot.getSelf().toASCIIString().startsWith(expectedAgent.getSelf().toASCIIString())); assertTrue(slot.getExternalUri().toASCIIString().startsWith(expectedAgent.getExternalUri().toASCIIString())); }
@Test public void testGetAllSlotsStatus() { Response response = agentResource.getAllSlotsStatus(); AgentStatusRepresentation actual = (AgentStatusRepresentation) response.getEntity(); assertEquals(actual.getAgentId(), agent.getAgentId()); assertEquals(actual.getState(), ONLINE); assertEquals(actual.getInstanceType(), null); assertEquals(actual.getResources(), TEST_RESOURCES); assertEquals(actual.getLocation(), agent.getLocation()); }
execute("show", "-h", agent.getInternalHost()); slots = slotsByAssignment(); assertEquals(slots.get(APPLE_ASSIGNMENT).size(), 2); execute("show", "-H", agent.getInternalHost()); slots = slotsByAssignment(); assertEquals(slots.get(APPLE_ASSIGNMENT).size(), 0); execute("show", "-h", agent.getInternalIp()); slots = slotsByAssignment(); assertEquals(slots.get(APPLE_ASSIGNMENT).size(), 2); execute("show", "-H", agent.getInternalIp()); slots = slotsByAssignment(); assertEquals(slots.get(APPLE_ASSIGNMENT).size(), 0); execute("show", "-h", agent.getExternalHost()); slots = slotsByAssignment(); assertEquals(slots.get(APPLE_ASSIGNMENT).size(), 2); execute("show", "-H", agent.getExternalHost()); slots = slotsByAssignment(); assertEquals(slots.get(APPLE_ASSIGNMENT).size(), 0); execute("show", "-m", agent.getInstanceId()); slots = slotsByAssignment(); assertEquals(slots.get(APPLE_ASSIGNMENT).size(), 2); execute("show", "-M", agent.getInstanceId());
private List<AgentStatusRepresentation> waitForAgentsToStart(List<String> instanceIds) { for (int loop = 0; true; loop++) { try { URI uri = uriBuilderFrom(coordinatorUri).replacePath("v1/admin/agent").build(); Request request = Request.Builder.prepareGet() .setUri(uri) .build(); List<AgentStatusRepresentation> agents = client.execute(request, createJsonResponseHandler(AGENTS_CODEC)); Map<String, AgentStatusRepresentation> runningAgents = newHashMap(); for (AgentStatusRepresentation agent : agents) { if (agent.getState() == AgentLifecycleState.ONLINE) { runningAgents.put(agent.getInstanceId(), agent); } } if (runningAgents.keySet().containsAll(instanceIds)) { WaitUtils.clearWaitMessage(); runningAgents.keySet().retainAll(instanceIds); return ImmutableList.copyOf(runningAgents.values()); } } catch (Exception ignored) { } WaitUtils.wait(loop); } }
List<String> instanceIds = newArrayList(); for (AgentStatusRepresentation agent : agents) { instanceIds.add(agent.getInstanceId());
uuidFilter.uuid.add(agent.getAgentId());
assertEquals(actual.getAgentId(), agentId); assertEquals(actual.getState(), AgentLifecycleState.ONLINE); assertEquals(actual.getInstanceId(), instanceId); assertEquals(actual.getLocation(), location); assertEquals(actual.getInstanceType(), instanceType); assertEquals(actual.getSelf(), internalUri); assertEquals(actual.getExternalUri(), externalUri); assertEquals(actual.getResources(), resources);
return agentStatus.getShortAgentId(); case uuid: return agentStatus.getAgentId(); case machine: return agentStatus.getInstanceId(); case internalHost: return agentStatus.getInternalHost(); case internalIp: return agentStatus.getInternalIp(); case externalHost: return agentStatus.getExternalHost(); case status: return agentStatus.getState(); case location: return agentStatus.getLocation(); case shortLocation: return agentStatus.getShortLocation(); case instanceType: return agentStatus.getInstanceType(); case internalUri: return agentStatus.getSelf(); case externalUri: return agentStatus.getExternalUri(); default: return null;