Merge pull request #43908 from geoand/#43899

Add a test using `QuarkusRestClientBuilder` and `RestClientBuilderListener`

extensions/resteasy-reactive/rest-client/deployment/src/test/java/io/quarkus/rest/client/reactive/headers/ReactiveClientHeadersFromBuilderListenerTest.java

@@ -0,0 +1,145 @@
+package io.quarkus.rest.client.reactive.headers;
+
+import static io.restassured.RestAssured.given;
+import static java.lang.String.format;
+import static org.assertj.core.api.Assertions.assertThat;
+
+import java.net.URI;
+import java.util.List;
+import java.util.Map;
+import java.util.function.Supplier;
+
+import jakarta.enterprise.context.ApplicationScoped;
+import jakarta.ws.rs.GET;
+import jakarta.ws.rs.HeaderParam;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.Context;
+import jakarta.ws.rs.core.HttpHeaders;
+import jakarta.ws.rs.core.MultivaluedHashMap;
+import jakarta.ws.rs.core.MultivaluedMap;
+
+import org.eclipse.microprofile.rest.client.RestClientBuilder;
+import org.eclipse.microprofile.rest.client.spi.RestClientBuilderListener;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.RegisterExtension;
+
+import io.quarkus.arc.Arc;
+import io.quarkus.arc.Unremovable;
+import io.quarkus.rest.client.reactive.QuarkusRestClientBuilder;
+import io.quarkus.rest.client.reactive.ReactiveClientHeadersFactory;
+import io.quarkus.rest.client.reactive.TestJacksonBasicMessageBodyReader;
+import io.quarkus.rest.client.reactive.runtime.context.ClientHeadersFactoryContextResolver;
+import io.quarkus.test.QuarkusUnitTest;
+import io.quarkus.test.common.http.TestHTTPResource;
+import io.smallrye.mutiny.Uni;
+
+public class ReactiveClientHeadersFromBuilderListenerTest {
+
+ private static final String HEADER_NAME = "my-header";
+ private static final String HEADER_VALUE = "oifajrofijaeoir5gjaoasfaxcvcz";
+ public static final String COPIED_INCOMING_HEADER = "copied-incoming-header";
+ public static final String INCOMING_HEADER = "incoming-header";
+ public static final String DIRECT_HEADER_PARAM = "direct-header-param";
+ public static final String DIRECT_HEADER_PARAM_VAL = "direct-header-param-val";
+
+ @TestHTTPResource
+ URI baseUri;
+
+ @RegisterExtension
+ static final QuarkusUnitTest config = new QuarkusUnitTest()
+ .withApplicationRoot((jar) -> jar
+ .addClasses(ReactiveClientHeadersFromBuilderTest.Client.class, TestJacksonBasicMessageBodyReader.class)
+ .addAsServiceProvider(RestClientBuilderListener.class, CustomRestClientBuilderListener.class))
+ .overrideRuntimeConfigKey("my.property-value", HEADER_VALUE);
+
+ @Test
+ void shouldPropagateHeaders() {
+ // we're calling a resource that sets "incoming-header" header
+ // this header should be dropped by the client and its value should be put into copied-incoming-header
+ String propagatedHeaderValue = "propag8ed header";
+ // @formatter:off
+ var response =
+ given()
+ .header(INCOMING_HEADER, propagatedHeaderValue)
+ .body(baseUri.toString())
+ .when()
+ .post("/call-client")
+ .thenReturn();
+ // @formatter:on
+ assertThat(response.statusCode()).isEqualTo(200);
+ assertThat(response.jsonPath().getString(INCOMING_HEADER)).isNull();
+ assertThat(response.jsonPath().getString(COPIED_INCOMING_HEADER)).isEqualTo(format("[%s]", propagatedHeaderValue));
+ assertThat(response.jsonPath().getString(HEADER_NAME)).isEqualTo(format("[%s]", HEADER_VALUE));
+ assertThat(response.jsonPath().getString(DIRECT_HEADER_PARAM)).isEqualTo(format("[%s]", DIRECT_HEADER_PARAM_VAL));
+ }
+
+ @Path("/")
+ @ApplicationScoped
+ public static class Resource {
+
+ @GET
+ @Produces("application/json")
+ public Map<String, List<String>> returnHeaderValues(@Context HttpHeaders headers) {
+ return headers.getRequestHeaders();
+ }
+
+ @Path("/call-client")
+ @POST
+ public Map<String, List<String>> callClient(String uri) {
+ ReactiveClientHeadersFromBuilderTest.Client client = QuarkusRestClientBuilder.newBuilder()
+ .baseUri(URI.create(uri))
+ .register(new TestJacksonBasicMessageBodyReader())
+ .build(ReactiveClientHeadersFromBuilderTest.Client.class);
+ return client.getWithHeader(DIRECT_HEADER_PARAM_VAL);
+ }
+ }
+
+ @ApplicationScoped
+ public static class Service {
+
+ public String getValue() {
+ return HEADER_VALUE;
+ }
+ }
+
+ public interface Client {
+ @GET
+ Map<String, List<String>> getWithHeader(@HeaderParam(DIRECT_HEADER_PARAM) String directHeaderParam);
+ }
+
+ @Unremovable
+ @ApplicationScoped
+ public static class CustomReactiveClientHeadersFactory extends ReactiveClientHeadersFactory {
+
+ private final Service service;
+
+ public CustomReactiveClientHeadersFactory(Service service) {
+ this.service = service;
+ }
+
+ @Override
+ public Uni<MultivaluedMap<String, String>> getHeaders(MultivaluedMap<String, String> incomingHeaders,
+ MultivaluedMap<String, String> clientOutgoingHeaders) {
+ return Uni.createFrom().item(new Supplier<>() {
+ @Override
+ public MultivaluedHashMap<String, String> get() {
+ MultivaluedHashMap<String, String> newHeaders = new MultivaluedHashMap<>();
+ newHeaders.add(HEADER_NAME, service.getValue());
+ newHeaders.add(COPIED_INCOMING_HEADER, incomingHeaders.getFirst(INCOMING_HEADER));
+ return newHeaders;
+ }
+ });
+ }
+ }
+
+ public static class CustomRestClientBuilderListener implements RestClientBuilderListener {
+
+ @Override
+ public void onNewBuilder(RestClientBuilder builder) {
+ builder.register(new ClientHeadersFactoryContextResolver(
+ Arc.container().instance(CustomReactiveClientHeadersFactory.class).get()));
+ }
+ }
+}