From 2f5fbd3d813d0e9b3f8f8b7e4bce6bbcd26e36ba Mon Sep 17 00:00:00 2001
From: Pekka Nurmi <pekka.nurmi@upcloud.com>
Date: Fri, 14 Jul 2023 09:22:03 +0300
Subject: [PATCH] feat(identity): add tests (#83)

---
 internal/identity/identity_test.go | 69 ++++++++++++++++++++++++++++++
 1 file changed, 69 insertions(+)
 create mode 100644 internal/identity/identity_test.go

diff --git a/internal/identity/identity_test.go b/internal/identity/identity_test.go
new file mode 100644
index 0000000..6829598
--- /dev/null
+++ b/internal/identity/identity_test.go
@@ -0,0 +1,69 @@
+package identity_test
+
+import (
+	"context"
+	"testing"
+
+	"github.com/UpCloudLtd/upcloud-csi/internal/identity"
+	"github.com/UpCloudLtd/upcloud-csi/internal/logger"
+	"github.com/container-storage-interface/spec/lib/go/csi"
+	"github.com/stretchr/testify/require"
+)
+
+func TestIdentity_GetPluginInfo(t *testing.T) {
+	t.Parallel()
+
+	l := logger.New("error")
+	id := identity.NewIdentity("test", l.WithField("package", "identity_test"))
+	want := csi.GetPluginInfoResponse{
+		Name: "test",
+	}
+	got, err := id.GetPluginInfo(context.TODO(), nil)
+	require.NoError(t, err)
+	require.Equal(t, want, *got)
+}
+
+func TestIdentity_GetPluginCapabilities(t *testing.T) {
+	t.Parallel()
+
+	l := logger.New("error")
+	id := identity.NewIdentity("test", l.WithField("package", "identity_test"))
+	want := csi.GetPluginCapabilitiesResponse{
+		Capabilities: []*csi.PluginCapability{
+			{
+				Type: &csi.PluginCapability_Service_{
+					Service: &csi.PluginCapability_Service{
+						Type: csi.PluginCapability_Service_CONTROLLER_SERVICE,
+					},
+				},
+			},
+			{
+				Type: &csi.PluginCapability_Service_{
+					Service: &csi.PluginCapability_Service{
+						Type: csi.PluginCapability_Service_VOLUME_ACCESSIBILITY_CONSTRAINTS,
+					},
+				},
+			},
+			{
+				Type: &csi.PluginCapability_VolumeExpansion_{
+					VolumeExpansion: &csi.PluginCapability_VolumeExpansion{
+						Type: csi.PluginCapability_VolumeExpansion_OFFLINE,
+					},
+				},
+			},
+		},
+	}
+	got, err := id.GetPluginCapabilities(context.TODO(), nil)
+	require.NoError(t, err)
+	require.Equal(t, want, *got)
+}
+
+func TestIdentity_Probe(t *testing.T) {
+	t.Parallel()
+
+	l := logger.New("error")
+	id := identity.NewIdentity("test", l.WithField("package", "identity_test"))
+	got, err := id.Probe(context.TODO(), nil)
+	require.NoError(t, err)
+	require.True(t, got.Ready.Value)
+}