chore: add testing for locality

internal/services/baremetal/offer_data_source.go

@@ -204,7 +204,7 @@
 	zonedID := datasource.NewZonedID(offer.ID, zone)
 	d.SetId(zonedID)
 	_ = d.Set("offer_id", zonedID)
-	_ = d.Set("zone", zone)
+	_ = d.Set("zone", offer.zone)
 	_ = d.Set("name", offer.Name)
 	_ = d.Set("subscription_period", offer.SubscriptionPeriod)
 	_ = d.Set("include_disabled", !offer.Enable)

internal/services/inference/model.go

@@ -203,6 +203,7 @@ func ResourceModelRead(ctx context.Context, d *schema.ResourceData, m any) diag.
 	_ = d.Set("updated_at", types.FlattenTime(model.UpdatedAt))
 	_ = d.Set("has_eula", model.HasEula)
 	_ = d.Set("nodes_support", flattenNodeSupport(model.NodesSupport))
+	_ = d.Set("region", model.Region.String())
 
 	return nil
 }

internal/services/instance/ip.go

@@ -172,7 +172,7 @@ func ResourceInstanceIPRead(ctx context.Context, d *schema.ResourceData, m any)
 
 	_ = d.Set("address", address)
 	_ = d.Set("prefix", prefix)
-	_ = d.Set("zone", zone)
+	_ = d.Set("zone", res.IP.Zone.String())
 	_ = d.Set("organization_id", res.IP.Organization)
 	_ = d.Set("project_id", res.IP.Project)
 	_ = d.Set("reverse", res.IP.Reverse)

internal/services/instance/placement_group.go

@@ -112,7 +112,7 @@ func ResourceInstancePlacementGroupRead(ctx context.Context, d *schema.ResourceD
 	}
 
 	_ = d.Set("name", res.PlacementGroup.Name)
-	_ = d.Set("zone", string(zone))
+	_ = d.Set("zone", res.PlacementGroup.Zone.String())
 	_ = d.Set("organization_id", res.PlacementGroup.Organization)
 	_ = d.Set("project_id", res.PlacementGroup.Project)
 	_ = d.Set("policy_mode", res.PlacementGroup.PolicyMode.String())

internal/services/instance/private_nic.go

@@ -178,25 +178,25 @@
 		return diag.FromErr(err)
 	}
 
-	_ = d.Set("zone", zone)
+	_ = d.Set("zone", privateNIC.)
 	_ = d.Set("server_id", zonal.NewID(zone, privateNIC.ServerID).String())
 	_ = d.Set("private_network_id", regional.NewIDString(fetchRegion, privateNIC.PrivateNetworkID))
 	_ = d.Set("mac_address", privateNIC.MacAddress)
 
 	if len(privateNIC.Tags) > 0 {
 		_ = d.Set("tags", privateNIC.Tags)
 	}
 
 	// Get private NIC's private IPs if possible
 	diags := diag.Diagnostics{}
 
 	region, err := zone.Region()
 	if err != nil {
 		return append(diags, diag.Diagnostic{
 			Severity: diag.Warning,
 			Summary:  "Unable to get private NIC's private IPs",
 			Detail:   err.Error(),
 		})
 	}
 
 	projectID, err := getServerProjectID(ctx, instanceAPI, zone, privateNIC.ServerID)

internal/services/instance/security_group.go

@@ -160,7 +160,7 @@ func ResourceInstanceSecurityGroupRead(ctx context.Context, d *schema.ResourceDa
 		return diag.FromErr(err)
 	}
 
-	_ = d.Set("zone", zone)
+	_ = d.Set("zone", res.SecurityGroup.Zone.String())
 	_ = d.Set("organization_id", res.SecurityGroup.Organization)
 	_ = d.Set("project_id", res.SecurityGroup.Project)
 	_ = d.Set("name", res.SecurityGroup.Name)

internal/services/instance/server.go

@@ -670,7 +670,7 @@ func ResourceInstanceServerRead(ctx context.Context, d *schema.ResourceData, m a
 	}
 
 	_ = d.Set("state", state)
-	_ = d.Set("zone", string(zone))
+	_ = d.Set("zone", server.Zone.String())
 	_ = d.Set("name", server.Name)
 	_ = d.Set("boot_type", server.BootType)
 

internal/services/instance/snapshot.go

