From 3086c72650347901bb46090eb4e8e95d0d15b32c Mon Sep 17 00:00:00 2001 From: Danielle Lancashire Date: Thu, 23 Jan 2020 12:20:36 +0100 Subject: [PATCH] csi: Add csi.NodeStageVolume to the NodeClient Implements a fake version of NodeStageVolume as a dependency of implementing the client.NodeStageVolume request --- plugins/csi/client.go | 2 ++ plugins/csi/testing/client.go | 5 +++++ 2 files changed, 7 insertions(+) diff --git a/plugins/csi/client.go b/plugins/csi/client.go index 556f9142a..08add6719 100644 --- a/plugins/csi/client.go +++ b/plugins/csi/client.go @@ -67,6 +67,8 @@ type CSIControllerClient interface { type CSINodeClient interface { NodeGetCapabilities(ctx context.Context, in *csipbv1.NodeGetCapabilitiesRequest, opts ...grpc.CallOption) (*csipbv1.NodeGetCapabilitiesResponse, error) NodeGetInfo(ctx context.Context, in *csipbv1.NodeGetInfoRequest, opts ...grpc.CallOption) (*csipbv1.NodeGetInfoResponse, error) + NodeStageVolume(ctx context.Context, in *csipbv1.NodeStageVolumeRequest, opts ...grpc.CallOption) (*csipbv1.NodeStageVolumeResponse, error) + // NodeUnstageVolume(ctx context.Context, in *NodeUnstageVolumeRequest, opts ...grpc.CallOption) (*NodeUnstageVolumeResponse, error) } type client struct { diff --git a/plugins/csi/testing/client.go b/plugins/csi/testing/client.go index 75f20929d..f341634a8 100644 --- a/plugins/csi/testing/client.go +++ b/plugins/csi/testing/client.go @@ -83,6 +83,7 @@ type NodeClient struct { NextErr error NextCapabilitiesResponse *csipbv1.NodeGetCapabilitiesResponse NextGetInfoResponse *csipbv1.NodeGetInfoResponse + NextStageVolumeResponse *csipbv1.NodeStageVolumeResponse } // NewNodeClient returns a new ControllerClient @@ -103,3 +104,7 @@ func (c *NodeClient) NodeGetCapabilities(ctx context.Context, in *csipbv1.NodeGe func (c *NodeClient) NodeGetInfo(ctx context.Context, in *csipbv1.NodeGetInfoRequest, opts ...grpc.CallOption) (*csipbv1.NodeGetInfoResponse, error) { return c.NextGetInfoResponse, c.NextErr } + +func (c *NodeClient) NodeStageVolume(ctx context.Context, in *csipbv1.NodeStageVolumeRequest, opts ...grpc.CallOption) (*csipbv1.NodeStageVolumeResponse, error) { + return c.NextStageVolumeResponse, c.NextErr +}