Adding fake tests for snapshot_inspector.go

kastenhq · Jul 29, 2024 · 3a2d883 · 3a2d883
1 parent e8ad4f8
commit 3a2d883
Showing 1 changed file with 209 additions and 0 deletions.
diff --git a/pkg/csi/snapshot_inspector_test.go b/pkg/csi/snapshot_inspector_test.go
@@ -0,0 +1,209 @@
+package csi
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/golang/mock/gomock"
+	"github.com/kastenhq/kubestr/pkg/csi/mocks"
+	"github.com/kastenhq/kubestr/pkg/csi/types"
+	snapv1 "github.com/kubernetes-csi/external-snapshotter/client/v4/apis/volumesnapshot/v1"
+	. "gopkg.in/check.v1"
+	v1 "k8s.io/api/core/v1"
+	sv1 "k8s.io/api/storage/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/client-go/dynamic"
+	fakedynamic "k8s.io/client-go/dynamic/fake"
+	"k8s.io/client-go/kubernetes"
+	"k8s.io/client-go/kubernetes/fake"
+)
+
+func (s *CSITestSuite) TestRunSnapshotBrowseHelper(c *C) {
+	ctx := context.Background()
+	type fields struct {
+		stepperOps *mocks.MockSnapshotBrowserStepper
+	}
+	for _, tc := range []struct {
+		kubeCli    kubernetes.Interface
+		dynCli     dynamic.Interface
+		args       *types.SnapshotBrowseArgs
+		prepare    func(f *fields)
+		errChecker Checker
+	}{
+		{
+			// success
+			kubeCli: fake.NewSimpleClientset(),
+			dynCli:  fakedynamic.NewSimpleDynamicClient(runtime.NewScheme()),
+			args:    &types.SnapshotBrowseArgs{},
+			prepare: func(f *fields) {
+				gomock.InOrder(
+					f.stepperOps.EXPECT().ValidateArgs(gomock.Any(), gomock.Any()).Return(
+						&sv1.StorageClass{}, nil,
+					),
+					f.stepperOps.EXPECT().FetchVS(gomock.Any(), gomock.Any()).Return(
+						&snapv1.VolumeSnapshot{
+							ObjectMeta: metav1.ObjectMeta{
+								Name:      "snap1",
+								Namespace: "ns",
+							},
+						},
+						nil,
+					),
+					f.stepperOps.EXPECT().CreateInspectorApplication(gomock.Any(), gomock.Any(),
+						&snapv1.VolumeSnapshot{
+							ObjectMeta: metav1.ObjectMeta{
+								Name:      "snap1",
+								Namespace: "ns",
+							},
+						}, &sv1.StorageClass{},
+					).Return(
+						&v1.Pod{
+							ObjectMeta: metav1.ObjectMeta{
+								Name:      "pod1",
+								Namespace: "ns",
+							},
+						},
+						&v1.PersistentVolumeClaim{
+							ObjectMeta: metav1.ObjectMeta{
+								Name:      "pvc1",
+								Namespace: "ns",
+							},
+						},
+						nil,
+					),
+					f.stepperOps.EXPECT().PortForwardAPod(gomock.Any(),
+						&v1.Pod{
+							ObjectMeta: metav1.ObjectMeta{
+								Name:      "pod1",
+								Namespace: "ns",
+							},
+						}, gomock.Any(),
+					).Return(nil),
+					f.stepperOps.EXPECT().Cleanup(gomock.Any(),
+						&v1.PersistentVolumeClaim{
+							ObjectMeta: metav1.ObjectMeta{
+								Name:      "pvc1",
+								Namespace: "ns",
+							},
+						},
+						&v1.Pod{
+							ObjectMeta: metav1.ObjectMeta{
+								Name:      "pod1",
+								Namespace: "ns",
+							},
+						},
+					),
+				)
+			},
+			errChecker: IsNil,
+		},
+		{
+			// portforward failure
+			kubeCli: fake.NewSimpleClientset(),
+			dynCli:  fakedynamic.NewSimpleDynamicClient(runtime.NewScheme()),
+			args:    &types.SnapshotBrowseArgs{},
+			prepare: func(f *fields) {
+				gomock.InOrder(
+					f.stepperOps.EXPECT().ValidateArgs(gomock.Any(), gomock.Any()).Return(nil, nil),
+					f.stepperOps.EXPECT().FetchVS(gomock.Any(), gomock.Any()).Return(nil, nil),
+					f.stepperOps.EXPECT().CreateInspectorApplication(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, nil, nil),
+					f.stepperOps.EXPECT().PortForwardAPod(gomock.Any(), gomock.Any(), gomock.Any()).Return(fmt.Errorf("portforward error")),
+					f.stepperOps.EXPECT().Cleanup(gomock.Any(), gomock.Any(), gomock.Any()),
+				)
+			},
+			errChecker: NotNil,
+		},
+		{
+			// createapp failure
+			kubeCli: fake.NewSimpleClientset(),
+			dynCli:  fakedynamic.NewSimpleDynamicClient(runtime.NewScheme()),
+			args:    &types.SnapshotBrowseArgs{},
+			prepare: func(f *fields) {
+				gomock.InOrder(
+					f.stepperOps.EXPECT().ValidateArgs(gomock.Any(), gomock.Any()).Return(nil, nil),
+					f.stepperOps.EXPECT().FetchVS(gomock.Any(), gomock.Any()).Return(nil, nil),
+					f.stepperOps.EXPECT().CreateInspectorApplication(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, nil, fmt.Errorf("createapp error")),
+					f.stepperOps.EXPECT().Cleanup(gomock.Any(), gomock.Any(), gomock.Any()),
+				)
+			},
+			errChecker: NotNil,
+		},
+		{
+			// fetch snapshot failure
+			kubeCli: fake.NewSimpleClientset(),
+			dynCli:  fakedynamic.NewSimpleDynamicClient(runtime.NewScheme()),
+			args:    &types.SnapshotBrowseArgs{},
+			prepare: func(f *fields) {
+				gomock.InOrder(
+					f.stepperOps.EXPECT().ValidateArgs(gomock.Any(), gomock.Any()).Return(nil, nil),
+					f.stepperOps.EXPECT().FetchVS(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf("snapshot error")),
+					f.stepperOps.EXPECT().Cleanup(gomock.Any(), gomock.Any(), gomock.Any()),
+				)
+			},
+			errChecker: NotNil,
+		},
+		{
+			// validate failure
+			kubeCli: fake.NewSimpleClientset(),
+			dynCli:  fakedynamic.NewSimpleDynamicClient(runtime.NewScheme()),
+			args:    &types.SnapshotBrowseArgs{},
+			prepare: func(f *fields) {
+				gomock.InOrder(
+					f.stepperOps.EXPECT().ValidateArgs(gomock.Any(), gomock.Any()).Return(nil, fmt.Errorf("snapshot error")),
+					f.stepperOps.EXPECT().Cleanup(gomock.Any(), gomock.Any(), gomock.Any()),
+				)
+			},
+			errChecker: NotNil,
+		},
+		{
+			// emptycli failure
+			kubeCli: nil,
+			dynCli:  fakedynamic.NewSimpleDynamicClient(runtime.NewScheme()),
+			args:    &types.SnapshotBrowseArgs{},
+			prepare: func(f *fields) {
+				gomock.InOrder(
+					f.stepperOps.EXPECT().Cleanup(gomock.Any(), gomock.Any(), gomock.Any()),
+				)
+			},
+			errChecker: NotNil,
+		},
+		{
+			// emptydyncli failure
+			kubeCli: fake.NewSimpleClientset(),
+			dynCli:  nil,
+			args:    &types.SnapshotBrowseArgs{},
+			prepare: func(f *fields) {
+				gomock.InOrder(
+					f.stepperOps.EXPECT().Cleanup(gomock.Any(), gomock.Any(), gomock.Any()),
+				)
+			},
+			errChecker: NotNil,
+		},
+	} {
+		ctrl := gomock.NewController(c)
+		defer ctrl.Finish()
+		f := fields{
+			stepperOps: mocks.NewMockSnapshotBrowserStepper(ctrl),
+		}
+		if tc.prepare != nil {
+			tc.prepare(&f)
+		}
+		runner := &SnapshotBrowseRunner{
+			KubeCli:      tc.kubeCli,
+			DynCli:       tc.dynCli,
+			browserSteps: f.stepperOps,
+		}
+		err := runner.RunSnapshotBrowseHelper(ctx, tc.args)
+		c.Check(err, tc.errChecker)
+	}
+}
+
+func (s *CSITestSuite) TestSnapshotBrowseRunner(c *C) {
+	ctx := context.Background()
+	r := &SnapshotBrowseRunner{
+		browserSteps: &snapshotBrowserSteps{},
+	}
+	err := r.RunSnapshotBrowseHelper(ctx, nil)
+	c.Check(err, NotNil)
+}