fix(edge-services): fetch zone from provider config when field is missing

internal/services/edgeservices/backend_stage.go

@@ -112,7 +112,10 @@ func ResourceBackendStage() *schema.Resource {
 }
 
 func ResourceBackendStageCreate(ctx context.Context, d *schema.ResourceData, m any) diag.Diagnostics {
-	api := NewEdgeServicesAPI(m)
+	api, zone, err := edgeServicesAPIWithZone(d, m)
+	if err != nil {
+		return diag.FromErr(err)
+	}
 
 	req := &edgeservices.CreateBackendStageRequest{
 		PipelineID: d.Get("pipeline_id").(string),
@@ -123,7 +126,7 @@ func ResourceBackendStageCreate(ctx context.Context, d *schema.ResourceData, m a
 	}
 
 	if lbConfig, ok := d.GetOk("lb_backend_config"); ok {
-		req.ScalewayLB = expandLBBackendConfig(lbConfig)
+		req.ScalewayLB = expandLBBackendConfig(zone, lbConfig)
 	}
 
 	backendStage, err := api.CreateBackendStage(req, scw.WithContext(ctx))
@@ -137,7 +140,10 @@ func ResourceBackendStageCreate(ctx context.Context, d *schema.ResourceData, m a
 }
 
 func ResourceBackendStageRead(ctx context.Context, d *schema.ResourceData, m any) diag.Diagnostics {
-	api := NewEdgeServicesAPI(m)
+	api, zone, err := edgeServicesAPIWithZone(d, m)
+	if err != nil {
+		return diag.FromErr(err)
+	}
 
 	backendStage, err := api.GetBackendStage(&edgeservices.GetBackendStageRequest{
 		BackendStageID: d.Id(),
@@ -161,14 +167,17 @@ func ResourceBackendStageRead(ctx context.Context, d *schema.ResourceData, m any
 	}
 
 	if backendStage.ScalewayLB != nil {
-		_ = d.Set("lb_backend_config", flattenLBBackendConfig(backendStage.ScalewayLB))
+		_ = d.Set("lb_backend_config", flattenLBBackendConfig(zone, backendStage.ScalewayLB))
 	}
 
 	return nil
 }
 
 func ResourceBackendStageUpdate(ctx context.Context, d *schema.ResourceData, m any) diag.Diagnostics {
-	api := NewEdgeServicesAPI(m)
+	api, zone, err := edgeServicesAPIWithZone(d, m)
+	if err != nil {
+		return diag.FromErr(err)
+	}
 
 	hasChanged := false
 
@@ -182,7 +191,7 @@ func ResourceBackendStageUpdate(ctx context.Context, d *schema.ResourceData, m a
 	}
 
 	if d.HasChange("lb_backend_config") {
-		updateRequest.ScalewayLB = expandLBBackendConfig(d.Get("lb_backend_config"))
+		updateRequest.ScalewayLB = expandLBBackendConfig(zone, d.Get("lb_backend_config"))
 		hasChanged = true
 	}
 

internal/services/edgeservices/backend_stage_test.go

@@ -53,3 +53,90 @@ func TestAccEdgeServicesBackend_Basic(t *testing.T) {
 		},
 	})
 }
+
+func TestAccEdgeServicesBackend_LB(t *testing.T) {
+	tt := acctest.NewTestTools(t)
+	defer tt.Cleanup()
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck:          func() { acctest.PreCheck(t) },
+		ProviderFactories: tt.ProviderFactories,
+		CheckDestroy:      edgeservicestestfuncs.CheckEdgeServicesBackendDestroy(tt),
+		Steps: []resource.TestStep{
+			{
+				Config: `
+					resource "scaleway_lb_ip" "lb_ip" {}
+
+					resource "scaleway_lb" "lb01" {
+					  name  = "lb_name"
+					  ip_id = scaleway_lb_ip.lb_ip.id
+					  type  = "LB-S"
+					}
+
+					resource "scaleway_lb_backend" "bck01" {
+					  lb_id            = scaleway_lb.lb01.id
+					  name             = "backend"
+					  forward_protocol = "http"
+					  forward_port     = 80
+					  ssl_bridging     = true
+
+					  health_check_http {
+						uri    = "/healthcheck"
+						method = "GET"
+						code   = 200
+					  }
+					}
+
+					resource "scaleway_lb_frontend" "frt01" {
+					  lb_id        = scaleway_lb.lb01.id
+					  backend_id   = scaleway_lb_backend.bck01.id
+					  name         = "frontend"
+					  inbound_port = "443"
+					  certificate_ids = [
+						scaleway_lb_certificate.cert01.id,
+					  ]
+					}
+
+					resource "scaleway_lb_certificate" "cert01" {
+					  lb_id = scaleway_lb.lb01.id
+					  name  = "test-cert"
+					  letsencrypt {
+						common_name = "${replace(scaleway_lb_ip.lb_ip.ip_address, ".", "-")}.lb.${scaleway_lb.lb01.region}.scw.cloud"
+					  }
+					}
+
+					resource "scaleway_edge_services_pipeline" "main" {
+					  name        = "my-edge_services-pipeline"
+					  description = "pipeline description"
+					}
+
+					resource "scaleway_edge_services_backend_stage" "main" {
+                      pipeline_id = scaleway_edge_services_pipeline.main.id
+					  lb_backend_config {
+					    lb_config {
+						  id          = scaleway_lb.lb01.id
+						  frontend_id = scaleway_lb_frontend.frt01.id
+						  is_ssl      = true
+					    }
+					  }
+					}
+				`,
+				Check: resource.ComposeTestCheckFunc(
+					edgeservicestestfuncs.CheckEdgeServicesBackendExists(tt, "scaleway_edge_services_backend_stage.main"),
+					resource.TestCheckResourceAttrPair(
+						"scaleway_edge_services_backend_stage.main", "lb_backend_config.0.lb_config.0.id",
+						"scaleway_lb.lb01", "id"),
+					resource.TestCheckResourceAttrPair(
+						"scaleway_edge_services_backend_stage.main", "lb_backend_config.0.lb_config.0.frontend_id",
+						"scaleway_lb_frontend.frt01", "id"),
+					resource.TestCheckResourceAttrPair(
+						"scaleway_edge_services_pipeline.main", "id",
+						"scaleway_edge_services_backend_stage.main", "pipeline_id"),
+					resource.TestCheckResourceAttr("scaleway_edge_services_backend_stage.main", "lb_backend_config.0.lb_config.0.is_ssl", "true"),
+					resource.TestCheckResourceAttr("scaleway_edge_services_backend_stage.main", "lb_backend_config.0.lb_config.0.zone", "fr-par-1"),
+					resource.TestCheckResourceAttrSet("scaleway_edge_services_backend_stage.main", "created_at"),
+					resource.TestCheckResourceAttrSet("scaleway_edge_services_backend_stage.main", "updated_at"),
+				),
+			},
+		},
+	})
+}

internal/services/edgeservices/helpers.go

@@ -23,3 +23,15 @@ func NewEdgeServicesAPIWithRegion(d *schema.ResourceData, m any) (*edgeservices.
 
 	return api, region, err
 }
+
+// edgeServicesAPIWithZone returns a new edge_services API and the zone
+func edgeServicesAPIWithZone(d *schema.ResourceData, m interface{}) (*edgeservices.API, scw.Zone, error) {
+	api := edgeservices.NewAPI(meta.ExtractScwClient(m))
+
+	zone, err := meta.ExtractZone(d, m)
+	if err != nil {
+		return nil, "", err
+	}
+
+	return api, zone, nil
+}