Added unit test.

pkg/rbd/controllerserver_test.go

@@ -0,0 +1,81 @@
+package rbd
+
+import (
+	"testing"
+
+	"github.com/ceph/ceph-csi/pkg/util"
+)
+
+type testCachePersister struct {
+	volumes   map[string]rbdVolume
+	snapshots map[string]rbdSnapshot
+}
+
+func (t *testCachePersister) Create(identifier string, data interface{}) error {
+	return nil
+}
+
+func (t *testCachePersister) Get(identifier string, data interface{}) error {
+	return nil
+}
+
+func (t *testCachePersister) ForAll(pattern string, destObj interface{}, f util.ForAllFunc) error {
+
+	switch pattern {
+	case "csi-rbd-vol-":
+		for identifier, vol := range t.volumes {
+			*destObj.(*rbdVolume) = vol
+			if err := f(identifier); err != nil {
+				return err
+			}
+		}
+	case "csi-rbd-(.*)-snap-":
+		for identifier, snap := range t.snapshots {
+			*destObj.(*rbdSnapshot) = snap
+			if err := f(identifier); err != nil {
+				return err
+			}
+		}
+	}
+
+	return nil
+}
+
+func (t *testCachePersister) Delete(identifier string) error {
+	return nil
+}
+
+func TestLoadExDataFromMetadataStore(t *testing.T) {
+	cs := &ControllerServer{
+		MetadataStore: &testCachePersister{
+			volumes: map[string]rbdVolume{
+				"item1": rbdVolume{
+					VolID: "1",
+				},
+				"item2": rbdVolume{
+					VolID: "2",
+				},
+			},
+			snapshots: map[string]rbdSnapshot{
+				"item1": rbdSnapshot{
+					SnapID: "1",
+				},
+				"item2": rbdSnapshot{
+					SnapID: "2",
+				},
+			},
+		},
+	}
+
+	if err := cs.LoadExDataFromMetadataStore(); err != nil {
+		t.Error(err)
+	}
+
+	if rbdVolumes["item1"] == rbdVolumes["item2"] {
+		t.Error("rbd volume entries contain pointer to same volume")
+	}
+
+	if rbdSnapshots["item1"] == rbdSnapshots["item2"] {
+		t.Error("rbd snapshot entries contain pointer to same snapshot")
+	}
+}