rejects non 2xx http responses in tests

Signed-off-by: Jorge Prendes <[email protected]>
deislabs · Oct 25, 2023 · f5656b3 · f5656b3
1 parent d2efdc3
commit f5656b3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/tests/src/integration_test.rs b/tests/src/integration_test.rs
@@ -4,8 +4,8 @@ mod test {
     use crate::{list_pods, random_payload, retry_get, retry_put};
     use anyhow::Result;
 
-    const RETRY_TIMES: u32 = 5;
-    const INTERVAL_IN_SECS: u64 = 10;
+    const RETRY_TIMES: u32 = 25;
+    const INTERVAL_IN_SECS: u64 = 2;
 
     #[tokio::test]
     async fn slight_test() -> Result<()> {

diff --git a/tests/src/lib.rs b/tests/src/lib.rs
@@ -30,8 +30,9 @@ pub async fn retry_get(
         // verify res is ok and not 404
         match res {
             Ok(_) => {
-                if response_code != 404 {
-                    break;
+                println!("response_code: {}", response_code);
+                if response_code / 100 == 2 {
+                    break; // 2xx response
                 }
             }
             Err(e) => {