feat/response-lib

MadebyAe · Apr 13, 2024 · 137813e · 137813e
1 parent 091118e
commit 137813e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 14 deletions.
diff --git a/src/headers.rs b/src/headers.rs
@@ -1,6 +1,6 @@
 use lambda_http::Request;
-use std::collections::HashMap;
 use serde_json::{json, Value};
+use std::collections::HashMap;
 
 pub fn get_header_value(request: &Request, value: String) -> String {
     return request.headers().get(value).map_or_else(
@@ -46,21 +46,19 @@ pub fn get_header_cookies(request: &Request) -> Value {
 #[cfg(test)]
 mod headers_tests {
     use super::*;
-    use lambda_http::http::header::{HeaderValue, HeaderName};
-    use lambda_http::{Request, Body};
+    use lambda_http::http::header::{HeaderName, HeaderValue};
+    use lambda_http::{Body, Request};
 
     fn mock_request(headers: Vec<(&str, &str)>) -> Request {
         let mut request = Request::new(Body::Empty); // Correct use of Request::new with Body::Empty for lambda_http
 
         for (key, value) in headers {
             match key.parse::<HeaderName>() {
-                Ok(parsed_key) => {
-                    match HeaderValue::from_str(value) {
-                        Ok(header_value) => {
-                            request.headers_mut().insert(parsed_key, header_value);
-                        },
-                        Err(e) => eprintln!("Failed to create HeaderValue: {}", e),
+                Ok(parsed_key) => match HeaderValue::from_str(value) {
+                    Ok(header_value) => {
+                        request.headers_mut().insert(parsed_key, header_value);
                     }
+                    Err(e) => eprintln!("Failed to create HeaderValue: {}", e),
                 },
                 Err(e) => eprintln!("Failed to parse HeaderName: {}", e),
             }

diff --git a/src/mongodb.rs b/src/mongodb.rs
@@ -40,8 +40,8 @@ pub fn get_mongodb_client() -> Result<&'static Client, Box<dyn Error + Send + Sy
 
 #[cfg(test)]
 mod mongodb_tests {
-    use std::env;
     use super::*;
+    use std::env;
 
     #[tokio::test]
     async fn test_create_mongodb_client() {

diff --git a/src/response.rs b/src/response.rs
@@ -36,8 +36,8 @@ pub use json_error;
 #[macro_export]
 macro_rules! json_ok {
     ($data:tt) => {{
-        use lambda_http::Body;
         use lambda_http::http::Response;
+        use lambda_http::Body;
         use serde_json::json;
 
         let body = json!($data).to_string();
@@ -53,10 +53,10 @@ pub use json_ok;
 
 #[cfg(test)]
 mod response_tests {
-    use lambda_http::http::{StatusCode};
-    use lambda_http::{Body, Response};
-    use serde_json::{Value, from_slice};
     use super::*;
+    use lambda_http::http::StatusCode;
+    use lambda_http::{Body, Response};
+    use serde_json::{from_slice, Value};
 
     fn extract_body(response: Response<Body>) -> Value {
         let body = response.into_body();