From 5fc7a6f0d477b7dcb9489f7b6841b01d7964fdbe Mon Sep 17 00:00:00 2001
From: mo khan <mo@mokhan.ca>
Date: Thu, 19 Jun 2025 17:18:07 -0600
Subject: refactor: rename PolicyServer to CheckService

---
 .gitignore                 | 4 +---
 src/authorization.rs       | 8 ++++----
 src/lib.rs                 | 2 +-
 src/main.rs                | 9 +++------
 tests/integration_tests.rs | 6 +++---
 5 files changed, 12 insertions(+), 17 deletions(-)

diff --git a/.gitignore b/.gitignore
index 6bfb8923..eb5a316c 100644
--- a/.gitignore
+++ b/.gitignore
@@ -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;
diff --git a/src/lib.rs b/src/lib.rs
index cb28e34b..b3791974 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -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),
-- 
cgit v1.2.3