Address comments

Signed-off-by: Serguei Bezverkhi <[email protected]>

Author: sbezverk

cmd/livenessprobe_test.go

@@ -71,6 +71,14 @@ func TestProbe(t *testing.T) {
 	defer csiConn.Close()
 
 	var injectedErr error
+
+	// Setting up expected calls' responses
+	inPlugin := &csi.GetPluginInfoRequest{}
+	outPlugin := &csi.GetPluginInfoResponse{
+		Name: "foo/bar",
+	}
+	idServer.EXPECT().GetPluginInfo(gomock.Any(), inPlugin).Return(outPlugin, injectedErr).Times(1)
+
 	inProbe := &csi.ProbeRequest{}
 	outProbe := &csi.ProbeResponse{}
 	idServer.EXPECT().Probe(gomock.Any(), inProbe).Return(outProbe, injectedErr).Times(1)

cmd/main.go

@@ -42,6 +42,13 @@ var (
 )
 
 func runProbe(ctx context.Context, csiConn connection.CSIConnection) error {
+	// Get CSI driver name.
+	glog.Infof("Calling CSI driver to discover driver name.")
+	csiDriverName, err := csiConn.GetDriverName(ctx)
+	if err != nil {
+		return err
+	}
+	glog.Infof("CSI driver name: %q", csiDriverName)
 	// Sending Probe request
 	glog.Infof("Sending probe request to CSI driver.")
 	if err := csiConn.LivenessProbe(ctx); err != nil {

pkg/connection/connection.go

@@ -36,9 +36,6 @@ type CSIConnection interface {
 	// call.
 	GetDriverName(ctx context.Context) (string, error)
 
-	// NodeGetId returns node ID of the current according to the CSI driver.
-	NodeGetId(ctx context.Context) (string, error)
-
 	// Liveness Probe
 	LivenessProbe(ctx context.Context) error
 
@@ -126,22 +123,6 @@ func (c *csiConnection) LivenessProbe(ctx context.Context) error {
 	return nil
 }
 
-func (c *csiConnection) NodeGetId(ctx context.Context) (string, error) {
-	client := csi.NewNodeClient(c.conn)
-
-	req := csi.NodeGetIdRequest{}
-
-	rsp, err := client.NodeGetId(ctx, &req)
-	if err != nil {
-		return "", err
-	}
-	nodeID := rsp.GetNodeId()
-	if nodeID == "" {
-		return "", fmt.Errorf("node ID is empty")
-	}
-	return nodeID, nil
-}
-
 func (c *csiConnection) Close() error {
 	return c.conn.Close()
 }