test(drain): add test for custom controller pod

Signed-off-by: vadasambar <[email protected]>

cluster-autoscaler/utils/drain/drain_test.go

@@ -183,6 +183,20 @@ func TestDrain(t *testing.T) {
 		},
 	}
 
+	customControllerPod := &apiv1.Pod{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "bar",
+			Namespace: "default",
+			// Using names like FooController is discouraged
+			// https://github.com/kubernetes/community/blob/master/contributors/devel/sig-architecture/api-conventions.md#naming-conventions
+			// vadasambar: I am using it here just because `FooController`` is easier to understand than say `FooSet`
+			OwnerReferences: GenerateOwnerReferences("Foo", "FooController", "apps/v1", ""),
+		},
+		Spec: apiv1.PodSpec{
+			NodeName: "node",
+		},
+	}
+
 	nakedPod := &apiv1.Pod{
 		ObjectMeta: metav1.ObjectMeta{
 			Name:      "bar",
@@ -470,6 +484,14 @@ func TestDrain(t *testing.T) {
 			expectPods:          []*apiv1.Pod{},
 			expectDaemonSetPods: []*apiv1.Pod{},
 		},
+		{
+			description:         "Custom-controller-managed pod",
+			pods:                []*apiv1.Pod{customControllerPod},
+			pdbs:                []*policyv1.PodDisruptionBudget{},
+			expectFatal:         false,
+			expectPods:          []*apiv1.Pod{customControllerPod},
+			expectDaemonSetPods: []*apiv1.Pod{},
+		},
 		{
 			description:         "naked pod",
 			pods:                []*apiv1.Pod{nakedPod},