org.junit.jupiter.params.provider.ArgumentsSource Java Exaples

Source File: FileServiceTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(BaseUriProvider.class)
void testFileSystemGet_modifiedFile(String baseUri) throws Exception {
    final Path barFile = tmpDir.resolve("modifiedFile.html");
    final String expectedContentA = "<html/>";
    final String expectedContentB = "<html><body/></html>";
    writeFile(barFile, expectedContentA);
    final long barFileLastModified = Files.getLastModifiedTime(barFile).toMillis();

    try (CloseableHttpClient hc = HttpClients.createMinimal()) {
        final HttpUriRequest req = new HttpGet(baseUri + "/fs/modifiedFile.html");
        try (CloseableHttpResponse res = hc.execute(req)) {
            assert200Ok(res, "text/html", expectedContentA);
        }

        // Modify the file cached by the service. Update last modification time explicitly
        // so that it differs from the old value.
        writeFile(barFile, expectedContentB);
        Files.setLastModifiedTime(barFile, FileTime.fromMillis(barFileLastModified + 5000));

        try (CloseableHttpResponse res = hc.execute(req)) {
            assert200Ok(res, "text/html", expectedContentB);
        }
    }
}

Source File: ThriftServiceTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(SerializationFormatProvider.class)
void testIdentity_DevNullService_consume(SerializationFormat defaultSerializationFormat) throws Exception {
    final DevNullService.Client client = new DevNullService.Client.Factory().getClient(
            inProto(defaultSerializationFormat), outProto(defaultSerializationFormat));
    client.send_consume(FOO);
    assertThat(out.length()).isGreaterThan(0);

    final THttpService syncService = THttpService.of((DevNullService.Iface) value -> {
        // NOOP
    }, defaultSerializationFormat);

    final THttpService asyncService = THttpService.of(
            (DevNullService.AsyncIface) (value, resultHandler) ->
                    resultHandler.onComplete(null), defaultSerializationFormat);

    invokeTwice(syncService, asyncService);

    assertThat(promise.get()).isEqualTo(promise2.get());
}

Source File: DeferredStreamMessageTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(AbortCauseArgumentProvider.class)
void testLateAbortWithSubscriber(@Nullable Throwable cause) {
    final DeferredStreamMessage<Object> m = new DeferredStreamMessage<>();
    final DefaultStreamMessage<Object> d = new DefaultStreamMessage<>();
    @SuppressWarnings("unchecked")
    final Subscriber<Object> subscriber = mock(Subscriber.class);

    m.subscribe(subscriber, ImmediateEventExecutor.INSTANCE);
    m.delegate(d);
    verify(subscriber).onSubscribe(any());

    if (cause == null) {
        m.abort();
    } else {
        m.abort(cause);
    }
    if (cause == null) {
        verify(subscriber, times(1)).onError(isA(AbortedStreamException.class));
    } else {
        verify(subscriber, times(1)).onError(isA(cause.getClass()));
    }

    assertAborted(m, cause);
    assertAborted(d, cause);
}

Source File: DeferredStreamMessageTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(AbortCauseArgumentProvider.class)
void testEarlyAbortWithSubscriber(@Nullable Throwable cause) {
    final DeferredStreamMessage<Object> m = new DeferredStreamMessage<>();
    @SuppressWarnings("unchecked")
    final Subscriber<Object> subscriber = mock(Subscriber.class);
    m.subscribe(subscriber, ImmediateEventExecutor.INSTANCE);
    if (cause == null) {
        m.abort();
    } else {
        m.abort(cause);
    }
    assertAborted(m, cause);

    final DefaultStreamMessage<Object> d = new DefaultStreamMessage<>();
    m.delegate(d);
    assertAborted(d, cause);
}

Source File: ThriftServiceTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(SerializationFormatProvider.class)
void testAsync_FileService_create_reply(SerializationFormat defaultSerializationFormat) throws Exception {
    final FileService.Client client = new FileService.Client.Factory().getClient(
            inProto(defaultSerializationFormat), outProto(defaultSerializationFormat));
    client.send_create(BAR);
    assertThat(out.length()).isGreaterThan(0);

    final THttpService service = THttpService.of(
            (FileService.AsyncIface) (path, resultHandler) ->
                    resultHandler.onError(newFileServiceException()), defaultSerializationFormat);

    invoke(service);

    try {
        client.recv_create();
        fail(FileServiceException.class.getSimpleName() + " not raised.");
    } catch (FileServiceException ignored) {
        // Expected
    }
}

