Add a new field for ControllerPublish'd state

Author: gnufied

mock/service/controller.go

@@ -228,7 +228,7 @@ func (s *service) ControllerPublishVolume(
 	s.vols[i] = v
 
 	if volInfo, ok := MockVolumes[req.VolumeId]; ok {
-		volInfo.ISPublished = true
+		volInfo.ISControllerPublished = true
 		MockVolumes[req.VolumeId] = volInfo
 	}
 
@@ -628,8 +628,8 @@ func (s *service) ControllerExpandVolume(
 		return nil, status.Error(codes.NotFound, req.VolumeId)
 	}
 
-	if s.config.DisableOnlineExpansion && MockVolumes[v.GetVolumeId()].ISPublished {
-		return nil, status.Error(codes.Aborted, "volume is published and online volume expansion is not supported")
+	if s.config.DisableOnlineExpansion && MockVolumes[v.GetVolumeId()].ISControllerPublished {
+		return nil, status.Error(codes.FailedPrecondition, "volume is published and online volume expansion is not supported")
 	}
 
 	requestBytes := req.CapacityRange.RequiredBytes

mock/service/service.go

@@ -109,13 +109,14 @@ type service struct {
 }
 
 type Volume struct {
-	VolumeCSI       csi.Volume
-	NodeID          string
-	ISStaged        bool
-	ISPublished     bool
-	ISEphemeral     bool
-	StageTargetPath string
-	TargetPath      string
+	VolumeCSI             csi.Volume
+	NodeID                string
+	ISStaged              bool
+	ISPublished           bool
+	ISEphemeral           bool
+	ISControllerPublished bool
+	StageTargetPath       string
+	TargetPath            string
 }
 
 var MockVolumes map[string]Volume