From d8458293d2f36b802eb98fae43722cdbee3d89c0 Mon Sep 17 00:00:00 2001 From: Humble Chirammal Date: Mon, 27 May 2024 19:46:26 +0530 Subject: [PATCH] add controllerModifyVolume test function Signed-off-by: Humble Chirammal --- .../drivers/csi-test/driver/driver.mock.go | 8 ++++++++ .../drivers/csi-test/mock/service/controller.go | 15 +++++++++++++++ 2 files changed, 23 insertions(+) diff --git a/test/e2e/storage/drivers/csi-test/driver/driver.mock.go b/test/e2e/storage/drivers/csi-test/driver/driver.mock.go index 06d3470ceaf..1499fd8467c 100644 --- a/test/e2e/storage/drivers/csi-test/driver/driver.mock.go +++ b/test/e2e/storage/drivers/csi-test/driver/driver.mock.go @@ -319,6 +319,14 @@ func (mr *MockControllerServerMockRecorder) ValidateVolumeCapabilities(arg0, arg return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateVolumeCapabilities", reflect.TypeOf((*MockControllerServer)(nil).ValidateVolumeCapabilities), arg0, arg1) } +func (m *MockControllerServer) ControllerModifyVolume(arg0 context.Context, arg1 *csi.ControllerModifyVolumeRequest) (*csi.ControllerModifyVolumeResponse, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ModifyVolume", arg0, arg1) + ret0, _ := ret[0].(*csi.ControllerModifyVolumeResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + // MockNodeServer is a mock of NodeServer interface. type MockNodeServer struct { ctrl *gomock.Controller diff --git a/test/e2e/storage/drivers/csi-test/mock/service/controller.go b/test/e2e/storage/drivers/csi-test/mock/service/controller.go index ee1465077e6..33700104c25 100644 --- a/test/e2e/storage/drivers/csi-test/mock/service/controller.go +++ b/test/e2e/storage/drivers/csi-test/mock/service/controller.go @@ -567,6 +567,13 @@ func (s *service) ControllerGetCapabilities( }, }, }, + { + Type: &csi.ControllerServiceCapability_Rpc{ + Rpc: &csi.ControllerServiceCapability_RPC{ + Type: csi.ControllerServiceCapability_RPC_MODIFY_VOLUME, + }, + }, + }, } if !s.config.DisableAttach { @@ -736,6 +743,14 @@ func (s *service) ControllerExpandVolume( return resp, nil } +func (s *service) ControllerModifyVolume( + ctx context.Context, + req *csi.ControllerModifyVolumeRequest) (*csi.ControllerModifyVolumeResponse, error) { + // todo: implement the functionality while we add the modifyVolume test + resp := &csi.ControllerModifyVolumeResponse{} + return resp, nil +} + func getSnapshotById(s *service, req *csi.ListSnapshotsRequest) (*csi.ListSnapshotsResponse, error) { if len(req.GetSnapshotId()) != 0 { i, snapshot := s.snapshots.FindSnapshot("id", req.GetSnapshotId())