Source File: ThriftServiceTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(SerializationFormatProvider.class)
void testAsync_NameService_removeMiddle(SerializationFormat defaultSerializationFormat) throws Exception {
    final NameService.Client client = new NameService.Client.Factory().getClient(
            inProto(defaultSerializationFormat), outProto(defaultSerializationFormat));
    client.send_removeMiddle(new Name(BAZ, BAR, FOO));
    assertThat(out.length()).isGreaterThan(0);

    final THttpService service = THttpService.of(
            (NameService.AsyncIface) (name, resultHandler) ->
                    resultHandler.onComplete(new Name(name.first, null, name.last)),
            defaultSerializationFormat);

    invoke(service);

    assertThat(client.recv_removeMiddle()).isEqualTo(new Name(BAZ, null, FOO));
}

Source File: HttpClientResponseTimeoutTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(TimeoutDecoratorSource.class)
void setRequestTimeoutAtPendingTimeoutTask(Consumer<? super ClientRequestContext> timeoutCustomizer) {
    final WebClient client = WebClient
            .builder(server.httpUri())
            .option(ClientOption.RESPONSE_TIMEOUT_MILLIS.newValue(30L))
            .decorator((delegate, ctx, req) -> {
                // set timeout before initializing timeout controller
                timeoutCustomizer.accept(ctx);
                return delegate.execute(ctx, req);
            })
            .build();
    await().timeout(Duration.ofSeconds(5)).untilAsserted(() -> {
        assertThatThrownBy(() -> client.get("/no-timeout")
                                       .aggregate().join())
                .isInstanceOf(CompletionException.class)
                .hasCauseInstanceOf(ResponseTimeoutException.class);
    });
}

Source File: DeferredStreamMessageTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(AbortCauseArgumentProvider.class)
void testLateAbort(@Nullable Throwable cause) {
    final DeferredStreamMessage<Object> m = new DeferredStreamMessage<>();
    final DefaultStreamMessage<Object> d = new DefaultStreamMessage<>();

    m.delegate(d);
    if (cause == null) {
        m.abort();
    } else {
        m.abort(cause);
    }

    assertAborted(m, cause);
    assertAborted(d, cause);
}

Source File: InvestorTest.java From robozonky with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(SessionType.class)
void irrelevantFail(final SessionInfo sessionType) {
    final boolean isDryRun = sessionType.isDryRun();
    final Tenant t = mockTenant(zonky, isDryRun);
    doThrow(IllegalStateException.class).when(zonky)
        .invest(notNull(), anyInt());
    final Investor i = Investor.build(t);
    final RecommendedLoan r = DESCRIPTOR.recommend(Money.from(200))
        .orElse(null);
    if (isDryRun) { // the endpoint is not actually called, therefore cannot return error
        assertThat(i.invest(r)
            .get()).isEqualTo(Money.from(200));
    } else {
        assertThatThrownBy(() -> i.invest(r)).isInstanceOf(IllegalStateException.class);
    }
}

Source File: RequestCompressionPluginTest.java From riptide with MIT License

6 votes

@ParameterizedTest
@ArgumentsSource(RequestFactorySource.class)
void shouldCompressRequestBody(final ClientHttpRequestFactory factory) {
    driver.addExpectation(onRequestTo("/")
                    .withMethod(POST)
                    .withHeader("X-Content-Encoding", "gzip") // written by Jetty's GzipHandler
                    .withBody(equalTo("{}"), "application/json"),
            giveResponse("", "text/plain"));

    final Http http = buildHttp(factory, new RequestCompressionPlugin());
    http.post("/")
            .contentType(MediaType.APPLICATION_JSON)
            .body(new HashMap<>())
            .call(pass())
            .join();
}

Source File: RequestCompressionPluginTest.java From riptide with MIT License

6 votes

@ParameterizedTest
@ArgumentsSource(RequestFactorySource.class)
void shouldCompressWithGivenAlgorithm(final ClientHttpRequestFactory factory) {
    driver.addExpectation(onRequestTo("/")
                    .withMethod(POST)
                    .withHeader("Content-Encoding", "identity") // not handled by Jetty
                    .withoutHeader("X-Content-Encoding")
                    .withBody(equalTo("{}"), "application/json"),
            giveResponse("", "text/plain"));

    final Http http = buildHttp(factory, new RequestCompressionPlugin(Compression.of("identity", it -> it)));
    http.post("/")
            .contentType(MediaType.APPLICATION_JSON)
            .body(new HashMap<>())
            .call(pass())
            .join();
}

