dafult data cache enabled to false

Author: Sneha-at

cmd/gce-pd-csi-driver/main.go

@@ -245,7 +245,7 @@ func handle() {
 		if err != nil {
 			klog.Fatalf("Failed to set up metadata service: %v", err.Error())
 		}
-		isDataCacheEnabledNodePool, err := isDataCacheEnabledNodePool(ctx, *nodeName)
+		isDataCacheEnabledNodePool, err := driver.IsDataCacheEnabledNodePool(ctx, *nodeName, *enableDataCacheFlag)
 		if err != nil {
 			klog.Fatalf("Failed to get node info from API server: %v", err.Error())
 		}
@@ -351,17 +351,6 @@ func urlFlag(target **url.URL, name string, usage string) {
 	})
 }
 
-func isDataCacheEnabledNodePool(ctx context.Context, nodeName string) (bool, error) {
-	if !*enableDataCacheFlag {
-		return false, nil
-	}
-	if len(nodeName) > 0 && nodeName != common.TestNode { // disregard logic below when E2E testing.
-		dataCacheLSSDCount, err := driver.GetDataCacheCountFromNodeLabel(ctx, nodeName)
-		return dataCacheLSSDCount != 0, err
-	}
-	return true, nil
-}
-
 func fetchLssdsForRaiding(lssdCount int) ([]string, error) {
 	allLssds, err := driver.FetchAllLssds()
 	if err != nil {

pkg/gce-pd-csi-driver/node.go

@@ -526,7 +526,7 @@ func (ns *GCENodeServer) NodeUnstageVolume(ctx context.Context, req *csi.NodeUns
 
 	// The NodeUnstageVolume does not have any volume or publish context, we need to get the info from LVM locally
 	// Check if cache group cache-{volumeID} exist in LVM
-	if ns.EnableDataCache {
+	if ns.EnableDataCache && ns.DataCacheEnabledNodePool {
 		nodeId := ns.MetadataService.GetName()
 		err := cleanupCache(volumeID, nodeId)
 		if err != nil {

pkg/gce-pd-csi-driver/utils.go

@@ -331,3 +331,17 @@ func containsZone(zones []string, zone string) bool {
 
 	return false
 }
+
+func IsDataCacheEnabledNodePool(ctx context.Context, nodeName string, enableDataCacheFlag bool) (bool, error) {
+	if !enableDataCacheFlag {
+		return false, nil
+	}
+	if nodeName == common.TestNode { // disregard logic below when E2E testing.
+		return true, nil
+	}
+	if len(nodeName) > 0 {
+		dataCacheLSSDCount, err := GetDataCacheCountFromNodeLabel(ctx, nodeName)
+		return dataCacheLSSDCount != 0, err
+	}
+	return false, nil
+}

pkg/gce-pd-csi-driver/utils_test.go

@@ -18,8 +18,10 @@ limitations under the License.
 package gceGCEDriver
 
 import (
+	"context"
 	"fmt"
 	"testing"
+	"time"
 
 	csi "github.com/container-storage-interface/spec/lib/go/csi"
 	"github.com/google/go-cmp/cmp"
@@ -857,3 +859,54 @@ func TestGetHyperdiskAccessModeFromCapabilities(t *testing.T) {
 		}
 	}
 }
+
+func TestIsDataCacheEnabledNodePool(t *testing.T) {
+	for _, tc := range []struct {
+		name                 string
+		nodeName             string
+		wantDataCacheEnabled bool
+		dataCacheFlag        bool
+		wantErr              bool
+	}{
+		{
+			// Valid nod ename tries to fetch the data cache count from node labels resulting in an error
+			name:                 "node name is provided",
+			nodeName:             "gke-node-some-name",
+			dataCacheFlag:        true,
+			wantDataCacheEnabled: true,
+			wantErr:              true,
+		},
+		{
+			name:                 "no node name provided",
+			nodeName:             "",
+			dataCacheFlag:        true,
+			wantDataCacheEnabled: false,
+		},
+		{
+			name:                 "test node",
+			nodeName:             common.TestNode,
+			dataCacheFlag:        true,
+			wantDataCacheEnabled: true,
+		},
+		{
+			name:                 "node name provided but data cache feature disabled",
+			nodeName:             "",
+			dataCacheFlag:        false,
+			wantDataCacheEnabled: false,
+		},
+	} {
+		t.Logf("Running test: %v", tc.name)
+		ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+		defer cancel()
+		gotDataCacheEnabled, err := IsDataCacheEnabledNodePool(ctx, tc.nodeName, tc.dataCacheFlag)
+		if err != nil {
+			if !tc.wantErr {
+				t.Errorf("unexpected error, got %v", err)
+			}
+			continue
+		}
+		if gotDataCacheEnabled != tc.wantDataCacheEnabled {
+			t.Errorf("want %t, got %t", tc.wantDataCacheEnabled, gotDataCacheEnabled)
+		}
+	}
+}