Examples with TestingClusterClient org.apache.flink.client.program.TestingClusterClient used on opensource projects

Example 6 with TestingClusterClient

use of org.apache.flink.client.program.TestingClusterClient in project flink by apache.

the class CliFrontendStopWithSavepointTest method testUnknownJobId.

@Test
public void testUnknownJobId() throws Exception {
    // test unknown job Id
    JobID jid = new JobID();
    String[] parameters = { "-p", "test-target-dir", jid.toString() };
    String expectedMessage = "Test exception";
    FlinkException testException = new FlinkException(expectedMessage);
    TestingClusterClient<String> clusterClient = new TestingClusterClient<>();
    clusterClient.setStopWithSavepointFunction((jobID, advanceToEndOfEventTime, savepointDirectory, formatType) -> FutureUtils.completedExceptionally(testException));
    MockedCliFrontend testFrontend = new MockedCliFrontend(clusterClient);
    try {
        testFrontend.stop(parameters);
        fail("Should have failed.");
    } catch (FlinkException e) {
        assertTrue(ExceptionUtils.findThrowableWithMessage(e, expectedMessage).isPresent());
    }
}

Example 7 with TestingClusterClient

use of org.apache.flink.client.program.TestingClusterClient in project flink by apache.

the class CliFrontendStopWithSavepointTest method testStopOnlyWithMaxWM.

@Test
public void testStopOnlyWithMaxWM() throws Exception {
    JobID jid = new JobID();
    String[] parameters = { "-d", jid.toString() };
    OneShotLatch stopWithSavepointLatch = new OneShotLatch();
    TestingClusterClient<String> clusterClient = new TestingClusterClient<>();
    clusterClient.setStopWithSavepointFunction((jobID, advanceToEndOfEventTime, savepointDirectory, formatType) -> {
        assertThat(jobID, is(jid));
        assertThat(advanceToEndOfEventTime, is(true));
        assertNull(savepointDirectory);
        stopWithSavepointLatch.trigger();
        return CompletableFuture.completedFuture(savepointDirectory);
    });
    MockedCliFrontend testFrontend = new MockedCliFrontend(clusterClient);
    testFrontend.stop(parameters);
    stopWithSavepointLatch.await();
}

Example 8 with TestingClusterClient

use of org.apache.flink.client.program.TestingClusterClient in project flink by apache.

the class CliFrontendStopWithSavepointTest method testStopWithDefaultSavepointDir.

@Test
public void testStopWithDefaultSavepointDir() throws Exception {
    JobID jid = new JobID();
    String[] parameters = { jid.toString() };
    OneShotLatch stopWithSavepointLatch = new OneShotLatch();
    TestingClusterClient<String> clusterClient = new TestingClusterClient<>();
    clusterClient.setStopWithSavepointFunction((jobID, advanceToEndOfEventTime, savepointDirectory, formatType) -> {
        assertThat(jobID, is(jid));
        assertThat(advanceToEndOfEventTime, is(false));
        assertNull(savepointDirectory);
        stopWithSavepointLatch.trigger();
        return CompletableFuture.completedFuture(savepointDirectory);
    });
    MockedCliFrontend testFrontend = new MockedCliFrontend(clusterClient);
    testFrontend.stop(parameters);
    stopWithSavepointLatch.await();
}

Example 9 with TestingClusterClient

use of org.apache.flink.client.program.TestingClusterClient in project flink by apache.

the class CliFrontendStopWithSavepointTest method testStopWithMaxWMAndDefaultSavepointDir.

@Test
public void testStopWithMaxWMAndDefaultSavepointDir() throws Exception {
    JobID jid = new JobID();
    String[] parameters = { "-p", "-d", jid.toString() };
    OneShotLatch stopWithSavepointLatch = new OneShotLatch();
    TestingClusterClient<String> clusterClient = new TestingClusterClient<>();
    clusterClient.setStopWithSavepointFunction((jobID, advanceToEndOfEventTime, savepointDirectory, formatType) -> {
        assertThat(jobID, is(jid));
        assertThat(advanceToEndOfEventTime, is(true));
        assertNull(savepointDirectory);
        stopWithSavepointLatch.trigger();
        return CompletableFuture.completedFuture(savepointDirectory);
    });
    MockedCliFrontend testFrontend = new MockedCliFrontend(clusterClient);
    testFrontend.stop(parameters);
    stopWithSavepointLatch.await();
}

Example 10 with TestingClusterClient

use of org.apache.flink.client.program.TestingClusterClient in project flink by apache.

the class CliFrontendCancelTest method testCancel.

@Test
public void testCancel() throws Exception {
    // test cancel properly
    JobID jid = new JobID();
    OneShotLatch cancelLatch = new OneShotLatch();
    String[] parameters = { jid.toString() };
    TestingClusterClient<String> clusterClient = new TestingClusterClient<>();
    clusterClient.setCancelFunction(jobID -> {
        cancelLatch.trigger();
        return CompletableFuture.completedFuture(Acknowledge.get());
    });
    MockedCliFrontend testFrontend = new MockedCliFrontend(clusterClient);
    testFrontend.cancel(parameters);
    cancelLatch.await();
}