fix unit test

Signed-off-by: Varun Gupta <[email protected]>
vllm-project · Mar 3, 2025 · fac9b64 · fac9b64
1 parent 1153f19
commit fac9b64
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/pkg/plugins/gateway/algorithms/router_test.go b/pkg/plugins/gateway/algorithms/router_test.go
@@ -106,6 +106,10 @@ func TestWithIPPods(t *testing.T) {
 							Type:   v1.PodReady,
 							Status: v1.ConditionTrue,
 						},
+						{
+							Type:   v1.ContainersReady,
+							Status: v1.ConditionTrue,
+						},
 					},
 				},
 			},
@@ -121,6 +125,10 @@ func TestWithIPPods(t *testing.T) {
 							Type:   v1.PodReady,
 							Status: v1.ConditionTrue,
 						},
+						{
+							Type:   v1.ContainersReady,
+							Status: v1.ConditionTrue,
+						},
 					},
 				},
 			},
@@ -182,6 +190,10 @@ func TestSelectRandomPod(t *testing.T) {
 								Type:   v1.PodReady,
 								Status: v1.ConditionTrue,
 							},
+							{
+								Type:   v1.ContainersReady,
+								Status: v1.ConditionTrue,
+							},
 						},
 					},
 				},
@@ -199,6 +211,10 @@ func TestSelectRandomPod(t *testing.T) {
 								Type:   v1.PodReady,
 								Status: v1.ConditionTrue,
 							},
+							{
+								Type:   v1.ContainersReady,
+								Status: v1.ConditionTrue,
+							},
 						},
 					},
 				},
@@ -210,6 +226,10 @@ func TestSelectRandomPod(t *testing.T) {
 								Type:   v1.PodReady,
 								Status: v1.ConditionTrue,
 							},
+							{
+								Type:   v1.ContainersReady,
+								Status: v1.ConditionTrue,
+							},
 						},
 					},
 				},
@@ -221,6 +241,10 @@ func TestSelectRandomPod(t *testing.T) {
 								Type:   v1.PodReady,
 								Status: v1.ConditionTrue,
 							},
+							{
+								Type:   v1.ContainersReady,
+								Status: v1.ConditionTrue,
+							},
 						},
 					},
 				},