Add CSI info test for agent

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

agent/agent_test.go

@@ -437,6 +437,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.CSIInfo = append(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)