test: extract helper function to create a check request

1 files changed, 10 insertions, 4 deletions

diff --git a/src/authorization/cedar_authorizer.rs b/src/authorization/cedar_authorizer.rs
index c1f5455d..49659d35 100644
--- a/src/authorization/cedar_authorizer.rs
+++ b/src/authorization/cedar_authorizer.rs
@@ -41,15 +41,15 @@ mod tests {
     use envoy_types::pb::envoy::service::auth::v3::{AttributeContext, attribute_context};
     use std::collections::HashMap;
 
-    fn create_test_request_with_headers(headers: HashMap<String, String>) -> CheckRequest {
+    fn create_request(
+        f: impl std::ops::FnOnce(&mut attribute_context::HttpRequest),
+    ) -> CheckRequest {
         please::build_with(|item: &mut CheckRequest| {
             item.attributes = Some(please::build_with(|item: &mut AttributeContext| {
                 item.request = Some(please::build_with(
                     |item: &mut attribute_context::Request| {
                         item.http = Some(please::build_with(
-                            |item: &mut attribute_context::HttpRequest| {
-                                item.headers = headers;
-                            },
+                            |item: &mut attribute_context::HttpRequest| f(item),
                         ));
                     },
                 ));
@@ -57,6 +57,12 @@ mod tests {
         })
     }
 
+    fn create_test_request_with_headers(headers: HashMap<String, String>) -> CheckRequest {
+        return create_request(|item: &mut attribute_context::HttpRequest| {
+            item.headers = headers;
+        });
+    }
+
     #[test]
     fn test_cedar_authorizer_allows_valid_token() {
         let authorizer = CedarAuthorizer::new();