Unit tests for serving source

Signed-off-by: Sreekanth <[email protected]>
numaproj · Jan 3, 2025 · e893964 · e893964
1 parent 57c4cb7
commit e893964
Show file tree

Hide file tree

Showing 4 changed files with 225 additions and 11 deletions.
diff --git a/rust/Cargo.lock b/rust/Cargo.lock
diff --git a/rust/serving/Cargo.toml b/rust/serving/Cargo.toml
@@ -18,6 +18,7 @@ backoff.workspace = true
 axum.workspace = true
 axum-server.workspace = true
 bytes.workspace = true
+rustls.workspace = true
 axum-macros = "0.4.1"
 hyper-util = { version = "0.1.6", features = ["client-legacy"] }
 serde = { version = "1.0.204", features = ["derive"] }
@@ -35,4 +36,5 @@ prometheus-client = "0.22.3"
 thiserror = "1.0.63"
 
 [dev-dependencies]
+reqwest = {version= "0.12.12", features = ["json"]}
 rustls.workspace = true
diff --git a/rust/serving/src/lib.rs b/rust/serving/src/lib.rs
@@ -38,6 +38,9 @@ pub(crate) async fn serve<T>(
 where
     T: Clone + Send + Sync + Store + 'static,
 {
+    rustls::crypto::aws_lc_rs::default_provider()
+        .install_default()
+        .expect("Failed to set crypto provider");
     let (cert, key) = generate_certs()?;
 
     let tls_config = RustlsConfig::from_pem(cert.pem().into(), key.serialize_pem().into())