From d6221574e6260b6958707df50da63c13eac90c52 Mon Sep 17 00:00:00 2001
From: Chin-Ya Huang <chin-ya.huang@suse.com>
Date: Thu, 14 Nov 2024 09:35:05 +0800
Subject: [PATCH] test: func GetDeployment

longhorn/longhorn-9752

Signed-off-by: Chin-Ya Huang <chin-ya.huang@suse.com>
---
 kubernetes/deployment_test.go | 61 +++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 kubernetes/deployment_test.go

diff --git a/kubernetes/deployment_test.go b/kubernetes/deployment_test.go
new file mode 100644
index 00000000..fa1580d8
--- /dev/null
+++ b/kubernetes/deployment_test.go
@@ -0,0 +1,61 @@
+package kubernetes
+
+import (
+	"context"
+
+	"github.com/longhorn/go-common-libs/test"
+	. "gopkg.in/check.v1"
+
+	"k8s.io/client-go/kubernetes/fake"
+
+	appsv1 "k8s.io/api/apps/v1"
+	apierrors "k8s.io/apimachinery/pkg/api/errors"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+)
+
+func (s *TestSuite) TestGetDeployment(c *C) {
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	type testCase struct {
+		deployment     *appsv1.Deployment
+		expectNotFound bool
+	}
+	testCases := map[string]testCase{
+		"GetDeployment(...):": {
+			deployment: &appsv1.Deployment{
+				ObjectMeta: metav1.ObjectMeta{
+					Name:      "test",
+					Namespace: "default",
+				},
+			},
+		},
+		"GetDeployment(...): not found": {
+			deployment: &appsv1.Deployment{
+				ObjectMeta: metav1.ObjectMeta{
+					Name:      "test",
+					Namespace: "default",
+				},
+			},
+			expectNotFound: true,
+		},
+	}
+	for testName, testCase := range testCases {
+		c.Logf("testing kubernetes.%v", testName)
+
+		kubeClient := fake.NewSimpleClientset()
+
+		if !testCase.expectNotFound {
+			_, err := kubeClient.AppsV1().Deployments(testCase.deployment.Namespace).Create(ctx, testCase.deployment, metav1.CreateOptions{})
+			c.Assert(err, IsNil, Commentf(test.ErrErrorFmt, testName))
+		}
+
+		deployment, err := GetDeployment(kubeClient, testCase.deployment.Namespace, testCase.deployment.Name)
+		if testCase.expectNotFound {
+			c.Assert(apierrors.IsNotFound(err), Equals, true, Commentf(test.ErrResultFmt, testName))
+			continue
+		}
+		c.Assert(err, IsNil, Commentf(test.ErrErrorFmt, testName))
+		c.Assert(deployment.Name, Equals, testCase.deployment.Name, Commentf(test.ErrResultFmt, testName))
+	}
+}