fix(domain): allow single or multiple dns zone in waiter (#1839)

scaleway · Sep 20, 2023 · d2f8a2e · d2f8a2e
1 parent a731a00
commit d2f8a2e
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/api/domain/v2beta1/domain_utils.go b/api/domain/v2beta1/domain_utils.go
@@ -25,6 +25,7 @@ const (
 // WaitForDNSZoneRequest is used by WaitForDNSZone method.
 type WaitForDNSZoneRequest struct {
 	DNSZone       string
+	DNSZones      []string
 	Timeout       *time.Duration
 	RetryInterval *time.Duration
 }
@@ -51,10 +52,16 @@ func (s *API) WaitForDNSZone(
 
 	dns, err := async.WaitSync(&async.WaitSyncConfig{
 		Get: func() (interface{}, bool, error) {
+			listReq := &ListDNSZonesRequest{
+				DNSZones: req.DNSZones,
+			}
+
+			if req.DNSZone != "" {
+				listReq.DNSZone = &req.DNSZone
+			}
+
 			// listing dns zones and take the first one
-			DNSZones, err := s.ListDNSZones(&ListDNSZonesRequest{
-				DNSZones: []string{req.DNSZone},
-			}, opts...)
+			DNSZones, err := s.ListDNSZones(listReq, opts...)
 
 			if err != nil {
 				return nil, false, err