From d5e528022024b28d97a22d0541ba38915af73b5b Mon Sep 17 00:00:00 2001
From: Feny Mehta <fbm3307@gmail.com>
Date: Wed, 25 Sep 2024 16:21:19 +0530
Subject: [PATCH] test case for delete

Signed-off-by: Feny Mehta <fbm3307@gmail.com>
---
 pkg/cmd/adm/restart_test.go | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/pkg/cmd/adm/restart_test.go b/pkg/cmd/adm/restart_test.go
index 04b958e..aa412b7 100644
--- a/pkg/cmd/adm/restart_test.go
+++ b/pkg/cmd/adm/restart_test.go
@@ -12,6 +12,7 @@ import (
 
 	"github.com/stretchr/testify/require"
 	appsv1 "k8s.io/api/apps/v1"
+	corev1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
 	"k8s.io/apimachinery/pkg/runtime"
@@ -129,13 +130,16 @@ func TestRestartDeployment(t *testing.T) {
 
 			streams, _, buf, _ := genericclioptions.NewTestIOStreams()
 			term := NewFakeTerminalWithResponse("Y")
-			newClient, fakeClient := NewFakeClients(t, deployment1)
+			pod := newPod(test.NamespacedName("toolchain-host-operator", "host-operator-controller-manager"))
+			newClient, fakeClient := NewFakeClients(t, deployment1, pod)
 			ctx := clicontext.NewCommandContext(term, newClient)
 
 			//when
 			err := restartDeployment(ctx, fakeClient, namespacedName.Namespace, tf, streams)
 			if tc.labelValue == "kubesaw-controller-manager" {
 				require.NoError(t, err, "non-OLM based deployment not found in")
+				err2 := deleteAndWaitForPods(ctx, fakeClient, *deployment1, tf, streams)
+				require.NoError(t, err2)
 			} else if tc.labelValue == "codeready-toolchain" {
 				require.NoError(t, err, "OLM based deployment not found in")
 				err := restartNonOlmDeployments(*deployment1, tf, streams)
@@ -207,6 +211,25 @@ func newDeployment(namespacedName types.NamespacedName, replicas int32) *appsv1.
 		},
 		Spec: appsv1.DeploymentSpec{
 			Replicas: &replicas,
+			Selector: &metav1.LabelSelector{MatchLabels: map[string]string{"host": "controller"}},
+		},
+	}
+}
+
+func newPod(namespacedName types.NamespacedName) *corev1.Pod { //nolint:unparam
+	return &corev1.Pod{
+		TypeMeta: metav1.TypeMeta{
+			APIVersion: "v1",
+			Kind:       "Pod",
+		},
+		ObjectMeta: metav1.ObjectMeta{
+			Namespace: namespacedName.Namespace,
+			Name:      namespacedName.Name,
+			Labels:    map[string]string{"host": "controller"},
+		},
+		Spec: corev1.PodSpec{},
+		Status: corev1.PodStatus{
+			Phase: "Running",
 		},
 	}
 }