fix E2E tests when kube client not available

Author: Sneha-at

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

@@ -299,24 +299,26 @@ func urlFlag(target **url.URL, name string, usage string) {
 }
 
 func setupDataCache(ctx context.Context, nodeName string) error {
-	cfg, err := rest.InClusterConfig()
-	if err != nil {
-		return err
-	}
-	kubeClient, err := kubernetes.NewForConfig(cfg)
-	if err != nil {
-		return err
-	}
-	node, err := kubeClient.CoreV1().Nodes().Get(ctx, nodeName, metav1.GetOptions{})
-	if err != nil {
-		// We could retry, but this error will also crashloop the driver which may be as good a way to retry as any.
-		return err
-	}
-	if val, found := node.GetLabels()[dataCacheLabel]; !found || val != dataCacheLabelValue {
-		klog.V(2).Infof("Datacache not enabled for node %s; node label %s=%s and not %s", nodeName, dataCacheLabel, val, dataCacheLabelValue)
-		return nil
+	klog.V(2).Infof("Seting up data cache for node %s", nodeName)
+	if nodeName != common.TestNode {
+		cfg, err := rest.InClusterConfig()
+		if err != nil {
+			return err
+		}
+		kubeClient, err := kubernetes.NewForConfig(cfg)
+		if err != nil {
+			return err
+		}
+		node, err := kubeClient.CoreV1().Nodes().Get(ctx, nodeName, metav1.GetOptions{})
+		if err != nil {
+			// We could retry, but this error will also crashloop the driver which may be as good a way to retry as any.
+			return err
+		}
+		if val, found := node.GetLabels()[dataCacheLabel]; !found || val != dataCacheLabelValue {
+			klog.V(2).Infof("Datacache not enabled for node %s; node label %s=%s and not %s", nodeName, dataCacheLabel, val, dataCacheLabelValue)
+			return nil
+		}
 	}
-	// Setup data cache only if enabled fro nodes
 	klog.V(2).Info("Raiding local ssds to setup data cache")
 	if err := driver.RaidLocalSsds(); err != nil {
 		return fmt.Errorf("Failed to Raid local SSDs, unable to setup data caching, got error %v", err)

pkg/common/constants.go

@@ -42,4 +42,6 @@ const (
 
 	// Keys in the publish context
 	ContexLocalSsdCacheSize = "local-ssd-cache-size"
+	// Node name for E2E tests
+	TestNode = "test-node-csi-e2e"
 )

test/e2e/utils/utils.go

@@ -62,6 +62,7 @@ func GCEClientAndDriverSetup(instance *remote.InstanceInfo, computeEndpoint stri
 		"--use-instance-api-to-list-volumes-published-nodes",
 		"--enable-controller-data-cache",
 		"--enable-node-data-cache",
+		fmt.Sprintf("--node-name=%s", utilcommon.TestNode),
 	}
 	extra_flags = append(extra_flags, fmt.Sprintf("--compute-endpoint=%s", computeEndpoint))