@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(); }