fix test suite issue

Author: Sneha-at

test/e2e/tests/setup_e2e_test.go

@@ -81,8 +81,6 @@ var _ = BeforeSuite(func() {
 	defer close(hdtcc)
 
 	zones := strings.Split(*zones, ",")
-	// Create 2 instances for each zone as we need 2 instances each zone for certain test cases
-
 	rand.Seed(time.Now().UnixNano())
 
 	computeService, err = remote.GetComputeClient()
@@ -109,27 +107,31 @@ var _ = BeforeSuite(func() {
 
 	numberOfInstancesPerZone := 2
 
-	setupContext := func(zones []string, randInt int) {
-		for _, zone := range zones {
-			go func(curZone string) {
+	setupContext := func(zone string) {
+		// Create 2 instances for each zone as we need 2 instances each zone for certain test cases
+		for j := 0; j < numberOfInstancesPerZone; j++ {
+			go func(curZone string, randInt int) {
 				defer GinkgoRecover()
 				tcc <- NewDefaultTestContext(curZone, strconv.Itoa(randInt))
-			}(zone)
-			go func(curZone string) {
-				defer GinkgoRecover()
-				hdtcc <- NewTestContext(curZone, *hdMinCpuPlatform, *hdMachineType, strconv.Itoa(randInt))
-			}(zone)
+			}(zone, j)
 		}
+		go func(curZone string) {
+			defer GinkgoRecover()
+			hdtcc <- NewTestContext(curZone, *hdMinCpuPlatform, *hdMachineType, "0")
+		}(zone)
 	}
-	for j := 0; j < numberOfInstancesPerZone; j++ {
-		setupContext(zones, j)
+
+	for _, zone := range zones {
+		setupContext(zone)
 	}
 
 	for i := 0; i < len(zones)*numberOfInstancesPerZone; i++ {
 		tc := <-tcc
 		testContexts = append(testContexts, tc)
 		klog.Infof("Added TestContext for node %s", tc.Instance.GetName())
-		tc = <-hdtcc
+	}
+	for i := 0; i < len(zones); i++ {
+		tc := <-hdtcc
 		hyperdiskTestContexts = append(hyperdiskTestContexts, tc)
 		klog.Infof("Added TestContext for node %s", tc.Instance.GetName())
 	}
@@ -178,6 +180,11 @@ func NewTestContext(zone, minCpuPlatform, machineType string, instanceNumber str
 		ComputeService:            computeService,
 		LocalSSDCount:             localSSDCount,
 	}
+
+	if machineType == *hdMachineType {
+		// Machine type is defaulted to c3-standard-2 which doesn't support LSSD and we don't need LSSD for HdHA test context
+		instanceConfig.LocalSSDCount = 0
+	}
 	i, err := remote.SetupInstance(instanceConfig)
 	if err != nil {
 		klog.Fatalf("Failed to setup instance %v: %v", nodeID, err)