*.: Add more fake test

Author: anywhy

pkg/controller/pv_control.go

@@ -9,8 +9,10 @@ import (
 	apierrs "k8s.io/apimachinery/pkg/api/errors"
 	"k8s.io/apimachinery/pkg/types"
 	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
+	coreinformers "k8s.io/client-go/informers/core/v1"
 	"k8s.io/client-go/kubernetes"
 	corelisters "k8s.io/client-go/listers/core/v1"
+	"k8s.io/client-go/tools/cache"
 	"k8s.io/client-go/tools/record"
 	"k8s.io/client-go/util/retry"
 
@@ -148,3 +150,68 @@ func (rpc *realPVControl) recordPVEvent(verb string, tc *v1alpha1.Redis, pvName
 }
 
 var _ PVControlInterface = &realPVControl{}
+
+// FakePVControl is a fake PVControlInterface
+type FakePVControl struct {
+	PVCLister       corelisters.PersistentVolumeClaimLister
+	PVIndexer       cache.Indexer
+	updatePVTracker requestTracker
+}
+
+// NewFakePVControl returns a FakePVControl
+func NewFakePVControl(pvInformer coreinformers.PersistentVolumeInformer, pvcInformer coreinformers.PersistentVolumeClaimInformer) *FakePVControl {
+	return &FakePVControl{
+		pvcInformer.Lister(),
+		pvInformer.Informer().GetIndexer(),
+		requestTracker{0, nil, 0},
+	}
+}
+
+// SetUpdatePVError sets the error attributes of updatePVTracker
+func (fpc *FakePVControl) SetUpdatePVError(err error, after int) {
+	fpc.updatePVTracker.err = err
+	fpc.updatePVTracker.after = after
+}
+
+// PatchPVReclaimPolicy patchs the reclaim policy of PV
+func (fpc *FakePVControl) PatchPVReclaimPolicy(_ *v1alpha1.Redis, pv *corev1.PersistentVolume, reclaimPolicy corev1.PersistentVolumeReclaimPolicy) error {
+	defer fpc.updatePVTracker.inc()
+	if fpc.updatePVTracker.errorReady() {
+		defer fpc.updatePVTracker.reset()
+		return fpc.updatePVTracker.err
+	}
+	pv.Spec.PersistentVolumeReclaimPolicy = reclaimPolicy
+
+	return fpc.PVIndexer.Update(pv)
+}
+
+// UpdatePV update a pv in a Redis.
+func (fpc *FakePVControl) UpdatePV(rc *v1alpha1.Redis, pv *corev1.PersistentVolume) (*corev1.PersistentVolume, error) {
+	ns := rc.GetNamespace()
+	if pv.Labels == nil {
+		pv.Labels = make(map[string]string)
+	}
+	if pv.Annotations == nil {
+		pv.Annotations = make(map[string]string)
+	}
+
+	pvcName := pv.Spec.ClaimRef.Name
+	pvc, err := fpc.PVCLister.PersistentVolumeClaims(ns).Get(pvcName)
+	if err != nil {
+		return nil, err
+	}
+
+	component := pvc.Labels[label.ComponentLabelKey]
+	podName := pvc.Annotations[label.AnnPodNameKey]
+	pv.Labels[label.NamespaceLabelKey] = ns
+	pv.Labels[label.ComponentLabelKey] = component
+	pv.Labels[label.NameLabelKey] = pvc.Labels[label.NameLabelKey]
+	pv.Labels[label.ManagedByLabelKey] = pvc.Labels[label.ManagedByLabelKey]
+	pv.Labels[label.InstanceLabelKey] = pvc.Labels[label.InstanceLabelKey]
+
+	setIfNotEmpty(pv.Annotations, label.AnnPodNameKey, podName)
+
+	return pv, fpc.PVIndexer.Update(pv)
+}
+
+var _ PVControlInterface = &FakePVControl{}

pkg/controller/pvc_control.go

@@ -7,8 +7,10 @@ import (
 	"github.com/golang/glog"
 	corev1 "k8s.io/api/core/v1"
 	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
+	coreinformers "k8s.io/client-go/informers/core/v1"
 	"k8s.io/client-go/kubernetes"
 	corelisters "k8s.io/client-go/listers/core/v1"
+	"k8s.io/client-go/tools/cache"
 	"k8s.io/client-go/tools/record"
 	"k8s.io/client-go/util/retry"
 
@@ -117,3 +119,65 @@ func (rpc *realPVCControl) recordPVCEvent(verb string, rc *v1alpha1.Redis, pvcNa
 }
 
 var _ PVCControlInterface = &realPVCControl{}
+
+// FakePVCControl is a fake PVCControlInterface
+type FakePVCControl struct {
+	PVCIndexer       cache.Indexer
+	updatePVCTracker requestTracker
+	deletePVCTracker requestTracker
+}
+
+// NewFakePVCControl returns a FakePVCControl
+func NewFakePVCControl(pvcInformer coreinformers.PersistentVolumeClaimInformer) *FakePVCControl {
+	return &FakePVCControl{
+		pvcInformer.Informer().GetIndexer(),
+		requestTracker{0, nil, 0},
+		requestTracker{0, nil, 0},
+	}
+}
+
+// SetUpdatePVCError sets the error attributes of updatePVCTracker
+func (fpc *FakePVCControl) SetUpdatePVCError(err error, after int) {
+	fpc.updatePVCTracker.err = err
+	fpc.updatePVCTracker.after = after
+}
+
+// SetDeletePVCError sets the error attributes of deletePVCTracker
+func (fpc *FakePVCControl) SetDeletePVCError(err error, after int) {
+	fpc.deletePVCTracker.err = err
+	fpc.deletePVCTracker.after = after
+}
+
+// DeletePVC deletes the pvc
+func (fpc *FakePVCControl) DeletePVC(_ *v1alpha1.Redis, pvc *corev1.PersistentVolumeClaim) error {
+	defer fpc.deletePVCTracker.inc()
+	if fpc.deletePVCTracker.errorReady() {
+		defer fpc.deletePVCTracker.reset()
+		return fpc.deletePVCTracker.err
+	}
+
+	return fpc.PVCIndexer.Delete(pvc)
+}
+
+// UpdatePVC updates the annotation, labels and spec of pvc
+func (fpc *FakePVCControl) UpdatePVC(rc *v1alpha1.Redis, pvc *corev1.PersistentVolumeClaim, pod *corev1.Pod) (*corev1.PersistentVolumeClaim, error) {
+	defer fpc.updatePVCTracker.inc()
+	if fpc.updatePVCTracker.errorReady() {
+		defer fpc.updatePVCTracker.reset()
+		return nil, fpc.updatePVCTracker.err
+	}
+
+	if pvc.Annotations == nil {
+		pvc.Annotations = make(map[string]string)
+	}
+	if pvc.Labels == nil {
+		pvc.Labels = make(map[string]string)
+	}
+
+	podName := pod.GetName()
+	setIfNotEmpty(pvc.Annotations, label.AnnPodNameKey, podName)
+
+	return pvc, fpc.PVCIndexer.Update(pvc)
+}
+
+var _ PVCControlInterface = &FakePVCControl{}