Source File: CircuitBreakerRuleWithContentBuilderTest.java From armeria with Apache License 2.0

6 votes

@ArgumentsSource(CircuitBreakerRuleSource.class)
@ParameterizedTest
void shouldReportAsWithContent(String message, CircuitBreakerDecision decision) {
    final CircuitBreakerRuleWithContent<HttpResponse> rule =
            CircuitBreakerRuleWithContent
                    .<HttpResponse>builder()
                    .onResponse((unused, response) -> response.aggregate().thenApply(content -> false))
                    .onResponse((unused, response) -> response.aggregate().thenApply(content -> false))
                    .onResponse((unused, response) -> {
                        return response.aggregate()
                                       .thenApply(content -> content.contentUtf8().contains(message));
                    })
                    .thenSuccess()
                    .orElse(CircuitBreakerRuleWithContent
                                    .<HttpResponse>builder()
                                    .thenFailure());

    ctx1.logBuilder().responseHeaders(ResponseHeaders.of(HttpStatus.OK));
    assertFuture(rule.shouldReportAsSuccess(ctx1, HttpResponse.of("Hello Armeria!"), null))
            .isSameAs(decision);
}

Source File: Java9RequestContextAwareFutureTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(Java9CallbackArgumentsProvider.class)
void makeContextAwareCompletableFutureWithDifferentContext(
        BiConsumer<CompletableFuture<?>, AtomicBoolean> callback) {
    final HttpRequest req = HttpRequest.of(HttpMethod.GET, "/");
    final ServiceRequestContext ctx1 = ServiceRequestContext.builder(req).build();
    final ServiceRequestContext ctx2 = ServiceRequestContext.builder(req).build();
    try (SafeCloseable ignored = ctx1.push()) {
        final CompletableFuture<Object> future = new CompletableFuture<>();
        final CompletableFuture<Object> contextAwareFuture = ctx2.makeContextAware(future);
        final AtomicBoolean callbackCalled = new AtomicBoolean();
        callback.accept(contextAwareFuture, callbackCalled);

        future.complete(null);

        assertThat(callbackCalled.get()).isFalse();
        verify(appender, atLeast(0)).doAppend(eventCaptor.capture());
        assertThat(eventCaptor.getAllValues()).anySatisfy(event -> {
            assertThat(event.getLevel()).isEqualTo(Level.WARN);
            assertThat(event.getMessage()).startsWith("An error occurred while pushing");
        });
    }
}

Source File: GrpcServiceServerTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(BlockingClientProvider.class)
void error_withMessage(UnitTestServiceBlockingStub blockingClient) throws Exception {
    final StatusRuntimeException t = (StatusRuntimeException) catchThrowable(
            () -> blockingClient.errorWithMessage(REQUEST_MESSAGE));
    assertThat(t.getStatus().getCode()).isEqualTo(Code.ABORTED);
    assertThat(t.getStatus().getDescription()).isEqualTo("aborted call");
    assertThat(t.getTrailers().getAll(STRING_VALUE_KEY))
            .containsExactly(StringValue.newBuilder().setValue("custom metadata").build());
    assertThat(t.getTrailers().getAll(INT_32_VALUE_KEY))
            .containsExactly(Int32Value.newBuilder().setValue(10).build(),
                             Int32Value.newBuilder().setValue(20).build());
    assertThat(t.getTrailers().get(CUSTOM_VALUE_KEY)).isEqualTo("custom value");

    checkRequestLog((rpcReq, rpcRes, grpcStatus) -> {
        assertThat(rpcReq.method()).isEqualTo("armeria.grpc.testing.UnitTestService/ErrorWithMessage");
        assertThat(rpcReq.params()).containsExactly(REQUEST_MESSAGE);
        assertThat(grpcStatus).isNotNull();
        assertThat(grpcStatus.getCode()).isEqualTo(Code.ABORTED);
        assertThat(grpcStatus.getDescription()).isEqualTo("aborted call");
    });
}

