Add with_request_context to RequestExt (#523)

sgasse · web-flow · commit b28b8686cc68 · 2022-09-05T10:00:42.000-07:00
The trait `RequestExt` already includes functions to add query string parameters, path parameters and stage variables to a request. This commit adds another function to allow adding a `RequestContext` to the request. This is useful in cases where a lambda function is called by the API Gateway with a custom authorizer that adds parameters to the `RequestContext`. Related to #522 Co-authored-by: Simon Gasse <sgasse@users.noreply.github.com>
diff --git a/lambda-http/src/ext.rs b/lambda-http/src/ext.rs
@@ -165,6 +165,11 @@ pub trait RequestExt {
     /// Return request context data assocaited with the ALB or API gateway request
     fn request_context(&self) -> RequestContext;
 
+    /// Configures instance with request context
+    ///
+    /// This is intended for use in mock testing contexts.
+    fn with_request_context(self, context: RequestContext) -> Self;
+
     /// Return the Result of a payload parsed into a serde Deserializeable
     /// type
     ///
@@ -255,6 +260,12 @@ impl RequestExt for http::Request<Body> {
             .expect("Request did not contain a request context")
     }
 
+    fn with_request_context(self, context: RequestContext) -> Self {
+        let mut s = self;
+        s.extensions_mut().insert(context);
+        s
+    }
+
     fn lambda_context(&self) -> Context {
         self.extensions()
             .get::<Context>()
