From 27179629033d32939af7940db2d2fbe54cdba2a5 Mon Sep 17 00:00:00 2001
From: sangitaray2021 <sangitaray@microsoft.com>
Date: Thu, 19 Sep 2024 12:27:28 +0530
Subject: [PATCH] fixed unittest

Signed-off-by: sangitaray2021 <sangitaray@microsoft.com>
---
 pkg/restore/restore.go      |  1 +
 pkg/restore/restore_test.go | 34 ++++++++++++++++++++--------------
 pkg/util/kube/utils_test.go | 20 +++++++++++---------
 3 files changed, 32 insertions(+), 23 deletions(-)

diff --git a/pkg/restore/restore.go b/pkg/restore/restore.go
index 5c8d1f0664..c7164624a3 100644
--- a/pkg/restore/restore.go
+++ b/pkg/restore/restore.go
@@ -326,6 +326,7 @@ func (kr *kubernetesRestorer) RestoreWithResolvers(
 		backupVolumeInfoMap:            req.BackupVolumeInfoMap,
 		restoreVolumeInfoTracker:       req.RestoreVolumeInfoTracker,
 		hooksWaitExecutor:              hooksWaitExecutor,
+		namespaceDeletionStatusTracker: kr.namespaceDeletionStatusTracker,
 	}
 
 	return restoreCtx.execute()
diff --git a/pkg/restore/restore_test.go b/pkg/restore/restore_test.go
index 8e239e9a70..ef08253a27 100644
--- a/pkg/restore/restore_test.go
+++ b/pkg/restore/restore_test.go
@@ -58,6 +58,7 @@ import (
 	uploadermocks "github.com/vmware-tanzu/velero/pkg/podvolume/mocks"
 	"github.com/vmware-tanzu/velero/pkg/test"
 	"github.com/vmware-tanzu/velero/pkg/types"
+	"github.com/vmware-tanzu/velero/pkg/util/kube"
 	kubeutil "github.com/vmware-tanzu/velero/pkg/util/kube"
 	. "github.com/vmware-tanzu/velero/pkg/util/results"
 )
@@ -2292,10 +2293,11 @@ func TestShouldRestore(t *testing.T) {
 			h := newHarness(t)
 
 			ctx := &restoreContext{
-				log:                        h.log,
-				dynamicFactory:             client.NewDynamicFactory(h.DynamicClient),
-				namespaceClient:            h.KubeClient.CoreV1().Namespaces(),
-				resourceTerminatingTimeout: time.Millisecond,
+				log:                            h.log,
+				dynamicFactory:                 client.NewDynamicFactory(h.DynamicClient),
+				namespaceClient:                h.KubeClient.CoreV1().Namespaces(),
+				resourceTerminatingTimeout:     time.Millisecond,
+				namespaceDeletionStatusTracker: kube.NewNamespaceDeletionStatusTracker(),
 			}
 
 			for _, resource := range tc.apiResources {
@@ -3710,9 +3712,10 @@ func newHarness(t *testing.T) *harness {
 			fileSystem:                 test.NewFakeFileSystem(),
 
 			// unsupported
-			podVolumeRestorerFactory: nil,
-			podVolumeTimeout:         0,
-			kbClient:                 kbClient,
+			podVolumeRestorerFactory:       nil,
+			podVolumeTimeout:               0,
+			kbClient:                       kbClient,
+			namespaceDeletionStatusTracker: kube.NewNamespaceDeletionStatusTracker(),
 		},
 		log: log,
 	}
@@ -3899,9 +3902,10 @@ func TestIsAlreadyExistsError(t *testing.T) {
 		h := newHarness(t)
 
 		ctx := &restoreContext{
-			log:             h.log,
-			dynamicFactory:  client.NewDynamicFactory(h.DynamicClient),
-			namespaceClient: h.KubeClient.CoreV1().Namespaces(),
+			log:                            h.log,
+			dynamicFactory:                 client.NewDynamicFactory(h.DynamicClient),
+			namespaceClient:                h.KubeClient.CoreV1().Namespaces(),
+			namespaceDeletionStatusTracker: kube.NewNamespaceDeletionStatusTracker(),
 		}
 
 		if test.apiResource != nil {
@@ -4018,7 +4022,8 @@ func TestHasCSIVolumeSnapshot(t *testing.T) {
 		h := newHarness(t)
 
 		ctx := &restoreContext{
-			log: h.log,
+			log:                            h.log,
+			namespaceDeletionStatusTracker: kube.NewNamespaceDeletionStatusTracker(),
 		}
 
 		if tc.vs != nil {
@@ -4118,9 +4123,10 @@ func TestHasSnapshotDataUpload(t *testing.T) {
 		h := newHarness(t)
 
 		ctx := &restoreContext{
-			log:      h.log,
-			kbClient: h.restorer.kbClient,
-			restore:  tc.restore,
+			log:                            h.log,
+			kbClient:                       h.restorer.kbClient,
+			restore:                        tc.restore,
+			namespaceDeletionStatusTracker: kube.NewNamespaceDeletionStatusTracker(),
 		}
 
 		if tc.duResult != nil {
diff --git a/pkg/util/kube/utils_test.go b/pkg/util/kube/utils_test.go
index 6feb9bd946..4bc77e3bf7 100644
--- a/pkg/util/kube/utils_test.go
+++ b/pkg/util/kube/utils_test.go
@@ -47,14 +47,15 @@ func TestNamespaceAndName(t *testing.T) {
 
 func TestEnsureNamespaceExistsAndIsReady(t *testing.T) {
 	tests := []struct {
-		name                  string
-		expectNSFound         bool
-		nsPhase               corev1.NamespacePhase
-		nsDeleting            bool
-		expectCreate          bool
-		alreadyExists         bool
-		expectedResult        bool
-		expectedCreatedResult bool
+		name                           string
+		expectNSFound                  bool
+		nsPhase                        corev1.NamespacePhase
+		nsDeleting                     bool
+		expectCreate                   bool
+		alreadyExists                  bool
+		expectedResult                 bool
+		expectedCreatedResult          bool
+		namespaceDeletionStatusTracker NamespaceDeletionStatusTracker
 	}{
 		{
 			name:                  "namespace found, not deleting",
@@ -97,6 +98,7 @@ func TestEnsureNamespaceExistsAndIsReady(t *testing.T) {
 		},
 	}
 
+	namespaceDeletionStatusTracker := NewNamespaceDeletionStatusTracker()
 	for _, test := range tests {
 		t.Run(test.name, func(t *testing.T) {
 			namespace := &corev1.Namespace{
@@ -132,7 +134,7 @@ func TestEnsureNamespaceExistsAndIsReady(t *testing.T) {
 				nsClient.On("Create", namespace).Return(namespace, nil)
 			}
 
-			result, nsCreated, _ := EnsureNamespaceExistsAndIsReady(namespace, nsClient, timeout)
+			result, nsCreated, _ := EnsureNamespaceExistsAndIsReady(namespace, nsClient, timeout, namespaceDeletionStatusTracker)
 
 			assert.Equal(t, test.expectedResult, result)
 			assert.Equal(t, test.expectedCreatedResult, nsCreated)