Source File: GrpcServiceServerTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(StreamingClientProvider.class)
void requestContextSet(UnitTestServiceStub streamingClient) throws Exception {
    final StreamRecorder<SimpleResponse> response = StreamRecorder.create();
    final StreamObserver<SimpleRequest> request = streamingClient.checkRequestContext(response);
    request.onNext(REQUEST_MESSAGE);
    request.onNext(REQUEST_MESSAGE);
    request.onNext(REQUEST_MESSAGE);
    request.onCompleted();
    response.awaitCompletion();
    final SimpleResponse expectedResponse =
            SimpleResponse.newBuilder()
                          .setPayload(Payload.newBuilder()
                                             .setBody(ByteString.copyFromUtf8("3")))
                          .build();
    assertThat(response.getValues()).containsExactly(expectedResponse);

    checkRequestLog((rpcReq, rpcRes, grpcStatus) -> {
        assertThat(rpcReq.method()).isEqualTo(
                "armeria.grpc.testing.UnitTestService/CheckRequestContext");
        assertThat(rpcReq.params()).containsExactly(REQUEST_MESSAGE);
        assertThat(rpcRes.get()).isEqualTo(expectedResponse);
    });
}

Source File: GrpcServiceServerTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(BlockingClientProvider.class)
void error_noMessage(UnitTestServiceBlockingStub blockingClient) throws Exception {
    final StatusRuntimeException t = (StatusRuntimeException) catchThrowable(
            () -> blockingClient.errorNoMessage(REQUEST_MESSAGE));
    assertThat(t.getStatus().getCode()).isEqualTo(Code.ABORTED);
    assertThat(t.getStatus().getDescription()).isNull();

    checkRequestLog((rpcReq, rpcRes, grpcStatus) -> {
        assertThat(rpcReq.method()).isEqualTo("armeria.grpc.testing.UnitTestService/ErrorNoMessage");
        assertThat(rpcReq.params()).containsExactly(REQUEST_MESSAGE);
        assertThat(grpcStatus).isNotNull();
        assertThat(grpcStatus.getCode()).isEqualTo(Code.ABORTED);
        assertThat(grpcStatus.getDescription()).isNull();
    });
}

Source File: StreamMessageTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(StreamProvider.class)
void flowControlled_writeThenDemandThenProcess_eventLoop(StreamMessage<Integer> stream,
                                                         List<Integer> values) {
    writeTenIntegers(stream);
    eventLoop.get().submit(
            () ->
                    stream.subscribe(new ResultCollectingSubscriber() {
                        private Subscription subscription;

                        @Override
                        public void onSubscribe(Subscription s) {
                            subscription = s;
                            subscription.request(1);
                        }

                        @Override
                        public void onNext(Integer value) {
                            subscription.request(1);
                            super.onNext(value);
                        }
                    }, eventLoop.get())).syncUninterruptibly();
    assertSuccess(values);
}

Source File: ThriftServiceTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(SerializationFormatProvider.class)
void testIdentity_FileService_create_exception(SerializationFormat defaultSerializationFormat)
        throws Exception {
    final FileService.Client client = new FileService.Client.Factory().getClient(
            inProto(defaultSerializationFormat), outProto(defaultSerializationFormat));
    client.send_create(BAZ);
    assertThat(out.length()).isGreaterThan(0);

    final RuntimeException exception = Exceptions.clearTrace(new RuntimeException());
    final THttpService syncService = THttpService.of((FileService.Iface) path -> {
        throw exception;
    }, defaultSerializationFormat);

    final THttpService asyncService = THttpService.of(
            (FileService.AsyncIface) (path, resultHandler) ->
                    resultHandler.onError(exception), defaultSerializationFormat);

    invokeTwice(syncService, asyncService);

    assertThat(promise.get()).isEqualTo(promise2.get());
}

Source File: ThriftOverHttpClientTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(ParametersProvider.class)
void contextCaptorSync(
        ClientOptions clientOptions, SerializationFormat format, SessionProtocol protocol)
        throws Exception {
    final HelloService.Iface client = Clients.builder(uri(Handlers.HELLO, format, protocol))
                                             .options(clientOptions)
                                             .build(Handlers.HELLO.iface());
    try (ClientRequestContextCaptor ctxCaptor = Clients.newContextCaptor()) {
        client.hello("kukuman");
        final ClientRequestContext ctx = ctxCaptor.get();
        final RpcRequest rpcReq = ctx.rpcRequest();
        assertThat(rpcReq).isNotNull();
        assertThat(rpcReq.method()).isEqualTo("hello");
        assertThat(rpcReq.params()).containsExactly("kukuman");
    }
}

