update e2e unittest

Signed-off-by: dongjiang1989 <[email protected]>
alibaba · Jan 8, 2024 · d8411a0 · d8411a0
1 parent 400c0c4
commit d8411a0
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/test/e2e/main_test.go b/test/e2e/main_test.go
@@ -131,15 +131,24 @@ func testInsertAgent(ctx context.Context) func(t *testing.T) {
 				Annotations: map[string]string{},
 			},
 			Spec: appsv1.DaemonSetSpec{
+				Selector: &metav1.LabelSelector{
+					MatchLabels: map[string]string{
+						"name": "test-agent",
+					},
+				},
 				Template: corev1.PodTemplateSpec{
+					ObjectMeta: metav1.ObjectMeta{
+						Labels: map[string]string{
+							"name": "test-agent",
+						},
+					},
 					Spec: corev1.PodSpec{
 						Containers: []corev1.Container{
 							{
 								Name:  "agent",
 								Image: testframework.Image,
 							},
 						},
-						RestartPolicy: corev1.RestartPolicyOnFailure,
 					},
 				},
 			},