Fix [Bug] API GW V2 Payload setting same header on Single and MultiValue micronaut-projects#1858

Author: driverpt

function-aws-api-proxy-test/src/main/java/io/micronaut/function/aws/proxy/test/DefaultServletToAwsProxyResponseAdapter.java

@@ -73,7 +73,8 @@ private void populateHeaders(@NonNull ConversionService conversionService,
         MutableMapListOfStringAndMapStringConvertibleMultiValue entries = new MutableMapListOfStringAndMapStringConvertibleMultiValue(conversionService, multiValueHeaders, singleHeaders);
 
         for (String name: entries.names()) {
-            response.addHeader(name, String.join(",", entries.getAll(name)));
+            entries.getAll(name)
+                .forEach(headerValue -> response.addHeader(name, headerValue));
         }
     }
 

function-aws-api-proxy-test/src/test/groovy/io/micronaut/function/aws/proxy/test/AwsApiProxyTestServerSpec.groovy

@@ -1,5 +1,6 @@
 package io.micronaut.function.aws.proxy.test
 
+import io.micronaut.http.HttpHeaders
 import io.micronaut.http.HttpRequest
 import io.micronaut.http.HttpResponse
 import io.micronaut.http.HttpStatus
@@ -79,6 +80,19 @@ class AwsApiProxyTestServerSpec extends Specification {
         response.body.get() == (1..256).collect { it as byte } as byte[]
     }
 
+    @Issue('https://github.com/micronaut-projects/micronaut-aws/issues/1858')
+    void 'can return a ByteArray'() {
+        when:
+        HttpResponse<?> response = client.toBlocking()
+                .exchange(HttpRequest.GET('/multiple-headers'))
+
+        then:
+        response.status == HttpStatus.OK
+        response.headers.getAll(HttpHeaders.LINK) ==
+                ['<https://example1.com>; rel="next"', '<https://example2.com>; rel="prev"']
+        response.headers.get(HttpHeaders.CACHE_CONTROL) == "no-cache"
+    }
+
     @Controller
     static class TestController {
         @Get(value = '/test', produces = MediaType.TEXT_PLAIN)
@@ -105,5 +119,13 @@ class AwsApiProxyTestServerSpec extends Specification {
         HttpResponse<byte[]> byteArray() {
             return HttpResponse.ok((1..256).collect { it as byte } as byte[])
         }
+
+        @Post(value = '/multiple-headers', processes = MediaType.TEXT_PLAIN)
+        HttpResponse emptyBody() {
+            return HttpResponse.ok()
+                        .header(HttpHeaders.LINK, '<https://example1.com>; rel="next"')
+                        .header(HttpHeaders.LINK, '<https://example2.com>; rel="prev"')
+                        .header(HttpHeaders.CACHE_CONTROL,  "no-cache")
+        }
     }
 }

function-aws-api-proxy/src/main/java/io/micronaut/function/aws/proxy/MapCollapseUtils.java

@@ -43,7 +43,10 @@ private MapCollapseUtils() {
     public static Map<String, String> getSingleValueHeaders(MutableHttpHeaders headers) {
         Map<String, String> result = new HashMap<>();
         for (String paramName : headers.names()) {
-            result.put(paramName, headers.get(paramName));
+            List<String> values = headers.getAll(paramName);
+            if (values.size() == 1) {
+                result.put(paramName, values.get(0));
+            }
         }
         return result;
     }
@@ -57,7 +60,10 @@ public static Map<String, String> getSingleValueHeaders(MutableHttpHeaders heade
     public static Map<String, List<String>> getMultiHeaders(MutableHttpHeaders headers) {
         Map<String, List<String>> result = new HashMap<>();
         for (String paramName : headers.names()) {
-            result.put(paramName, headers.getAll(paramName));
+            List<String> values = headers.getAll(paramName);
+            if (values.size() > 1) {
+                result.put(paramName, values);
+            }
         }
         return result;
     }