Source File: FileServiceTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(BaseUriProvider.class)
void testGetWithoutPreCompression(String baseUri) throws Exception {
    try (CloseableHttpClient hc = HttpClients.createMinimal()) {
        final HttpGet request = new HttpGet(baseUri + "/compressed/foo_alone.txt");
        request.setHeader("Accept-Encoding", "gzip");
        try (CloseableHttpResponse res = hc.execute(request)) {
            assertThat(res.getFirstHeader("Content-Encoding")).isNull();
            assertThat(headerOrNull(res, "Content-Type")).isEqualTo(
                    "text/plain; charset=utf-8");
            final byte[] content = content(res);
            assertThat(new String(content, StandardCharsets.UTF_8)).isEqualTo("foo_alone");

            // Confirm path not cached when cache disabled.
            assertThat(PathAndQuery.cachedPaths())
                    .doesNotContain("/compressed/foo_alone.txt");
        }
    }
}

Source File: FileServiceTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(BaseUriProvider.class)
void testClassPathGetFromJar(String baseUri) throws Exception {
    try (CloseableHttpClient hc = HttpClients.createMinimal()) {
        // Read a class from a third-party library JAR.
        try (CloseableHttpResponse res =
                     hc.execute(new HttpGet(baseUri + "/classes/io/netty/util/NetUtil.class"))) {
            assert200Ok(res, null, content -> assertThat(content).isNotEmpty());
        }
        // Read a class from a third-party library JAR.
        try (CloseableHttpResponse res =
                     hc.execute(new HttpGet(baseUri + "/by-entry/classes/io/netty/util/NetUtil.class"))) {
            assert200Ok(res, null, content -> assertThat(content).isNotEmpty());
        }
    }
}

Source File: FileServiceTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(BaseUriProvider.class)
void testFileSystemGet_newFile(String baseUri) throws Exception {
    final String barFileName = baseUri.substring(baseUri.lastIndexOf('/') + 1) + "_newFile.html";
    assertThat(barFileName).isIn("cached_newFile.html", "uncached_newFile.html");

    final Path barFile = tmpDir.resolve(barFileName);
    final String expectedContentA = "<html/>";

    try (CloseableHttpClient hc = HttpClients.createMinimal()) {
        final HttpUriRequest req = new HttpGet(baseUri + "/fs/" + barFileName);
        try (CloseableHttpResponse res = hc.execute(req)) {
            assert404NotFound(res);
        }
        writeFile(barFile, expectedContentA);
        try (CloseableHttpResponse res = hc.execute(req)) {
            assert200Ok(res, "text/html", expectedContentA);
        }
    }
}

Source File: FileServiceTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(BaseUriProvider.class)
void testClassPathGetFromModule(String baseUri) throws Exception {
    try (CloseableHttpClient hc = HttpClients.createMinimal()) {
        // Read a class from a JDK module (java.base).
        try (CloseableHttpResponse res =
                     hc.execute(new HttpGet(baseUri + "/classes/java/lang/Object.class"))) {
            assert200Ok(res, null, content -> assertThat(content).isNotEmpty());
        }
        // Read a class from a JDK module (java.base).
        try (CloseableHttpResponse res =
                     hc.execute(new HttpGet(baseUri + "/by-entry/classes/java/lang/Object.class"))) {
            assert200Ok(res, null, content -> assertThat(content).isNotEmpty());
        }
    }
}

Source File: StreamMessageTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(StreamProvider.class)
void flowControlled_writeThenProcessThenDemand(StreamMessage<Integer> stream, List<Integer> values) {
    writeTenIntegers(stream);
    stream.subscribe(new ResultCollectingSubscriber() {
        private Subscription subscription;

        @Override
        public void onSubscribe(Subscription s) {
            subscription = s;
            subscription.request(1);
        }

        @Override
        public void onNext(Integer value) {
            super.onNext(value);
            subscription.request(1);
        }
    });
    assertSuccess(values);
}

Source File: ThriftServiceTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(SerializationFormatProvider.class)
void testAsync_HelloService_hello(SerializationFormat defaultSerializationFormat) throws Exception {
    final HelloService.Client client = new HelloService.Client.Factory().getClient(
            inProto(defaultSerializationFormat), outProto(defaultSerializationFormat));
    client.send_hello(FOO);
    assertThat(out.length()).isGreaterThan(0);

    final THttpService service = THttpService.of(
            (HelloService.AsyncIface) (name, resultHandler) ->
                    resultHandler.onComplete("Hello, " + name + '!'), defaultSerializationFormat);

    invoke(service);

    assertThat(client.recv_hello()).isEqualTo("Hello, foo!");
}

