azurerm_static_site_custom_domain - set validation_token in create instead of read

internal/services/web/static_site_custom_domain_resource.go

@@ -9,6 +9,7 @@ import (
 	"time"
 
 	"github.com/Azure/azure-sdk-for-go/services/web/mgmt/2021-02-01/web" // nolint: staticcheck
+	"github.com/hashicorp/go-azure-helpers/lang/pointer"
 	"github.com/hashicorp/terraform-provider-azurerm/helpers/tf"
 	"github.com/hashicorp/terraform-provider-azurerm/internal/clients"
 	"github.com/hashicorp/terraform-provider-azurerm/internal/services/web/parse"
@@ -161,6 +162,15 @@ func resourceStaticSiteCustomDomainCreateOrUpdate(d *pluginsdk.ResourceData, met
 		}
 	}
 
+	// we set `validation_token` into state here since it's removed from the API once it's been validated
+	// setting it in the read would overwrite the value with an empty string and cause a diff, since this
+	// is not a user specifiable field we don't need to be concerned with it at import time either
+	resp, err := client.GetStaticSiteCustomDomain(ctx, staticSiteId.ResourceGroup, id.StaticSiteName, id.CustomDomainName)
+	if err != nil {
+		return fmt.Errorf("retrieving %s: %+v", id, err)
+	}
+	d.Set("validation_token", pointer.From(resp.ValidationToken))
+
 	d.SetId(id.ID())
 
 	return resourceStaticSiteCustomDomainRead(d, meta)
@@ -188,10 +198,6 @@ func resourceStaticSiteCustomDomainRead(d *pluginsdk.ResourceData, meta interfac
 	d.Set("domain_name", id.CustomDomainName)
 	d.Set("static_site_id", parse.NewStaticSiteID(id.SubscriptionId, id.ResourceGroup, id.StaticSiteName).ID())
 
-	if props := resp.StaticSiteCustomDomainOverviewARMResourceProperties; props != nil {
-		d.Set("validation_token", resp.ValidationToken)
-	}
-
 	return nil
 }
 

internal/services/web/static_site_resource_custom_domain_test.go

@@ -30,7 +30,7 @@ func TestAccAzureStaticSiteCustomDomain_basic(t *testing.T) {
 				check.That(data.ResourceName).Key("validation_token").Exists(),
 			),
 		},
-		data.ImportStep("validation_type"),
+		data.ImportStep("validation_type", "validation_token"),
 	})
 }