github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-09-03 18:27:49 +00:00
Code Issues Projects Releases Wiki Activity

Rework createSELinuxMountPreHook to switch()

Browse Source
This commit is contained in:
Jan Safranek
2023-01-17 11:23:09 +01:00
parent d2bb866d3f
commit b9c2449569
1 changed files with 7 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
test/e2e/storage/csi_mock/base.go
View File

@@ -829,28 +829,22 @@ func compareCSICalls(ctx context.Context, trackedCalls []string, expectedCallSeq
func createSELinuxMountPreHook(nodeStageMountOpts, nodePublishMountOpts *[]string, stageCalls, unstageCalls, publishCalls, unpublishCalls *atomic.Int32) *drivers.Hooks { func createSELinuxMountPreHook(nodeStageMountOpts, nodePublishMountOpts *[]string, stageCalls, unstageCalls, publishCalls, unpublishCalls *atomic.Int32) *drivers.Hooks {
return &drivers.Hooks{ return &drivers.Hooks{
Pre: func(ctx context.Context, fullMethod string, request interface{}) (reply interface{}, err error) { Pre: func(ctx context.Context, fullMethod string, request interface{}) (reply interface{}, err error) {
nodeStageRequest, ok := request.(*csipbv1.NodeStageVolumeRequest) switch req := request.(type) {
if ok { case *csipbv1.NodeStageVolumeRequest:
stageCalls.Add(1) stageCalls.Add(1)
mountVolume := nodeStageRequest.GetVolumeCapability().GetMount() mountVolume := req.GetVolumeCapability().GetMount()
if mountVolume != nil { if mountVolume != nil {
*nodeStageMountOpts = mountVolume.MountFlags *nodeStageMountOpts = mountVolume.MountFlags
} }
} case *csipbv1.NodePublishVolumeRequest:
nodePublishRequest, ok := request.(*csipbv1.NodePublishVolumeRequest)
if ok {
publishCalls.Add(1) publishCalls.Add(1)
mountVolume := nodePublishRequest.GetVolumeCapability().GetMount() mountVolume := req.GetVolumeCapability().GetMount()
if mountVolume != nil { if mountVolume != nil {
*nodePublishMountOpts = mountVolume.MountFlags *nodePublishMountOpts = mountVolume.MountFlags
} }
} case *csipbv1.NodeUnstageVolumeRequest:
_, ok = request.(*csipbv1.NodeUnstageVolumeRequest)
if ok {
unstageCalls.Add(1) unstageCalls.Add(1)
} case *csipbv1.NodeUnpublishVolumeRequest:
_, ok = request.(*csipbv1.NodeUnpublishVolumeRequest)
if ok {
unpublishCalls.Add(1) unpublishCalls.Add(1)
} }
return nil, nil return nil, nil