Search in sources :

Example 1 with AsynchronousOperationResult

use of org.apache.flink.runtime.rest.handler.async.AsynchronousOperationResult in project flink by apache.

the class SavepointHandlersTest method testQueryStatusOfUnknownOperationReturnsError.

@Test
public void testQueryStatusOfUnknownOperationReturnsError() throws HandlerRequestException, RestHandlerException {
    final TestingRestfulGateway testingRestfulGateway = new TestingRestfulGateway.Builder().setGetSavepointStatusFunction(key -> FutureUtils.completedExceptionally(new UnknownOperationKeyException(key))).build();
    final CompletableFuture<AsynchronousOperationResult<SavepointInfo>> statusFuture = savepointStatusHandler.handleRequest(savepointStatusRequest(new TriggerId()), testingRestfulGateway);
    assertThat(statusFuture, RestMatchers.respondsWithError(HttpResponseStatus.NOT_FOUND));
}
Also used : AsynchronousOperationResult(org.apache.flink.runtime.rest.handler.async.AsynchronousOperationResult) TriggerId(org.apache.flink.runtime.rest.messages.TriggerId) SavepointFormatType(org.apache.flink.core.execution.SavepointFormatType) SavepointTestUtilities.setReferenceToOperationKey(org.apache.flink.runtime.rest.handler.job.savepoints.SavepointTestUtilities.setReferenceToOperationKey) QueueStatus(org.apache.flink.runtime.rest.messages.queue.QueueStatus) HashMap(java.util.HashMap) CompletableFuture(java.util.concurrent.CompletableFuture) TestingRestfulGateway(org.apache.flink.runtime.webmonitor.TestingRestfulGateway) AtomicReference(java.util.concurrent.atomic.AtomicReference) AsynchronousJobOperationKey(org.apache.flink.runtime.rest.handler.job.AsynchronousJobOperationKey) HttpResponseStatus(org.apache.flink.shaded.netty4.io.netty.handler.codec.http.HttpResponseStatus) EmptyRequestBody(org.apache.flink.runtime.rest.messages.EmptyRequestBody) TriggerIdPathParameter(org.apache.flink.runtime.rest.messages.TriggerIdPathParameter) Assert.assertThat(org.junit.Assert.assertThat) SavepointTestUtilities.getResultIfKeyMatches(org.apache.flink.runtime.rest.handler.job.savepoints.SavepointTestUtilities.getResultIfKeyMatches) FutureUtils(org.apache.flink.util.concurrent.FutureUtils) Map(java.util.Map) SavepointTriggerRequestBody(org.apache.flink.runtime.rest.messages.job.savepoints.SavepointTriggerRequestBody) TestLogger(org.apache.flink.util.TestLogger) HandlerRequest(org.apache.flink.runtime.rest.handler.HandlerRequest) Assert.fail(org.junit.Assert.fail) SavepointStatusMessageParameters(org.apache.flink.runtime.rest.messages.job.savepoints.SavepointStatusMessageParameters) UnknownOperationKeyException(org.apache.flink.runtime.dispatcher.UnknownOperationKeyException) HandlerRequestException(org.apache.flink.runtime.rest.handler.HandlerRequestException) RestMatchers(org.apache.flink.runtime.rest.RestMatchers) JobIDPathParameter(org.apache.flink.runtime.rest.messages.JobIDPathParameter) Nullable(javax.annotation.Nullable) Before(org.junit.Before) Matchers.notNullValue(org.hamcrest.Matchers.notNullValue) OperationResult(org.apache.flink.runtime.rest.handler.async.OperationResult) GatewayRetriever(org.apache.flink.runtime.webmonitor.retriever.GatewayRetriever) SavepointInfo(org.apache.flink.runtime.rest.messages.job.savepoints.SavepointInfo) RestfulGateway(org.apache.flink.runtime.webmonitor.RestfulGateway) Test(org.junit.Test) Acknowledge(org.apache.flink.runtime.messages.Acknowledge) RestHandlerException(org.apache.flink.runtime.rest.handler.RestHandlerException) Matchers.instanceOf(org.hamcrest.Matchers.instanceOf) SavepointTriggerMessageParameters(org.apache.flink.runtime.rest.messages.job.savepoints.SavepointTriggerMessageParameters) JobID(org.apache.flink.api.common.JobID) Matchers.equalTo(org.hamcrest.Matchers.equalTo) Collections(java.util.Collections) Time(org.apache.flink.api.common.time.Time) Assert.assertEquals(org.junit.Assert.assertEquals) UnknownOperationKeyException(org.apache.flink.runtime.dispatcher.UnknownOperationKeyException) AsynchronousOperationResult(org.apache.flink.runtime.rest.handler.async.AsynchronousOperationResult) TriggerId(org.apache.flink.runtime.rest.messages.TriggerId) TestingRestfulGateway(org.apache.flink.runtime.webmonitor.TestingRestfulGateway) Test(org.junit.Test)

Aggregations

Collections (java.util.Collections)1 HashMap (java.util.HashMap)1 Map (java.util.Map)1 CompletableFuture (java.util.concurrent.CompletableFuture)1 AtomicReference (java.util.concurrent.atomic.AtomicReference)1 Nullable (javax.annotation.Nullable)1 JobID (org.apache.flink.api.common.JobID)1 Time (org.apache.flink.api.common.time.Time)1 SavepointFormatType (org.apache.flink.core.execution.SavepointFormatType)1 UnknownOperationKeyException (org.apache.flink.runtime.dispatcher.UnknownOperationKeyException)1 Acknowledge (org.apache.flink.runtime.messages.Acknowledge)1 RestMatchers (org.apache.flink.runtime.rest.RestMatchers)1 HandlerRequest (org.apache.flink.runtime.rest.handler.HandlerRequest)1 HandlerRequestException (org.apache.flink.runtime.rest.handler.HandlerRequestException)1 RestHandlerException (org.apache.flink.runtime.rest.handler.RestHandlerException)1 AsynchronousOperationResult (org.apache.flink.runtime.rest.handler.async.AsynchronousOperationResult)1 OperationResult (org.apache.flink.runtime.rest.handler.async.OperationResult)1 AsynchronousJobOperationKey (org.apache.flink.runtime.rest.handler.job.AsynchronousJobOperationKey)1 SavepointTestUtilities.getResultIfKeyMatches (org.apache.flink.runtime.rest.handler.job.savepoints.SavepointTestUtilities.getResultIfKeyMatches)1 SavepointTestUtilities.setReferenceToOperationKey (org.apache.flink.runtime.rest.handler.job.savepoints.SavepointTestUtilities.setReferenceToOperationKey)1