@@ -199,6 +199,7 @@ func ResourceInstanceSnapshotRead(ctx context.Context, d *schema.ResourceData, m
 	_ = d.Set("created_at", snapshot.Snapshot.CreationDate.Format(time.RFC3339))
 	_ = d.Set("type", snapshot.Snapshot.VolumeType.String())
 	_ = d.Set("tags", snapshot.Snapshot.Tags)
+	_ = d.Set("zone", snapshot.Snapshot.Zone.String())
 
 	return nil
 }

internal/services/instance/volume.go

@@ -165,7 +165,7 @@ func ResourceInstanceVolumeRead(ctx context.Context, d *schema.ResourceData, m a
 	_ = d.Set("name", res.Volume.Name)
 	_ = d.Set("organization_id", res.Volume.Organization)
 	_ = d.Set("project_id", res.Volume.Project)
-	_ = d.Set("zone", string(zone))
+	_ = d.Set("zone", res.Volume.Zone.String())
 	_ = d.Set("type", res.Volume.VolumeType.String())
 	_ = d.Set("tags", res.Volume.Tags)
 

internal/services/iot/device.go

@@ -293,7 +293,7 @@ func ResourceIotDeviceRead(ctx context.Context, d *schema.ResourceData, m any) d
 		return diag.FromErr(err)
 	}
 
-	_ = d.Set("name", device.Name)
+	_ = d.Set("name", device.regName)
 	_ = d.Set("status", device.Status)
 	_ = d.Set("hub_id", regional.NewID(region, device.HubID).String())
 	_ = d.Set("created_at", device.CreatedAt.Format(time.RFC3339))

internal/services/iot/hub.go

@@ -229,7 +229,7 @@ func ResourceIotHubRead(ctx context.Context, d *schema.ResourceData, m any) diag
 		return diag.FromErr(err)
 	}
 
-	_ = d.Set("region", string(region))
+	_ = d.Set("region", response.Region.String())
 	_ = d.Set("organization_id", response.OrganizationID)
 	_ = d.Set("project_id", response.ProjectID)
 	_ = d.Set("name", response.Name)

internal/services/iot/network.go

@@ -135,7 +135,7 @@ func ResourceIotNetworkRead(ctx context.Context, d *schema.ResourceData, m any)
 	_ = d.Set("hub_id", regional.NewID(region, network.HubID).String())
 	_ = d.Set("created_at", network.CreatedAt.Format(time.RFC3339))
 	_ = d.Set("topic_prefix", network.TopicPrefix)
-	_ = d.Set("region", string(region))
+	_ = d.Set("region", network.string(region))
 
 	return nil
 }

internal/services/iot/routes.go

@@ -272,7 +272,7 @@
 		return diag.FromErr(err)
 	}
 
-	_ = d.Set("region", string(region))
+	_ = d.Set("region", response.reg string(region))
 	_ = d.Set("name", response.Name)
 	_ = d.Set("hub_id", response.HubID)
 	_ = d.Set("topic", response.Topic)

internal/services/ipam/ip.go

@@ -281,7 +281,7 @@ func ResourceIPAMIPRead(ctx context.Context, d *schema.ResourceData, m any) diag
 	_ = d.Set("created_at", types.FlattenTime(res.CreatedAt))
 	_ = d.Set("updated_at", types.FlattenTime(res.UpdatedAt))
 	_ = d.Set("is_ipv6", res.IsIPv6)
-	_ = d.Set("region", region)
+	_ = d.Set("region", res.Region.String())
 
 	if res.Zone != nil {
 		_ = d.Set("zone", res.Zone.String())

internal/services/ipam/ip_reverse_dns.go

@@ -121,7 +121,7 @@ func ResourceIPAMIPReverseDNSRead(ctx context.Context, d *schema.ResourceData, m
 		}
 	}
 
-	_ = d.Set("region", region)
+	_ = d.Set("region", res.Region.String())
 
 	return nil
 }

internal/services/k8s/acl.go

@@ -152,7 +152,7 @@
 	}
 
 	_ = d.Set("cluster_id", regional.NewIDString(region, clusterID))
-	_ = d.Set("region", region)
+	_ = d.Set("region", acls.)
 	_ = d.Set("acl_rules", flattenACL(acls.Rules))
 
 	return nil

internal/services/k8s/cluster.go

@@ -550,7 +550,7 @@ func ResourceK8SClusterRead(ctx context.Context, d *schema.ResourceData, m any)
 		return diag.FromErr(err)
 	}
 
