test: fix test panic

kubernetes-sigs · Jan 31, 2024 · 4c1e5c2 · 4c1e5c2
1 parent 60bd25f
commit 4c1e5c2
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/test/e2e/testsuites/testsuites.go b/test/e2e/testsuites/testsuites.go
@@ -989,10 +989,12 @@ func (t *TestPod) GetZoneForVolume(ctx context.Context, index int) string {
 	framework.ExpectNoError(err)
 
 	zone := ""
-	for _, term := range pv.Spec.NodeAffinity.Required.NodeSelectorTerms {
-		for _, ex := range term.MatchExpressions {
-			if ex.Key == "topology.disk.csi.azure.com/zone" && ex.Operator == v1.NodeSelectorOpIn {
-				zone = ex.Values[0]
+	if pv.Spec.NodeAffinity == nil || pv.Spec.NodeAffinity.Required == nil {
+		for _, term := range pv.Spec.NodeAffinity.Required.NodeSelectorTerms {
+			for _, ex := range term.MatchExpressions {
+				if ex.Key == "topology.disk.csi.azure.com/zone" && ex.Operator == v1.NodeSelectorOpIn {
+					zone = ex.Values[0]
+				}
 			}
 		}
 	}