@Test
public void shouldNotAssignWorkToCanceledAgentsRegisteredInAgentRemoteHandler() {
AgentConfig agentConfig = AgentMother.remoteAgent();
configHelper.addAgent(agentConfig);
fixture.createPipelineWithFirstStageScheduled();
AgentRuntimeInfo info =
AgentRuntimeInfo.fromServer(agentConfig, true, "location", 1000000l, "OS");
info.setCookie("cookie");
agentRemoteHandler.process(agent, new Message(Action.ping, info));
AgentInstance agentInstance = agentService.findAgentAndRefreshStatus(info.getUUId());
agentInstance.cancel();
buildAssignmentService.onTimer();
assertThat(
"Should not assign work when agent status is Canceled", agent.messages.size(), is(0));
}
@Test
public void shouldThrowExceptionWhenADuplicateAgentTriesToUpdateStatus() throws Exception {
AgentRuntimeInfo runtimeInfo =
new AgentRuntimeInfo(
agentIdentifier, AgentRuntimeStatus.Idle, currentWorkingDirectory(), null, null);
runtimeInfo.setCookie("invalid_cookie");
AgentInstance original =
AgentInstance.createFromLiveAgent(
new AgentRuntimeInfo(
agentIdentifier, AgentRuntimeStatus.Idle, currentWorkingDirectory(), null, null),
new SystemEnvironment());
try {
when(agentService.findAgentAndRefreshStatus(runtimeInfo.getUUId())).thenReturn(original);
agentService.updateRuntimeInfo(runtimeInfo);
fail("should throw exception when cookie mismatched");
} catch (Exception e) {
assertThat(
e.getMessage(),
is(format("Agent [%s] has invalid cookie", runtimeInfo.agentInfoDebugString())));
assertThat(
Arrays.asList(logFixture.getMessages()),
hasItem(
format(
"Found agent [%s] with duplicate uuid. Please check the agent installation.",
runtimeInfo.agentInfoDebugString())));
verify(serverHealthService)
.update(
ServerHealthState.warning(
format(
"[%s] has duplicate unique identifier which conflicts with [%s]",
runtimeInfo.agentInfoForDisplay(), original.agentInfoForDisplay()),
"Please check the agent installation. Click <a href='http://www.go.cd/documentation/user/current/faq/agent_guid_issue.html' target='_blank'>here</a> for more info.",
HealthStateType.duplicateAgent(
HealthStateScope.forAgent(runtimeInfo.getCookie())),
Timeout.THIRTY_SECONDS));
}
verify(agentInstances).findAgentAndRefreshStatus(runtimeInfo.getUUId());
verifyNoMoreInteractions(agentInstances);
}