Source File: ThriftServiceTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(SerializationFormatProvider.class)
void testSync_FileService_create_exception(SerializationFormat defaultSerializationFormat)
        throws Exception {
    final FileService.Client client = new FileService.Client.Factory().getClient(
            inProto(defaultSerializationFormat), outProto(defaultSerializationFormat));
    client.send_create(BAZ);
    assertThat(out.length()).isGreaterThan(0);

    final RuntimeException exception = Exceptions.clearTrace(new RuntimeException());
    final THttpService service = THttpService.of((FileService.Iface) path -> {
        throw exception;
    }, defaultSerializationFormat);

    invoke(service);

    try {
        client.recv_create();
        fail(TApplicationException.class.getSimpleName() + " not raised.");
    } catch (TApplicationException e) {
        assertThat(e.getType()).isEqualTo(TApplicationException.INTERNAL_ERROR);
        assertThat(e.getMessage()).contains(exception.toString());
    }
}

Source File: ThriftServiceTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(SerializationFormatProvider.class)
void testAsync_HelloService_hello_with_null(SerializationFormat defaultSerializationFormat)
        throws Exception {
    final HelloService.Client client = new HelloService.Client.Factory().getClient(
            inProto(defaultSerializationFormat), outProto(defaultSerializationFormat));
    client.send_hello(null);
    assertThat(out.length()).isGreaterThan(0);

    final THttpService service = THttpService.of(
            (HelloService.AsyncIface) (name, resultHandler) ->
                    resultHandler.onComplete(String.valueOf(name != null)), defaultSerializationFormat);

    invoke(service);

    assertThat(client.recv_hello()).isEqualTo("false");
}

Source File: ThriftServiceTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(SerializationFormatProvider.class)
void testIdentity_HelloService_hello(SerializationFormat defaultSerializationFormat) throws Exception {
    final HelloService.Client client = new HelloService.Client.Factory().getClient(
            inProto(defaultSerializationFormat), outProto(defaultSerializationFormat));
    client.send_hello(FOO);
    assertThat(out.length()).isGreaterThan(0);

    final THttpService syncService = THttpService.of(
            (HelloService.Iface) name -> "Hello, " + name + '!', defaultSerializationFormat);

    final THttpService asyncService = THttpService.of(
            (HelloService.AsyncIface) (name, resultHandler) ->
                    resultHandler.onComplete("Hello, " + name + '!'), defaultSerializationFormat);

    invokeTwice(syncService, asyncService);

    assertThat(promise.get()).isEqualTo(promise2.get());
}

Source File: ThriftServiceTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(SerializationFormatProvider.class)
void testSync_OnewayHelloService_hello(SerializationFormat defaultSerializationFormat) throws Exception {
    final AtomicReference<String> actualName = new AtomicReference<>();

    final OnewayHelloService.Client client =
            new OnewayHelloService.Client.Factory().getClient(
                    inProto(defaultSerializationFormat), outProto(defaultSerializationFormat));
    client.send_hello(FOO);
    assertThat(out.length()).isGreaterThan(0);

    final THttpService service = THttpService.of(
            (OnewayHelloService.Iface) actualName::set, defaultSerializationFormat);

    invoke(service);

    assertThat(promise.get().isEmpty()).isTrue();
    assertThat(actualName.get()).isEqualTo(FOO);
}

Source File: ThriftServiceTest.java From armeria with Apache License 2.0

6 votes

@ParameterizedTest
@ArgumentsSource(SerializationFormatProvider.class)
void testAsync_OnewayHelloService_hello(SerializationFormat defaultSerializationFormat) throws Exception {
    final AtomicReference<String> actualName = new AtomicReference<>();

    final OnewayHelloService.Client client =
            new OnewayHelloService.Client.Factory().getClient(
                    inProto(defaultSerializationFormat), outProto(defaultSerializationFormat));
    client.send_hello(FOO);
    assertThat(out.length()).isGreaterThan(0);

    final THttpService service = THttpService.of((OnewayHelloService.AsyncIface) (name, resultHandler) -> {
        actualName.set(name);
        resultHandler.onComplete(null);
    }, defaultSerializationFormat);

    invoke(service);

    assertThat(promise.get().isEmpty()).isTrue();
    assertThat(actualName.get()).isEqualTo(FOO);
}

org.junit.jupiter.params.provider.ArgumentsSource Java Examples