-	_ = d.Set("region", string(region))
+	_ = d.Set("region", cluster.Region.String())
 	_ = d.Set("name", cluster.Name)
 	_ = d.Set("type", cluster.Type)
 	_ = d.Set("organization_id", cluster.OrganizationID)

internal/services/k8s/pool.go

@@ -431,8 +431,8 @@ func ResourceK8SPoolRead(ctx context.Context, d *schema.ResourceData, m any) dia
 	_ = d.Set("updated_at", pool.UpdatedAt.Format(time.RFC3339))
 	_ = d.Set("status", pool.Status)
 	_ = d.Set("kubelet_args", flattenKubeletArgs(pool.KubeletArgs))
-	_ = d.Set("region", region)
-	_ = d.Set("zone", pool.Zone)
+	_ = d.Set("region", pool.Region.String())
+	_ = d.Set("zone", pool.Zone.String())
 	_ = d.Set("upgrade_policy", poolUpgradePolicyFlatten(pool))
 	_ = d.Set("public_ip_disabled", pool.PublicIPDisabled)
 	_ = d.Set("security_group_id", pool.SecurityGroupID)

internal/services/marketplace/image_data_source.go

@@ -57,7 +57,7 @@ func DataSourceMarketplaceImageRead(ctx context.Context, d *schema.ResourceData,
 
 	zonedID := datasource.NewZonedID(image.ID, zone)
 	d.SetId(zonedID)
-	_ = d.Set("zone", zone)
+	_ = d.Set("zone", image.Zone.String())
 	_ = d.Set("label", image.Label)
 	_ = d.Set("instance_type", d.Get("instance_type").(string))
 	_ = d.Set("image_type", image.Type)

internal/services/mnq/nats_credentials.go

@@ -93,7 +93,7 @@ func ResourceMNQNatsCredentialsRead(ctx context.Context, d *schema.ResourceData,
 
 	_ = d.Set("account_id", credentials.NatsAccountID)
 	_ = d.Set("name", credentials.Name)
-	_ = d.Set("region", region)
+	_ = d.Set("region", credentials)
 
 	return nil
 }

internal/services/rdb/instance.go

@@ -663,7 +663,7 @@ func ResourceRdbInstanceRead(ctx context.Context, d *schema.ResourceData, m any)
 	}
 
 	_ = d.Set("read_replicas", []string{})
-	_ = d.Set("region", string(region))
+	_ = d.Set("region", res.Region.String())
 	_ = d.Set("organization_id", res.OrganizationID)
 	_ = d.Set("project_id", res.ProjectID)
 

internal/services/rdb/read_replica.go

@@ -225,9 +225,8 @@ func ResourceRdbReadReplicaRead(ctx context.Context, d *schema.ResourceData, m a
 	_ = d.Set("direct_access", directAccess)
 	_ = d.Set("private_network", privateNetwork)
 
-	regionStr := region.String()
 	_ = d.Set("same_zone", rr.SameZone)
-	_ = d.Set("region", regionStr)
+	_ = d.Set("region", rr.Region.String())
 	_ = d.Set("instance_id", regional.NewIDString(region, rr.InstanceID))
 
 	return nil

internal/services/rdb/snapshots.go

@@ -160,7 +160,7 @@ func ResourceRdbSnapshotRead(ctx context.Context, d *schema.ResourceData, meta a
 		_ = d.Set("size", int(*res.Size))
 	}
 
-	_ = d.Set("region", region)
+	_ = d.Set("region", res.Region.String())
 
 	return nil
 }

internal/services/secret/secret.go

@@ -266,7 +266,7 @@ func ResourceSecretRead(ctx context.Context, d *schema.ResourceData, m any) diag
 	_ = d.Set("updated_at", types.FlattenTime(secretResponse.UpdatedAt))
 	_ = d.Set("status", secretResponse.Status.String())
 	_ = d.Set("version_count", int(versions.TotalCount))
-	_ = d.Set("region", string(region))
+	_ = d.Set("region", secretResponse.Region.String())
 	_ = d.Set("project_id", secretResponse.ProjectID)
 	_ = d.Set("path", secretResponse.Path)
 	_ = d.Set("protected", secretResponse.Protected)

internal/services/secret/version.go

