tests(lib) catch possible 'dispatch_http_call' failures

Kong · Nov 13, 2024 · 9238d4a · 9238d4a
1 parent 8aadfa5
commit 9238d4a
Show file tree

Hide file tree

Showing 6 changed files with 54 additions and 11 deletions.
diff --git a/t/lib/Cargo.lock b/t/lib/Cargo.lock
diff --git a/t/lib/proxy-wasm-tests/context-checks/src/lib.rs b/t/lib/proxy-wasm-tests/context-checks/src/lib.rs
@@ -69,13 +69,23 @@ impl HttpContext for TestHttp {
             .get_config("on_http_dispatch_response")
             .is_some()
         {
-            let _ = self.dispatch_http_call(
+            match self.dispatch_http_call(
                 self.tester.get_config("host").unwrap_or(""),
                 vec![(":path", "/dispatch"), (":method", "GET")],
                 None,
                 vec![],
                 Duration::from_secs(0),
-            );
+            ) {
+                Ok(_) => {}
+                Err(status) => match status {
+                    Status::BadArgument => error!("dispatch_http_call: bad argument"),
+                    Status::InternalFailure => error!("dispatch_http_call: internal failure"),
+                    status => panic!(
+                        "dispatch_http_call: unexpected status \"{}\"",
+                        status as u32
+                    ),
+                },
+            }
 
             return Action::Pause;
         }

diff --git a/t/lib/proxy-wasm-tests/hostcalls/src/lib.rs b/t/lib/proxy-wasm-tests/hostcalls/src/lib.rs
@@ -156,14 +156,23 @@ impl RootContext for TestRoot {
 
                 info!("call {}", self.n_sync_calls);
 
-                self.dispatch_http_call(
+                match self.dispatch_http_call(
                     self.get_config("host").unwrap_or(""),
                     headers,
                     self.get_config("body").map(|v| v.as_bytes()),
                     vec![],
                     timeout,
-                )
-                .unwrap();
+                ) {
+                    Ok(_) => {}
+                    Err(status) => match status {
+                        Status::BadArgument => error!("dispatch_http_call: bad argument"),
+                        Status::InternalFailure => error!("dispatch_http_call: internal failure"),
+                        status => panic!(
+                            "dispatch_http_call: unexpected status \"{}\"",
+                            status as u32
+                        ),
+                    },
+                }
             }
             _ => (),
         }

diff --git a/t/lib/proxy-wasm-tests/hostcalls/src/types/test_http.rs b/t/lib/proxy-wasm-tests/hostcalls/src/types/test_http.rs
@@ -247,13 +247,22 @@ impl TestHttp {
             None => self.get_config("host").unwrap_or(""),
         };
 
-        self.dispatch_http_call(
+        match self.dispatch_http_call(
             host,
             headers,
             self.get_config("body").map(|v| v.as_bytes()),
             vec![],
             timeout,
-        )
-        .unwrap();
+        ) {
+            Ok(_) => {}
+            Err(status) => match status {
+                Status::BadArgument => error!("dispatch_http_call: bad argument"),
+                Status::InternalFailure => error!("dispatch_http_call: internal failure"),
+                status => panic!(
+                    "dispatch_http_call: unexpected status \"{}\"",
+                    status as u32
+                ),
+            },
+        }
     }
 }
diff --git a/t/lib/proxy-wasm-tests/rust-sdk-ver-zero-one/Cargo.toml b/t/lib/proxy-wasm-tests/rust-sdk-ver-zero-one/Cargo.toml
@@ -11,3 +11,4 @@ crate-type = ["cdylib"]
 [dependencies]
 proxy-wasm = "0.1"
 http = "0.2"
+log = "0.4"
diff --git a/t/lib/proxy-wasm-tests/rust-sdk-ver-zero-one/src/filter.rs b/t/lib/proxy-wasm-tests/rust-sdk-ver-zero-one/src/filter.rs
@@ -1,6 +1,7 @@
 #![allow(clippy::single_match)]
 
 use http::StatusCode;
+use log::*;
 use proxy_wasm::{traits::*, types::*};
 use std::collections::HashMap;
 use std::time::Duration;
@@ -88,7 +89,7 @@ impl HttpContext for TestHttpHostcalls {
         if let Some(test) = self.config.get("test") {
             match test.as_str() {
                 "dispatch" => {
-                    self.dispatch_http_call(
+                    match self.dispatch_http_call(
                         self.config.get("host").map(|v| v.as_str()).unwrap_or(""),
                         vec![
                             (":method", "GET"),
@@ -104,8 +105,20 @@ impl HttpContext for TestHttpHostcalls {
                         self.config.get("body").map(|v| v.as_bytes()),
                         vec![],
                         Duration::from_secs(3),
-                    )
-                    .unwrap();
+                    ) {
+                        Ok(_) => {}
+                        Err(status) => match status {
+                            Status::BadArgument => error!("dispatch_http_call: bad argument"),
+                            Status::InternalFailure => {
+                                error!("dispatch_http_call: internal failure")
+                            }
+                            status => panic!(
+                                "dispatch_http_call: unexpected status \"{}\"",
+                                status as u32
+                            ),
+                        },
+                    }
+
                     return Action::Pause;
                 }
                 "proxy_get_header_map_value" => {