Run snapshot class tests separately from other tests

Author: mattcary

test/k8s-integration/driver-config.go

@@ -34,7 +34,7 @@ const (
 
 // generateDriverConfigFile loads a testdriver config template and creates a file
 // with the test-specific configuration
-func generateDriverConfigFile(testParams *testParameters, storageClassFile string) (string, error) {
+func generateDriverConfigFile(testParams *testParameters) (string, error) {
 	// Load template
 	t, err := template.ParseFiles(filepath.Join(testParams.pkgDir, testConfigDir, configTemplateFile))
 	if err != nil {
@@ -130,16 +130,18 @@ func generateDriverConfigFile(testParams *testParameters, storageClassFile strin
 	caps = append(caps, "pvcDataSource")
 	minimumVolumeSize := "5Gi"
 	numAllowedTopologies := 1
-	if storageClassFile == regionalPDStorageClass {
+	if testParams.storageClassFile == regionalPDStorageClass {
 		minimumVolumeSize = "200Gi"
 		numAllowedTopologies = 2
 	}
 	timeouts := map[string]string{
 		dataSourceProvisionTimeoutKey: dataSourceProvisionTimeout,
 	}
+	extLoc := strings.LastIndex(testParams.storageClassFile, ".")
+	scName := testParams.storageClassFile[:extLoc]
 	params := driverConfig{
-		StorageClassFile:     filepath.Join(testParams.pkgDir, testConfigDir, storageClassFile),
-		StorageClass:         storageClassFile[:strings.LastIndex(storageClassFile, ".")],
+		StorageClassFile:     filepath.Join(testParams.pkgDir, testConfigDir, testParams.storageClassFile),
+		StorageClass:         scName,
 		SnapshotClassFile:    absSnapshotClassFilePath,
 		SnapshotClass:        snapshotClassName,
 		SupportedFsType:      fsTypes,

test/k8s-integration/main.go

@@ -93,6 +93,7 @@ type testParameters struct {
 	k8sSourceDir         string
 	testFocus            string
 	testSkip             string
+	storageClassFile     string
 	snapshotClassFile    string
 	cloudProviderArgs    []string
 	deploymentStrategy   string
@@ -501,21 +502,27 @@ func handle() error {
 				applicableSnapshotClassFiles = append(applicableSnapshotClassFiles, snapshotClassFile)
 			}
 		}
-		if len(applicableSnapshotClassFiles) == 0 {
-			// when no snapshot class specified, we run the tests without snapshot capability
-			applicableSnapshotClassFiles = append(applicableSnapshotClassFiles, "")
-		}
 		var ginkgoErrors []string
 		var testOutputDirs []string
+
+		// Run non-snapshot tests.
+		testParams.snapshotClassFile = ""
 		for _, scFile := range applicableStorageClassFiles {
 			outputDir := strings.TrimSuffix(scFile, ".yaml")
+			testOutputDirs = append(testOutputDirs, outputDir)
+			testParams.storageClassFile = scFile
+			if err = runCSITests(testParams, outputDir); err != nil {
+				ginkgoErrors = append(ginkgoErrors, err.Error())
+			}
+		}
+		// Run snapshot tests, if there are applicable files, using the first storage class.
+		if len(applicableStorageClassFiles) > 0 {
+			testParams.storageClassFile = applicableStorageClassFiles[0]
 			for _, snapshotClassFile := range applicableSnapshotClassFiles {
-				if len(snapshotClassFile) != 0 {
-					outputDir = fmt.Sprintf("%s--%s", outputDir, strings.TrimSuffix(snapshotClassFile, ".yaml"))
-				}
-				testOutputDirs = append(testOutputDirs, outputDir)
 				testParams.snapshotClassFile = snapshotClassFile
-				if err = runCSITests(testParams, scFile, outputDir); err != nil {
+				outputDir := strings.TrimSuffix(snapshotClassFile, ".yaml")
+				testOutputDirs = append(testOutputDirs, outputDir)
+				if err = runCSITests(testParams, outputDir); err != nil {
 					ginkgoErrors = append(ginkgoErrors, err.Error())
 				}
 			}
@@ -633,8 +640,8 @@ func runMigrationTests(testParams *testParameters) error {
 	return runTestsWithConfig(testParams, "--storage.migratedPlugins=kubernetes.io/gce-pd", "")
 }
 
-func runCSITests(testParams *testParameters, storageClassFile string, reportPrefix string) error {
-	testDriverConfigFile, err := generateDriverConfigFile(testParams, storageClassFile)
+func runCSITests(testParams *testParameters, reportPrefix string) error {
+	testDriverConfigFile, err := generateDriverConfigFile(testParams)
 	if err != nil {
 		return fmt.Errorf("failed to generated driver config: %w", err)
 	}
@@ -668,7 +675,17 @@ func runTestsWithConfig(testParams *testParameters, testConfigArg, reportPrefix
 			kubetestDumpDir = artifactsDir
 		}
 	}
-	ginkgoArgs := fmt.Sprintf("--ginkgo.focus=%s --ginkgo.skip=%s", testParams.testFocus, testParams.testSkip)
+
+	focus := testParams.testFocus
+	skip := testParams.testSkip
+	if testParams.snapshotClassFile == "" && !strings.Contains(skip, "VolumeSnapshotDataSource") {
+		skipString = skip + "|VolumeSnapshotDataSource"
+	} else if testParams.snapshotClassFile != "" && strings.Contains(skip, "VolumeSnapshotDataSource") {
+		return fmt.Errorf("Snapshot class file %s specified, but snapshot tests are skipped: %s", testParams.snapshotClassFile, skip)
+	} else if testParams.snapshotClassFile != "" and !strings.Contains(focus, "VolumeSnapshotDataSource") {
+		focus = focus + "|VolumeSnapshotDataSource"
+	}
+	ginkgoArgs := fmt.Sprintf("--ginkgo.focus=%s --ginkgo.skip=%s", focus, skip)
 
 	windowsArgs := ""
 	if testParams.platform == "windows" {