Header data class (#27)

Author: raoulk

router/src/main/kotlin/io/moia/router/APIGatewayProxyEventExtensions.kt

@@ -6,6 +6,9 @@ import com.google.common.net.MediaType
 import java.net.URI
 import java.util.Base64
 
+/** Data class that represents an HTTP header */
+data class Header(val name: String, val value: String)
+
 fun APIGatewayProxyRequestEvent.acceptHeader() = getHeaderCaseInsensitive("accept")
 fun APIGatewayProxyRequestEvent.acceptedMediaTypes() = acceptHeader()
     ?.split(",")
@@ -54,9 +57,15 @@ fun APIGatewayProxyRequestEvent.location(path: String): URI {
 fun APIGatewayProxyRequestEvent.withHeader(name: String, value: String) =
     this.also { if (headers == null) headers = mutableMapOf() }.also { headers[name] = value }
 
+fun APIGatewayProxyRequestEvent.withHeader(header: Header) =
+    this.withHeader(header.name, header.value)
+
 fun APIGatewayProxyResponseEvent.withHeader(name: String, value: String) =
     this.also { if (headers == null) headers = mutableMapOf() }.also { headers[name] = value }
 
+fun APIGatewayProxyResponseEvent.withHeader(header: Header) =
+    this.withHeader(header.name, header.value)
+
 fun APIGatewayProxyResponseEvent.withLocationHeader(request: APIGatewayProxyRequestEvent, path: String) =
     this.also { if (headers == null) headers = mutableMapOf() }.also { headers["location"] = request.location(path).toString() }
 

router/src/test/kotlin/io/moia/router/APIGatewayProxyEventExtensionsTest.kt

@@ -1,5 +1,6 @@
 package io.moia.router
 
+import com.amazonaws.services.lambda.runtime.events.APIGatewayProxyRequestEvent
 import com.amazonaws.services.lambda.runtime.events.APIGatewayProxyResponseEvent
 import org.assertj.core.api.BDDAssertions.then
 import org.junit.jupiter.api.Test
@@ -56,4 +57,20 @@ class APIGatewayProxyEventExtensionsTest {
 
         then(location.toString()).isEqualTo("http://example.com/some/path")
     }
+
+    @Test
+    fun `header class should work as expected with APIGatewayProxyRequestEvent`() {
+
+        val request = APIGatewayProxyRequestEvent().withHeader(Header("foo", "bar"))
+
+        then(request.headers["foo"]).isEqualTo("bar")
+    }
+
+    @Test
+    fun `header class should work as expected with APIGatewayProxyResponseEvent`() {
+
+        val request = APIGatewayProxyResponseEvent().withHeader(Header("foo", "bar"))
+
+        then(request.headers["foo"]).isEqualTo("bar")
+    }
 }