feat: check CoreDNS

kubernetes/deployment.go

@@ -0,0 +1,23 @@
+package kubernetes
+
+import (
+	"context"
+
+	"github.com/sirupsen/logrus"
+
+	appsv1 "k8s.io/api/apps/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	kubeclient "k8s.io/client-go/kubernetes"
+)
+
+// GetDeployment returns the Deployment with the given name in the given namespace.
+func GetDeployment(kubeClient kubeclient.Interface, namespace, name string) (*appsv1.Deployment, error) {
+	log := logrus.WithFields(logrus.Fields{
+		"kind":      "Deployment",
+		"namespace": namespace,
+		"name":      name,
+	})
+	log.Trace("Getting resource")
+
+	return kubeClient.AppsV1().Deployments(namespace).Get(context.Background(), name, metav1.GetOptions{})
+}

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))
+	}
+}