From cd8dc20f7f83c2da51b99768dbcee9c31479d97a Mon Sep 17 00:00:00 2001
From: liyanhui1228 <yanhuil@google.com>
Date: Wed, 3 Feb 2021 13:57:44 -0800
Subject: [PATCH] Add e2e test setup for windows vsphere driver

---
 test/e2e/storage/drivers/in_tree.go | 25 +++++++++++++++++++++++++
 test/e2e/storage/in_tree_volumes.go |  1 +
 2 files changed, 26 insertions(+)

diff --git a/test/e2e/storage/drivers/in_tree.go b/test/e2e/storage/drivers/in_tree.go
index 2e3682bc69c..e578c4cc956 100644
--- a/test/e2e/storage/drivers/in_tree.go
+++ b/test/e2e/storage/drivers/in_tree.go
@@ -1293,6 +1293,30 @@ func InitWindowsGcePdDriver() storageframework.TestDriver {
 	}
 }
 
+func InitWindowsVSphereDriver() storageframework.TestDriver {
+	return &vSphereDriver{
+		driverInfo: storageframework.DriverInfo{
+			Name:             "windows-vsphere",
+			InTreePluginName: "kubernetes.io/vsphere-volume",
+			MaxFileSize:      storageframework.FileSizeMedium,
+			SupportedSizeRange: e2evolume.SizeRange{
+				Min: "1Gi",
+			},
+			SupportedFsType: sets.NewString(
+				"ntfs",
+			),
+			TopologyKeys: []string{v1.LabelFailureDomainBetaZone},
+			Capabilities: map[storageframework.Capability]bool{
+				storageframework.CapPersistence: true,
+				storageframework.CapFsGroup:     true,
+				storageframework.CapExec:        true,
+				storageframework.CapMultiPODs:   true,
+				storageframework.CapTopology:    true,
+			},
+		},
+	}
+}
+
 func (g *gcePdDriver) GetDriverInfo() *storageframework.DriverInfo {
 	return &g.driverInfo
 }
@@ -1416,6 +1440,7 @@ func InitVSphereDriver() storageframework.TestDriver {
 			SupportedFsType: sets.NewString(
 				"", // Default fsType
 				"ext4",
+				"ntfs",
 			),
 			TopologyKeys: []string{v1.LabelFailureDomainBetaZone},
 			Capabilities: map[storageframework.Capability]bool{
diff --git a/test/e2e/storage/in_tree_volumes.go b/test/e2e/storage/in_tree_volumes.go
index b5572837d0e..7ed9e5277cf 100644
--- a/test/e2e/storage/in_tree_volumes.go
+++ b/test/e2e/storage/in_tree_volumes.go
@@ -38,6 +38,7 @@ var testDrivers = []func() storageframework.TestDriver{
 	drivers.InitGcePdDriver,
 	drivers.InitWindowsGcePdDriver,
 	drivers.InitVSphereDriver,
+	drivers.InitWindowsVSphereDriver,
 	drivers.InitAzureDiskDriver,
 	drivers.InitAwsDriver,
 	drivers.InitLocalDriverWithVolumeType(utils.LocalVolumeDirectory),