Fix unit tests

Signed-off-by: Sreekanth <[email protected]>
numaproj · Feb 20, 2025 · e88ae20 · e88ae20
1 parent 60a0487
commit e88ae20
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rust/serving/src/config.rs b/rust/serving/src/config.rs
@@ -281,7 +281,7 @@ mod tests {
             app_listen_port: 8443,
             metrics_server_listen_port: 3001,
             upstream_addr: "localhost:8888".into(),
-            drain_timeout_secs: 10,
+            drain_timeout_secs: 120,
             redis: RedisConfig {
                 addr: "redis://redis:6379".into(),
                 max_tasks: 50,
@@ -333,7 +333,7 @@ mod tests {
             app_listen_port: 8443,
             metrics_server_listen_port: 3001,
             upstream_addr: "localhost:8888".into(),
-            drain_timeout_secs: 10,
+            drain_timeout_secs: 120,
             redis: RedisConfig {
                 addr: "redis://redis:6379".into(),
                 max_tasks: 50,