fmt

Author: PartiallyUntyped

lambda-integration-tests/src/tenant_id_test.rs

@@ -3,9 +3,9 @@ use serde_json::{json, Value};
 
 async fn function_handler(event: LambdaEvent<Value>) -> Result<Value, Error> {
     let (event, context) = event.into_parts();
-    
+
     tracing::info!("Processing request with tenant ID: {:?}", context.tenant_id);
-    
+
     let response = json!({
         "statusCode": 200,
         "body": json!({

lambda-integration-tests/tests/tenant_id_prod_test.rs

@@ -2,48 +2,56 @@ use serde_json::json;
 
 #[test]
 fn test_tenant_id_functionality_in_production() {
-    let function_name = std::env::var("TENANT_ID_TEST_FUNCTION")
-        .expect("TENANT_ID_TEST_FUNCTION environment variable not set");
-    
+    let function_name =
+        std::env::var("TENANT_ID_TEST_FUNCTION").expect("TENANT_ID_TEST_FUNCTION environment variable not set");
+
     // Test with tenant ID
     let payload_with_tenant = json!({
         "test": "tenant_id_test",
         "message": "Testing with tenant ID"
     });
-    
+
     let output = std::process::Command::new("aws")
         .args([
-            "lambda", "invoke",
-            "--function-name", &function_name,
-            "--payload", &payload_with_tenant.to_string(),
-            "--cli-binary-format", "raw-in-base64-out",
-            "/tmp/tenant_response.json"
+            "lambda",
+            "invoke",
+            "--function-name",
+            &function_name,
+            "--payload",
+            &payload_with_tenant.to_string(),
+            "--cli-binary-format",
+            "raw-in-base64-out",
+            "/tmp/tenant_response.json",
         ])
         .output()
         .expect("Failed to invoke Lambda function");
-    
-    assert!(output.status.success(), "Lambda invocation failed: {}", 
-            String::from_utf8_lossy(&output.stderr));
-    
+
+    assert!(
+        output.status.success(),
+        "Lambda invocation failed: {}",
+        String::from_utf8_lossy(&output.stderr)
+    );
+
     // Read and verify response
-    let response = std::fs::read_to_string("/tmp/tenant_response.json")
-        .expect("Failed to read response file");
-    
-    let response_json: serde_json::Value = serde_json::from_str(&response)
-        .expect("Failed to parse response JSON");
-    
+    let response = std::fs::read_to_string("/tmp/tenant_response.json").expect("Failed to read response file");
+
+    let response_json: serde_json::Value = serde_json::from_str(&response).expect("Failed to parse response JSON");
+
     // Verify the function executed successfully
     assert_eq!(response_json["statusCode"], 200);
-    
+
     // Parse the body to check tenant_id field exists (even if null)
-    let body: serde_json::Value = serde_json::from_str(
-        response_json["body"].as_str().expect("Body should be a string")
-    ).expect("Failed to parse body JSON");
-    
-    assert!(body.get("tenant_id").is_some(), "tenant_id field should be present in response");
+    let body: serde_json::Value =
+        serde_json::from_str(response_json["body"].as_str().expect("Body should be a string"))
+            .expect("Failed to parse body JSON");
+
+    assert!(
+        body.get("tenant_id").is_some(),
+        "tenant_id field should be present in response"
+    );
     assert!(body.get("request_id").is_some(), "request_id should be present");
     assert_eq!(body["message"], "Tenant ID test successful");
-    
+
     println!("✅ Tenant ID production test passed");
     println!("Response: {}", serde_json::to_string_pretty(&response_json).unwrap());
 }

lambda-runtime/tests/tenant_id_integration.rs

@@ -1,11 +1,11 @@
-use lambda_runtime::Context;
 use http::HeaderMap;
+use lambda_runtime::Context;
 use std::sync::Arc;
 
 #[test]
 fn test_context_tenant_id_extraction() {
     let config = Arc::new(lambda_runtime::Config::default());
-    
+
     // Test with tenant ID
     let mut headers = HeaderMap::new();
     headers.insert("lambda-runtime-aws-request-id", "test-id".parse().unwrap());
@@ -27,21 +27,27 @@ fn test_context_tenant_id_extraction() {
 #[test]
 fn test_context_tenant_id_with_special_characters() {
     let config = Arc::new(lambda_runtime::Config::default());
-    
+
     // Test with tenant ID containing special characters
     let mut headers = HeaderMap::new();
     headers.insert("lambda-runtime-aws-request-id", "test-id".parse().unwrap());
     headers.insert("lambda-runtime-deadline-ms", "123456789".parse().unwrap());
-    headers.insert("lambda-runtime-aws-tenant-id", "tenant-with-dashes_and_underscores.123".parse().unwrap());
+    headers.insert(
+        "lambda-runtime-aws-tenant-id",
+        "tenant-with-dashes_and_underscores.123".parse().unwrap(),
+    );
 
     let context = Context::new("test-id", config, &headers).unwrap();
-    assert_eq!(context.tenant_id, Some("tenant-with-dashes_and_underscores.123".to_string()));
+    assert_eq!(
+        context.tenant_id,
+        Some("tenant-with-dashes_and_underscores.123".to_string())
+    );
 }
 
 #[test]
 fn test_context_tenant_id_empty_value() {
     let config = Arc::new(lambda_runtime::Config::default());
-    
+
     // Test with empty tenant ID
     let mut headers = HeaderMap::new();
     headers.insert("lambda-runtime-aws-request-id", "test-id".parse().unwrap());