diff options
| -rw-r--r-- | .gitignore | 4 | ||||
| -rw-r--r-- | src/authorization.rs | 8 | ||||
| -rw-r--r-- | src/lib.rs | 2 | ||||
| -rw-r--r-- | src/main.rs | 9 | ||||
| -rw-r--r-- | tests/integration_tests.rs | 6 |
5 files changed, 12 insertions, 17 deletions
@@ -1,3 +1 @@ -tmp -*.pdf -/target +target diff --git a/src/authorization.rs b/src/authorization.rs index 22cd695c..279f62cc 100644 --- a/src/authorization.rs +++ b/src/authorization.rs @@ -3,10 +3,10 @@ use envoy_types::ext_authz::v3::pb::{CheckRequest, CheckResponse}; use tonic::{Request, Response, Status}; #[derive(Debug, Default)] -pub struct PolicyServer; +pub struct CheckService; #[tonic::async_trait] -impl envoy_types::ext_authz::v3::pb::Authorization for PolicyServer { +impl envoy_types::ext_authz::v3::pb::Authorization for CheckService { async fn check( &self, request: Request<CheckRequest>, @@ -75,7 +75,7 @@ mod tests { #[tokio::test] async fn test_check_allows_valid_bearer_token() { let token = String::from("valid-token"); - let server = PolicyServer::default(); + let server = CheckService::default(); let headers = create_headers_with_auth(&format!("Bearer {}", token)); let request = create_test_request_with_headers(headers); @@ -90,7 +90,7 @@ mod tests { #[tokio::test] async fn test_check_denies_invalid_bearer_token() { - let server = PolicyServer::default(); + let server = CheckService::default(); let request = create_test_request_with_headers(HashMap::new()); let response = server.check(request).await; @@ -1,3 +1,3 @@ pub mod authorization; -pub use authorization::PolicyServer; +pub use authorization::CheckService; diff --git a/src/main.rs b/src/main.rs index 25e2c88f..917cace5 100644 --- a/src/main.rs +++ b/src/main.rs @@ -2,16 +2,13 @@ use envoy_types::ext_authz::v3::pb::AuthorizationServer; use tonic::transport::Server; pub mod authorization; -use authorization::PolicyServer; +use authorization::CheckService; #[tokio::main] async fn main() -> Result<(), Box<dyn std::error::Error>> { let addr = "[::1]:50051".parse()?; - let authorization_service = AuthorizationServer::new(PolicyServer::default()); - let (health_reporter, health_service) = tonic_health::server::health_reporter(); - health_reporter - .set_serving::<AuthorizationServer<PolicyServer>>() - .await; + let authorization_service = AuthorizationServer::new(CheckService::default()); + let (_health_reporter, health_service) = tonic_health::server::health_reporter(); let reflection_service = tonic_reflection::server::Builder::configure() .register_encoded_file_descriptor_set(tonic_health::pb::FILE_DESCRIPTOR_SET) diff --git a/tests/integration_tests.rs b/tests/integration_tests.rs index 28cdf959..a3603c11 100644 --- a/tests/integration_tests.rs +++ b/tests/integration_tests.rs @@ -1,11 +1,11 @@ -use authzd::PolicyServer; +use authzd::CheckService; use envoy_types::ext_authz::v3::pb::Authorization; mod common; #[tokio::test] async fn test_success_response() { - let server = PolicyServer::default(); + let server = CheckService::default(); let headers = common::create_headers_with_auth("Bearer valid-token"); let request = common::create_test_request_with_headers(headers); @@ -22,7 +22,7 @@ async fn test_success_response() { #[tokio::test] async fn test_multiple() { - let server = PolicyServer::default(); + let server = CheckService::default(); let test_cases = vec![ ("Bearer valid-token", true), |