test: use mock mount to fix sanit test failures

Author: andyzhangx

pkg/blob/blob.go

@@ -95,6 +95,7 @@ type Driver struct {
 	csicommon.CSIDriver
 	cloud                   *azure.Cloud
 	blobfuseProxyEndpoint   string
+	enableBlobMockMount     bool
 	enableBlobfuseProxy     bool
 	blobfuseProxyConnTimout int
 	mounter                 *mount.SafeFormatAndMount
@@ -106,7 +107,7 @@ type Driver struct {
 
 // NewDriver Creates a NewCSIDriver object. Assumes vendor version is equal to driver version &
 // does not support optional driver plugin info manifest field. Refer to CSI spec for more details.
-func NewDriver(nodeID, blobfuseProxyEndpoint string, enableBlobfuseProxy bool, blobfuseProxyConnTimout int) *Driver {
+func NewDriver(nodeID, blobfuseProxyEndpoint string, enableBlobfuseProxy bool, blobfuseProxyConnTimout int, enableBlobMockMount bool) *Driver {
 	driver := Driver{}
 	driver.Name = DriverName
 	driver.Version = driverVersion
@@ -116,6 +117,7 @@ func NewDriver(nodeID, blobfuseProxyEndpoint string, enableBlobfuseProxy bool, b
 	driver.blobfuseProxyEndpoint = blobfuseProxyEndpoint
 	driver.enableBlobfuseProxy = enableBlobfuseProxy
 	driver.blobfuseProxyConnTimout = blobfuseProxyConnTimout
+	driver.enableBlobMockMount = enableBlobMockMount
 	return &driver
 }
 

pkg/blob/blob_test.go

@@ -44,19 +44,19 @@ const (
 )
 
 func NewFakeDriver() *Driver {
-	driver := NewDriver(fakeNodeID, "", false, 5)
+	driver := NewDriver(fakeNodeID, "", false, 5, false)
 	driver.Name = fakeDriverName
 	driver.Version = vendorVersion
 	return driver
 }
 
 func TestNewFakeDriver(t *testing.T) {
-	d := NewDriver(fakeNodeID, "", false, 5)
+	d := NewDriver(fakeNodeID, "", false, 5, false)
 	assert.NotNil(t, d)
 }
 
 func TestNewDriver(t *testing.T) {
-	driver := NewDriver(fakeNodeID, "", false, 5)
+	driver := NewDriver(fakeNodeID, "", false, 5, false)
 	fakedriver := NewFakeDriver()
 	fakedriver.Name = DriverName
 	fakedriver.Version = driverVersion

pkg/blob/nodeserver.go

@@ -92,6 +92,14 @@ func (d *Driver) NodePublishVolume(ctx context.Context, req *csi.NodePublishVolu
 	}
 
 	klog.V(2).Infof("NodePublishVolume: volume %s mounting %s at %s with mountOptions: %v", volumeID, source, target, mountOptions)
+	if d.enableBlobMockMount {
+		if err := volumehelper.MakeDir(target); err != nil {
+			klog.Errorf("MakeDir failed on target: %s (%v)", target, err)
+			return nil, err
+		}
+		return &csi.NodePublishVolumeResponse{}, nil
+	}
+
 	if err := d.mounter.Mount(source, target, "", mountOptions); err != nil {
 		if removeErr := os.Remove(target); removeErr != nil {
 			return nil, status.Errorf(codes.Internal, "Could not remove mount target %q: %v", target, removeErr)

pkg/blobplugin/main.go

@@ -44,6 +44,7 @@ var (
 	kubeconfig              = flag.String("kubeconfig", "", "Absolute path to the kubeconfig file. Required only when running out of cluster.")
 	enableBlobfuseProxy     = flag.Bool("enable-blobfuse-proxy", false, "Whether supports using Blobfuse proxy for mounts")
 	blobfuseProxyConnTimout = flag.Int("blobfuse-proxy-connect-timeout", 5, "blobfuse proxy connection timeout(seconds)")
+	enableBlobMockMount     = flag.Bool("enable-blob-mock-mount", false, "Whether enable mock mount")
 )
 
 func main() {
@@ -64,7 +65,7 @@ func main() {
 }
 
 func handle() {
-	driver := blob.NewDriver(*nodeID, *blobfuseProxyEndpoint, *enableBlobfuseProxy, *blobfuseProxyConnTimout)
+	driver := blob.NewDriver(*nodeID, *blobfuseProxyEndpoint, *enableBlobfuseProxy, *blobfuseProxyConnTimout, *enableBlobMockMount)
 	if driver == nil {
 		klog.Fatalln("Failed to initialize Azure Blob Storage CSI driver")
 	}

test/e2e/suite_test.go

@@ -110,7 +110,7 @@ var _ = ginkgo.BeforeSuite(func() {
 	nodeid := os.Getenv("nodeid")
 	kubeconfig := os.Getenv(kubeconfigEnvVar)
 	_, useBlobfuseProxy := os.LookupEnv("ENABLE_BLOBFUSE_PROXY")
-	blobDriver = blob.NewDriver(nodeid, "", useBlobfuseProxy, 5)
+	blobDriver = blob.NewDriver(nodeid, "", useBlobfuseProxy, 5, false)
 	go func() {
 		os.Setenv("AZURE_CREDENTIAL_FILE", credentials.TempAzureCredentialFilePath)
 		blobDriver.Run(fmt.Sprintf("unix:///tmp/csi-%s.sock", uuid.NewUUID().String()), kubeconfig, false)

test/sanity/run-test.sh

@@ -33,7 +33,7 @@ if [[ "$#" -gt 0 ]] && [[ -n "$1" ]]; then
 fi
 
 _output/blobplugin --endpoint "$controllerendpoint" -v=5 &
-_output/blobplugin --endpoint "$nodeendpoint" --nodeid "$nodeid" -v=5 &
+_output/blobplugin --endpoint "$nodeendpoint" --nodeid "$nodeid" -v=5 --enable-blob-mock-mount &
 
 echo "Begin to run sanity test..."
 readonly CSI_SANITY_BIN='csi-sanity'