From 137813e1e64031e20c5a37cc665d616b371c5169 Mon Sep 17 00:00:00 2001 From: Ae Date: Sat, 13 Apr 2024 16:41:10 -0700 Subject: [PATCH] feat/response-lib --- src/headers.rs | 16 +++++++--------- src/mongodb.rs | 2 +- src/response.rs | 8 ++++---- 3 files changed, 12 insertions(+), 14 deletions(-) diff --git a/src/headers.rs b/src/headers.rs index 9ae0624..cddc016 100644 --- 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::() { - 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 index 383c37c..4d48b38 100644 --- a/src/mongodb.rs +++ b/src/mongodb.rs @@ -40,8 +40,8 @@ pub fn get_mongodb_client() -> Result<&'static Client, Box {{ - 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) -> Value { let body = response.into_body();