@@ -131,7 +131,7 @@
 	_ = d.Set("created_at", types.FlattenTime(secretResponse.CreatedAt))
 	_ = d.Set("updated_at", types.FlattenTime(secretResponse.UpdatedAt))
 	_ = d.Set("status", secretResponse.Status.String())
-	_ = d.Set("region", string(region))
+	_ = d.Set("region", secretResponse.)
 
 	return nil
 }

internal/services/tem/domain.go

@@ -259,7 +259,7 @@ func ResourceDomainRead(ctx context.Context, d *schema.ResourceData, m any) diag
 	_ = d.Set("smtps_port_alternative", tem.SMTPSPortAlternative)
 	_ = d.Set("mx_blackhole", tem.MXBlackhole)
 	_ = d.Set("reputation", flattenDomainReputation(domain.Reputation))
-	_ = d.Set("region", string(region))
+	_ = d.Set("region", domain.Region.String())
 	_ = d.Set("project_id", domain.ProjectID)
 	_ = d.Set("smtps_auth_user", domain.ProjectID)
 

internal/services/vpc/vpc.go

@@ -142,7 +142,7 @@ func ResourceVPCRead(ctx context.Context, d *schema.ResourceData, m any) diag.Di
 	_ = d.Set("is_default", res.IsDefault)
 	_ = d.Set("enable_routing", res.RoutingEnabled)
 	_ = d.Set("enable_custom_routes_propagation", res.CustomRoutesPropagationEnabled)
-	_ = d.Set("region", region)
+	_ = d.Set("region", res.Region.String())
 
 	if len(res.Tags) > 0 {
 		_ = d.Set("tags", res.Tags)

internal/services/vpcgw/dhcp_reservation.go

@@ -149,7 +149,7 @@ func ResourceVPCPublicGatewayDHCPReservationRead(ctx context.Context, d *schema.
 	_ = d.Set("gateway_network_id", zonal.NewIDString(zone, entry.GatewayNetworkID))
 	_ = d.Set("created_at", entry.CreatedAt.Format(time.RFC3339))
 	_ = d.Set("updated_at", entry.UpdatedAt.Format(time.RFC3339))
-	_ = d.Set("zone", zone)
+	_ = d.Set("zone", entry.Zone.String())
 
 	return nil
 }

internal/services/vpcgw/ip.go

@@ -124,7 +124,7 @@ func ResourceIPRead(ctx context.Context, d *schema.ResourceData, m any) diag.Dia
 	_ = d.Set("project_id", ip.ProjectID)
 	_ = d.Set("created_at", ip.CreatedAt.Format(time.RFC3339))
 	_ = d.Set("updated_at", ip.UpdatedAt.Format(time.RFC3339))
-	_ = d.Set("zone", zone)
+	_ = d.Set("zone", ip.Zone.String())
 	_ = d.Set("tags", ip.Tags)
 	_ = d.Set("reverse", ip.Reverse)
 

internal/services/vpcgw/ip_reverse.go

@@ -104,7 +104,7 @@ func ResourceVPCPublicGatewayIPReverseDNSRead(ctx context.Context, d *schema.Res
 		return diag.FromErr(err)
 	}
 
-	_ = d.Set("zone", string(zone))
+	_ = d.Set("zone", res.Zonestring(zone))
 	_ = d.Set("reverse", res.Reverse)
 
 	return nil

internal/services/vpcgw/pat_rule.go

@@ -157,7 +157,7 @@ func ResourceVPCPublicGatewayPATRuleRead(ctx context.Context, d *schema.Resource
 	_ = d.Set("private_port", int(patRule.PrivatePort))
 	_ = d.Set("public_port", int(patRule.PublicPort))
 	_ = d.Set("protocol", patRule.Protocol.String())
-	_ = d.Set("zone", zone)
+	_ = d.Set("zone", patRule.Zone.String())
 
 	return nil
 }

internal/services/webhosting/webhosting.go

@@ -325,7 +325,7 @@
 	_ = d.Set("dns_status", webhostingResponse.DNSStatus.String()) //nolint:staticcheck
 	_ = d.Set("cpanel_urls", flattenHostingCpanelUrls(webhostingResponse.Platform.ControlPanel.URLs))
 	_ = d.Set("username", webhostingResponse.User.Username)
-	_ = d.Set("region", string(region))
+	_ = d.Set("region", webhostingResponse.Regionstring(region))
 	_ = d.Set("organization_id", "")
 	_ = d.Set("project_id", webhostingResponse.ProjectID)