K8SPS-339: add dataSource and dataSourceRef fields (#998)

* K8SPS-339: add dataSource and dataSourceRef fields

https://perconadev.atlassian.net/browse/K8SPS-339

* update unit-test

---------

Co-authored-by: George Kechagias <[email protected]>
Co-authored-by: Viacheslav Sarzhan <[email protected]>

Author: 3 people

deploy/cr.yaml

@@ -133,6 +133,15 @@ spec:
 #        path: /data
 #        type: Directory
       persistentVolumeClaim:
+#        dataSource:
+#          apiGroup: ""
+#          kind: PersistentVolumeClaim
+#          name: source-pvc
+#        dataSourceRef:
+#          apiGroup: ""
+#          kind: PersistentVolumeClaim
+#          name: source-pvc
+#          namespace: default
 #        storageClassName: standard
 #        accessModes: [ "ReadWriteOnce" ]
         resources:

pkg/k8s/pvc.go

@@ -16,6 +16,8 @@ func PVC(name string, spec *apiv1alpha1.VolumeSpec) corev1.PersistentVolumeClaim
 			StorageClassName: spec.PersistentVolumeClaim.StorageClassName,
 			AccessModes:      spec.PersistentVolumeClaim.AccessModes,
 			Resources:        spec.PersistentVolumeClaim.Resources,
+			DataSource:       spec.PersistentVolumeClaim.DataSource,
+			DataSourceRef:    spec.PersistentVolumeClaim.DataSourceRef,
 		},
 	}
 }

pkg/k8s/pvc_test.go

@@ -3,11 +3,12 @@ package k8s
 import (
 	"testing"
 
-	apiv1alpha1 "github.com/percona/percona-server-mysql-operator/api/v1alpha1"
-
 	"github.com/stretchr/testify/assert"
 	corev1 "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/resource"
+	"k8s.io/utils/ptr"
+
+	apiv1alpha1 "github.com/percona/percona-server-mysql-operator/api/v1alpha1"
 )
 
 func TestPVC(t *testing.T) {
@@ -16,6 +17,17 @@ func TestPVC(t *testing.T) {
 	requests := corev1.ResourceList{
 		corev1.ResourceStorage: resource.MustParse("2Gi"),
 	}
+	dataSource := &corev1.TypedLocalObjectReference{
+		APIGroup: ptr.To("some-api-group"),
+		Kind:     "some-kind",
+		Name:     "some-name",
+	}
+	dataSourceRef := &corev1.TypedObjectReference{
+		APIGroup:  ptr.To("some-api-group-2"),
+		Kind:      "some-kind-2",
+		Name:      "some-name-2",
+		Namespace: ptr.To("namespace"),
+	}
 
 	spec := &apiv1alpha1.VolumeSpec{
 		PersistentVolumeClaim: &corev1.PersistentVolumeClaimSpec{
@@ -24,6 +36,8 @@ func TestPVC(t *testing.T) {
 			Resources: corev1.VolumeResourceRequirements{
 				Requests: requests,
 			},
+			DataSource:    dataSource,
+			DataSourceRef: dataSourceRef,
 		},
 	}
 
@@ -36,4 +50,6 @@ func TestPVC(t *testing.T) {
 	assert.Equal(t, storageClassName, *pvc.Spec.StorageClassName)
 	assert.Equal(t, accessModes, pvc.Spec.AccessModes)
 	assert.Equal(t, requests, pvc.Spec.Resources.Requests)
+	assert.Equal(t, dataSource, pvc.Spec.DataSource)
+	assert.Equal(t, dataSourceRef, pvc.Spec.DataSourceRef)
 }