Add CSI info test for agent

Signed-off-by: Ameya Gawde <agawde@mirantis.com>

agent/agent_test.go

@@ -15,6 +15,7 @@ import (
 	"google.golang.org/grpc/credentials"
 
 	events "github.com/docker/go-events"
+	"github.com/docker/swarmkit/agent/csi"
 	agentutils "github.com/docker/swarmkit/agent/testutils"
 	"github.com/docker/swarmkit/api"
 	"github.com/docker/swarmkit/ca"
@@ -437,6 +438,25 @@ func TestAgentExitsBasedOnSessionTracker(t *testing.T) {
 	require.Len(t, closedSessions, 3)
 }
 
+func TestAgentCSIInfo(t *testing.T) {
+	tlsCh := make(chan events.Event, 1)
+	defer close(tlsCh)
+	tester := agentTestEnv(t, nil, tlsCh)
+	fakeNodePlugin := &csi.NodePlugin{
+		Name:   "testDriver",
+		NodeID: "node1",
+	}
+	tester.agent.CSIPlugins = append(tester.agent.CSIPlugins, fakeNodePlugin)
+	defer tester.cleanup()
+	defer tester.StartAgent(t)
+
+	currSession, closedSessions := tester.dispatcher.GetSessions()
+	require.NotNil(t, currSession)
+	require.NotNil(t, currSession.Description)
+	require.NotNil(t, currSession.Description.CSIInfo)
+	require.Equal(t, currSession.Description.CSIInfo[0].NodeID, fakeNodePlugin.NodeID)
+}
+
 // If we pass a session tracker, established sessions get tracked.
 func TestAgentRegistersSessionsWithSessionTracker(t *testing.T) {
 	tlsCh := make(chan events.Event, 1)