Merge pull request #521 from BIBSYSDEV/NP-47966-request-info-should-n…

…ot-use-real-http-client-in-tests refactor(ApiGateway): Make HttpClient overridable in RequestInfo #NP-47966
BIBSYSDEV · Oct 30, 2024 · a364ec1 · a364ec1
2 parents 083fc45 + 6cd6b23
commit a364ec1
Show file tree

Hide file tree

Showing 15 changed files with 155 additions and 127 deletions.
diff --git a/apigateway/src/main/java/nva/commons/apigateway/ApiGatewayHandler.java b/apigateway/src/main/java/nva/commons/apigateway/ApiGatewayHandler.java
@@ -17,6 +17,7 @@
 import java.io.BufferedWriter;
 import java.io.IOException;
 import java.io.OutputStreamWriter;
+import java.net.http.HttpClient;
 import java.nio.charset.StandardCharsets;
 import java.util.Arrays;
 import java.util.Collections;
@@ -51,23 +52,16 @@ public abstract class ApiGatewayHandler<I, O> extends RestRequestHandler<I, O> {
     public static final String ORIGIN_DELIMITER = ",";
     public static final String FALLBACK_ORIGIN  = "https://nva.sikt.no";
 
-    private final ObjectMapper objectMapper;
 
     private Supplier<Map<String, String>> additionalSuccessHeadersSupplier;
     private boolean isBase64Encoded;
 
     public ApiGatewayHandler(Class<I> iclass) {
-        this(iclass, new Environment());
+        this(iclass, new Environment(), HttpClient.newBuilder().build());
     }
 
-    public ApiGatewayHandler(Class<I> iclass, Environment environment) {
-        this(iclass, environment, defaultRestObjectMapper);
-        this.additionalSuccessHeadersSupplier = Collections::emptyMap;
-    }
-
-    public ApiGatewayHandler(Class<I> iclass, Environment environment, ObjectMapper objectMapper) {
-        super(iclass, environment);
-        this.objectMapper = objectMapper;
+    public ApiGatewayHandler(Class<I> iclass, Environment environment, HttpClient httpClient) {
+        super(iclass, environment, defaultRestObjectMapper, httpClient);
         this.additionalSuccessHeadersSupplier = Collections::emptyMap;
     }
 

diff --git a/apigateway/src/main/java/nva/commons/apigateway/ApiGatewayProxyHandler.java b/apigateway/src/main/java/nva/commons/apigateway/ApiGatewayProxyHandler.java
@@ -1,7 +1,7 @@
 package nva.commons.apigateway;
 
 import com.amazonaws.services.lambda.runtime.Context;
-import com.fasterxml.jackson.databind.ObjectMapper;
+import java.net.http.HttpClient;
 import nva.commons.apigateway.exceptions.ApiGatewayException;
 import nva.commons.core.Environment;
 import nva.commons.core.JacocoGenerated;
@@ -18,17 +18,12 @@ public abstract class ApiGatewayProxyHandler<I, O> extends ApiGatewayHandler<I,
 
     @JacocoGenerated
     protected ApiGatewayProxyHandler(Class<I> iclass) {
-        this(iclass, new Environment());
+        this(iclass, new Environment(), HttpClient.newBuilder().build());
     }
 
     @JacocoGenerated
-    protected ApiGatewayProxyHandler(Class<I> iclass, Environment environment) {
-        super(iclass, environment);
-    }
-
-    @JacocoGenerated
-    protected ApiGatewayProxyHandler(Class<I> iclass, Environment environment, ObjectMapper objectMapper) {
-        super(iclass, environment, objectMapper);
+    protected ApiGatewayProxyHandler(Class<I> iclass, Environment environment, HttpClient httpClient) {
+        super(iclass, environment, httpClient);
     }
 
     @Override

diff --git a/apigateway/src/main/java/nva/commons/apigateway/ApiS3GatewayHandler.java b/apigateway/src/main/java/nva/commons/apigateway/ApiS3GatewayHandler.java
@@ -1,7 +1,7 @@
 package nva.commons.apigateway;
 
 import com.amazonaws.services.lambda.runtime.Context;
-import com.fasterxml.jackson.databind.ObjectMapper;
+import java.net.http.HttpClient;
 import java.time.Duration;
 import nva.commons.apigateway.exceptions.BadRequestException;
 import nva.commons.core.Environment;
@@ -29,8 +29,8 @@ public ApiS3GatewayHandler(Class<I> iclass,
                                S3Client s3client,
                                S3Presigner s3Presigner,
                                Environment environment,
-                               ObjectMapper objectMapper) {
-        super(iclass, s3Presigner, environment, objectMapper);
+                               HttpClient httpClient) {
+        super(iclass, s3Presigner, environment, httpClient);
         this.s3client = s3client;
     }
 

diff --git a/apigateway/src/main/java/nva/commons/apigateway/ApiS3PresignerGatewayHandler.java b/apigateway/src/main/java/nva/commons/apigateway/ApiS3PresignerGatewayHandler.java
@@ -1,8 +1,8 @@
 package nva.commons.apigateway;
 
 import com.amazonaws.services.lambda.runtime.Context;
-import com.fasterxml.jackson.databind.ObjectMapper;
 import java.net.HttpURLConnection;
+import java.net.http.HttpClient;
 import java.time.Duration;
 import java.util.Map;
 import nva.commons.apigateway.exceptions.ApiGatewayException;
@@ -31,8 +31,8 @@ public ApiS3PresignerGatewayHandler(Class<I> iclass, S3Presigner s3Presigner) {
     public ApiS3PresignerGatewayHandler(Class<I> iclass,
                                         S3Presigner s3Presigner,
                                         Environment environment,
-                                        ObjectMapper objectMapper) {
-        super(iclass, environment, objectMapper);
+                                        HttpClient httpClient) {
+        super(iclass, environment, httpClient);
         this.s3presigner = s3Presigner;
     }
 

diff --git a/apigateway/src/main/java/nva/commons/apigateway/RequestInfo.java b/apigateway/src/main/java/nva/commons/apigateway/RequestInfo.java
@@ -39,6 +39,7 @@
 import com.fasterxml.jackson.annotation.JsonProperty;
 import com.fasterxml.jackson.core.JsonPointer;
 import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import com.google.common.net.HttpHeaders;
 import java.io.InputStream;
 import java.net.URI;
@@ -55,14 +56,15 @@
 import java.util.stream.Stream;
 import no.unit.nva.auth.CognitoUserInfo;
 import no.unit.nva.auth.FetchUserInfo;
-import no.unit.nva.commons.json.JsonUtils;
+import nva.commons.apigateway.exceptions.ApiIoException;
 import nva.commons.apigateway.exceptions.BadRequestException;
 import nva.commons.apigateway.exceptions.UnauthorizedException;
 import nva.commons.core.JacocoGenerated;
 import nva.commons.core.SingletonCollector;
 import nva.commons.core.StringUtils;
 import nva.commons.core.attempt.Failure;
 import nva.commons.core.exceptions.ExceptionUtils;
+import nva.commons.core.ioutils.IoUtils;
 import nva.commons.core.paths.UriWrapper;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -71,9 +73,9 @@
 public class RequestInfo {
 
     public static final String ERROR_FETCHING_COGNITO_INFO = "Could not fetch user information from Cognito:{}";
-    private static final HttpClient DEFAULT_HTTP_CLIENT = HttpClient.newBuilder().build();
     private static final Logger logger = LoggerFactory.getLogger(RequestInfo.class);
-    private final HttpClient httpClient;
+    private static final ObjectMapper mapper = defaultRestObjectMapper;
+    private HttpClient httpClient;
     private final Supplier<URI> cognitoUri;
     private final Supplier<URI> e2eTestsUserInfoUri;
     @JsonProperty(HEADERS_FIELD)
@@ -100,29 +102,36 @@ public RequestInfo(HttpClient httpClient, Supplier<URI> cognitoUri, Supplier<URI
         this.e2eTestsUserInfoUri = e2eTestsUserInfoUri;
     }
 
-    public RequestInfo() {
+    private RequestInfo() {
         this.headers = new HashMap<>();
         this.pathParameters = new HashMap<>();
         this.queryParameters = new HashMap<>();
         this.multiValueQueryStringParameters = new HashMap<>();
         this.otherProperties = new LinkedHashMap<>(); // ordinary HashMap and ConcurrentHashMap fail.
         this.requestContext = defaultRestObjectMapper.createObjectNode();
-        this.httpClient = DEFAULT_HTTP_CLIENT;
+        this.httpClient = HttpClient.newBuilder().build();
         this.cognitoUri = DEFAULT_COGNITO_URI;
         this.e2eTestsUserInfoUri = RequestInfoConstants.E2E_TESTING_USER_INFO_ENDPOINT;
     }
 
-    public static RequestInfo fromRequest(InputStream requestStream) {
-        return attempt(() -> JsonUtils.dtoObjectMapper.readValue(requestStream, RequestInfo.class)).orElseThrow();
+    public static RequestInfo fromRequest(InputStream requestStream, HttpClient httpClient) throws ApiIoException {
+        String inputString = IoUtils.streamToString(requestStream);
+        return fromString(inputString, httpClient);
+    }
+
+    public static RequestInfo fromString(String inputString, HttpClient httpClient) throws ApiIoException {
+        var requestInfo = new ApiMessageParser<>(mapper).getRequestInfo(inputString);
+        requestInfo.setHttpClient(httpClient);
+        return requestInfo;
     }
 
     @JsonIgnore
     public String getHeader(String header) {
         return getHeaders().entrySet().stream()
-                .filter(entry -> entry.getKey().equalsIgnoreCase(header))
-                .findFirst()
-                .map(Map.Entry::getValue)
-                .orElseThrow(() -> new IllegalArgumentException(MISSING_FROM_HEADERS + header));
+                   .filter(entry -> entry.getKey().equalsIgnoreCase(header))
+                   .findFirst()
+                   .map(Map.Entry::getValue)
+                   .orElseThrow(() -> new IllegalArgumentException(MISSING_FROM_HEADERS + header));
     }
 
     @JsonIgnore
@@ -199,6 +208,11 @@ public void setOtherProperties(Map<String, Object> otherProperties) {
         this.otherProperties = otherProperties;
     }
 
+    @JacocoGenerated
+    public void setHttpClient(HttpClient httpClient) {
+        this.httpClient = httpClient;
+    }
+
     public Map<String, String> getHeaders() {
         return headers;
     }
@@ -340,7 +354,7 @@ public URI getPersonCristinId() throws UnauthorizedException {
     @JsonIgnore
     public URI getPersonAffiliation() throws UnauthorizedException {
         return extractPersonAffiliationForTests().or(this::fetchPersonAffiliation)
-            .orElseThrow(UnauthorizedException::new);
+                   .orElseThrow(UnauthorizedException::new);
     }
 
     @JsonIgnore
@@ -455,7 +469,7 @@ private List<AccessRightEntry> fetchAvailableRights() {
         return userInfo
                    .map(CognitoUserInfo::getAccessRights)
                    .map(accessRightEntryStr -> AccessRightEntry.fromCsvForCustomer(accessRightEntryStr, userInfo.get()
-                                                                                                 .getCurrentCustomer()))
+                                                                                                            .getCurrentCustomer()))
                    .map(stream -> stream.collect(Collectors.toList()))
                    .orElseGet(Collections::emptyList);
     }

diff --git a/apigateway/src/main/java/nva/commons/apigateway/RestRequestHandler.java b/apigateway/src/main/java/nva/commons/apigateway/RestRequestHandler.java
@@ -5,11 +5,13 @@
 import static nva.commons.core.exceptions.ExceptionUtils.stackTraceInSingleLine;
 import com.amazonaws.services.lambda.runtime.Context;
 import com.amazonaws.services.lambda.runtime.RequestStreamHandler;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import com.google.common.net.HttpHeaders;
 import com.google.common.net.MediaType;
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStream;
+import java.net.http.HttpClient;
 import java.util.Arrays;
 import java.util.List;
 import java.util.stream.Collectors;
@@ -42,7 +44,9 @@ public abstract class RestRequestHandler<I, O> implements RequestStreamHandler {
     protected final Environment environment;
     private static final Logger logger = LoggerFactory.getLogger(RestRequestHandler.class);
     private final transient Class<I> iclass;
-    private final transient ApiMessageParser<I> inputParser = new ApiMessageParser<>();
+    private final transient ApiMessageParser<I> inputParser;
+    protected final ObjectMapper objectMapper;
+    private final HttpClient httpClient;
 
     protected transient OutputStream outputStream;
     protected transient String allowedOrigin;
@@ -130,9 +134,13 @@ private MediaType defaultResponseContentTypeWhenNotSpecifiedByClientRequest() {
      * @param iclass      The class object of the input class.
      * @param environment the Environment from where the handler will read ENV variables.
      */
-    public RestRequestHandler(Class<I> iclass, Environment environment) {
+    public RestRequestHandler(Class<I> iclass, Environment environment, ObjectMapper objectMapper,
+                              HttpClient httpClient) {
         this.iclass = iclass;
         this.environment = environment;
+        this.inputParser = new ApiMessageParser<>(objectMapper);
+        this.objectMapper = objectMapper;
+        this.httpClient = httpClient;
     }
 
     @Override
@@ -145,7 +153,8 @@ public void handleRequest(InputStream inputStream, OutputStream outputStream, Co
             inputObject = attempt(() -> parseInput(inputString))
                               .orElseThrow(this::parsingExceptionToBadRequestException);
 
-            RequestInfo requestInfo = inputParser.getRequestInfo(inputString);
+            RequestInfo requestInfo = RequestInfo.fromString(inputString, httpClient);
+            requestInfo.setHttpClient(httpClient);
             setAllowedOrigin(requestInfo);
 
             validateRequest(inputObject, requestInfo, context);