@Test
public void testTriggerSavepointFailure() throws Exception {
replaceStdOutAndStdErr();
try {
JobID jobId = new JobID();
ActorGateway jobManager = mock(ActorGateway.class);
Promise<Object> triggerResponse = new scala.concurrent.impl.Promise.DefaultPromise<>();
when(jobManager.ask(
Mockito.eq(new JobManagerMessages.TriggerSavepoint(jobId)),
Mockito.any(FiniteDuration.class)))
.thenReturn(triggerResponse.future());
Exception testException = new Exception("expectedTestException");
triggerResponse.success(new JobManagerMessages.TriggerSavepointFailure(jobId, testException));
CliFrontend frontend = new MockCliFrontend(CliFrontendTestUtils.getConfigDir(), jobManager);
String[] parameters = {jobId.toString()};
int returnCode = frontend.savepoint(parameters);
assertTrue(returnCode != 0);
verify(jobManager, times(1))
.ask(
Mockito.eq(new JobManagerMessages.TriggerSavepoint(jobId)),
Mockito.any(FiniteDuration.class));
assertTrue(buffer.toString().contains("expectedTestException"));
} finally {
restoreStdOutAndStdErr();
}
}
@Test
public void testGenerateScopeCustom() {
MetricRegistry registry = new MetricRegistry(new Configuration());
TaskManagerScopeFormat tmFormat = new TaskManagerScopeFormat("abc");
TaskManagerJobScopeFormat jmFormat = new TaskManagerJobScopeFormat("def", tmFormat);
TaskScopeFormat taskFormat =
new TaskScopeFormat("<tm_id>.<job_id>.<task_id>.<task_attempt_id>", jmFormat);
JobID jid = new JobID();
AbstractID vertexId = new AbstractID();
AbstractID executionId = new AbstractID();
TaskManagerMetricGroup tmGroup =
new TaskManagerMetricGroup(registry, "theHostName", "test-tm-id");
JobMetricGroup jmGroup = new JobMetricGroup(registry, tmGroup, jid, "myJobName");
TaskMetricGroup taskGroup =
new TaskMetricGroup(
registry, jmGroup, taskFormat, vertexId, executionId, "aTaskName", 13, 2);
assertArrayEquals(
new String[] {"test-tm-id", jid.toString(), vertexId.toString(), executionId.toString()},
taskGroup.getScopeComponents());
assertEquals(
String.format("test-tm-id.%s.%s.%s", jid, vertexId, executionId),
taskGroup.getScopeString());
}
protected String internalJobId() {
return jobID.toString();
}
