use of alluxio.hub.proto.AlluxioCluster in project alluxio by Alluxio.
the class AgentProcessContextTest method testHeartbeatSuccess.
@Test
public void testHeartbeatSuccess() throws Exception {
InstancedConfiguration config = getTestSpecificConfig();
AgentProcessContext ctx = new AgentProcessContext(config, mClient);
CommonUtils.waitFor("cluster size to be > 0", () -> mContext.getAlluxioCluster().size() > 0, WaitForOptions.defaults().setInterval(100).setTimeoutMs(5000));
assertEquals(1, mContext.getAlluxioCluster().size());
AlluxioCluster c = mContext.getAlluxioCluster().toProto();
AlluxioNodeStatus nodeStatus = c.getNode(0);
String host = NetworkAddressUtils.getConnectHost(NetworkAddressUtils.ServiceType.HUB_AGENT_RPC, config);
assertEquals(host, nodeStatus.getHostname());
assertThrows(Exception.class, ctx::startHeartbeat);
}
use of alluxio.hub.proto.AlluxioCluster in project alluxio by Alluxio.
the class ManagerAgentServiceTest method agentHeartbeatTest.
@Test
public void agentHeartbeatTest() {
AlluxioNodeStatus s = AlluxioNodeStatus.newBuilder().setHostname(UUID.randomUUID().toString()).addProcess(AlluxioProcessStatus.newBuilder().setState(ProcessState.RUNNING).setNodeType(AlluxioNodeType.MASTER).build()).build();
HubNodeAddress addr = HubNodeAddress.newBuilder().setHostname(UUID.randomUUID().toString()).setRpcPort(56565).build();
AgentHeartbeatRequest request = AgentHeartbeatRequest.newBuilder().setAlluxioStatus(s).setHubNode(addr).build();
AgentHeartbeatResponse resp = mClient.agentHeartbeat(request);
assertTrue(resp.hasOk());
assertTrue(resp.getOk());
assertEquals(1, mContext.getAlluxioCluster().size());
AlluxioCluster c = mContext.getAlluxioCluster().toProto();
AlluxioNodeStatus nodeStatus = c.getNode(0);
assertEquals(s.getHostname(), nodeStatus.getHostname());
assertEquals(AlluxioNodeType.MASTER, nodeStatus.getProcess(0).getNodeType());
assertEquals(ProcessState.RUNNING, nodeStatus.getProcess(0).getState());
}
Aggregations