test: GetDeployment

longhorn/longhorn-9752

Signed-off-by: Chin-Ya Huang <[email protected]>

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))
+		}
+
+		daemonSet, err := GetDeployment(kubeClient, testCase.deployment.Namespace, testCase.deployment.Name)
+		if testCase.expectNotFound {
+			c.Assert(apierrors.IsNotFound(err), Equals, true, Commentf(test.ErrResultFmt, testName))
+			return
+		}
+		c.Assert(err, IsNil, Commentf(test.ErrErrorFmt, testName))
+		c.Assert(daemonSet.Name, Equals, testCase.deployment.Name, Commentf(test.ErrResultFmt, testName))
+	}
+}