Pass cookies from LambdaRequest::ApiGatewayV2 to Request (#258)

endor · web-flow · commit ed40f5a7b7e8 · 2020-09-16T18:55:56.000-04:00
diff --git a/lambda-http/src/request.rs b/lambda-http/src/request.rs
@@ -311,15 +311,22 @@ impl<'a> From<LambdaRequest<'a>> for http::Request<Body> {
             LambdaRequest::ApiGatewayV2 {
                 raw_path,
                 raw_query_string,
-                headers,
+                mut headers,
                 query_string_parameters,
                 path_parameters,
                 stage_variables,
                 body,
                 is_base64_encoded,
                 request_context,
+                cookies,
                 ..
             } => {
+                if let Some(cookies) = cookies {
+                    if let Ok(header_value) = http::header::HeaderValue::from_str(&cookies.join(";")) {
+                        headers.append(http::header::COOKIE, header_value);
+                    }
+                }
+
                 let builder = http::Request::builder()
                     .method(request_context.http.method.as_ref())
                     .uri({
@@ -569,8 +576,15 @@ mod tests {
             format!("event was not parsed as expected {:?} given {}", result, input)
         );
         let req = result.expect("failed to parse request");
+        let cookie_header = req
+            .headers()
+            .get(http::header::COOKIE)
+            .ok_or_else(|| "Cookie header not found".to_string())
+            .and_then(|v| v.to_str().map_err(|e| e.to_string()));
+
         assert_eq!(req.method(), "POST");
         assert_eq!(req.uri(), "https://id.execute-api.us-east-1.amazonaws.com/my/path?parameter1=value1&parameter1=value2&parameter2=value");
+        assert_eq!(cookie_header, Ok("cookie1=value1;cookie2=value2"));
     }
 
     #[test]
diff --git a/lambda-http/tests/data/apigw_v2_proxy_request.json b/lambda-http/tests/data/apigw_v2_proxy_request.json
@@ -3,7 +3,7 @@
     "routeKey": "$default",
     "rawPath": "/my/path",
     "rawQueryString": "parameter1=value1&parameter1=value2&parameter2=value",
-    "cookies": [ "cookie1", "cookie2" ],
+    "cookies": [ "cookie1=value1", "cookie2=value2" ],
     "headers": {
       "Header1": "value1",
       "Header2": "value2"
