wip

Signed-off-by: Yuji Ito <llamerada.jp@gmail.com>

Author: llamerada-jp

Makefile

@@ -81,6 +81,7 @@ vet: ## Run go vet against code.
 
 .PHONY: mock
 mock: mockgen
+	$(MOCKGEN) -source=internal/ceph/ceph.go -destination=internal/ceph/ceph_mock.go -package=ceph
 	$(MOCKGEN) -source=internal/ceph/command.go -destination=internal/ceph/command_mock.go -package=ceph
 	$(MOCKGEN) -source=pkg/controller/proto/controller_grpc.pb.go -destination=pkg/controller/proto/controller_grpc.pb_mock.go -package=proto
 	$(MOCKGEN) -source=internal/controller/internal/objectstorage/objectstorage.go -destination=internal/controller/internal/objectstorage/objectstorage_mock.go -package=objectstorage

internal/ceph/ceph_mock.go

@@ -1,15 +1,61 @@
 package controller
 
 import (
+	"github.com/cybozu-go/mantle/internal/ceph"
 	"github.com/cybozu-go/mantle/test/util"
 	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
+	"go.uber.org/mock/gomock"
 	corev1 "k8s.io/api/core/v1"
 	storagev1 "k8s.io/api/storage/v1"
 	"k8s.io/apimachinery/pkg/api/resource"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 )
 
+var _ = Describe("util.createCloneByPV", func() {
+	DescribeTable("matrix test",
+		func(ctx SpecContext,
+			pv *corev1.PersistentVolume, snapshotName, cloneName string, // arguments
+			lsIsCalled bool, lsPool string, images []string, // for RBDLs
+			infoIsCalled bool, infoPool, infoImage string, info *ceph.RBDImageInfo, // for RBDInfo
+			cloneIsCalled bool, clonePool, cloneImage, cloneFeatures string, // for RBDClone
+			expectedErr bool) {
+			ctrl := gomock.NewController(GinkgoT())
+			defer ctrl.Finish()
+
+			cmdMock := ceph.NewMockCephCmd(ctrl)
+			if lsIsCalled {
+				cmdMock.EXPECT().RBDLs(lsPool).Return(images, nil)
+			}
+			if infoIsCalled {
+				cmdMock.EXPECT().RBDInfo(infoPool, infoImage).Return(info, nil)
+			}
+			if cloneIsCalled {
+				cmdMock.EXPECT().RBDClone(clonePool, cloneImage, snapshotName, cloneName, cloneFeatures).Return(nil)
+			}
+			createCloneByPV(ctx, cmdMock, pv, snapshotName, cloneName)
+		},
+		Entry("clone does not exist, create clone",
+			&corev1.PersistentVolume{
+				Spec: corev1.PersistentVolumeSpec{
+					PersistentVolumeSource: corev1.PersistentVolumeSource{
+						CSI: &corev1.CSIPersistentVolumeSource{
+							VolumeAttributes: map[string]string{
+								"imageName": "bkImage",
+								"pool":      "bkPool",
+							},
+						},
+					},
+				},
+			},
+			true, "bkPool", []string{"bkImage"},
+			false, "", "", nil,
+			true, "bkPool", "bkImage", "deep-flatten",
+			false,
+		),
+	)
+})
+
 var _ = Describe("util.getCephClusterIDFromPVC", func() {
 	var namespace string
 	storageClassName := util.GetUniqueName("sc-")