r/advanced_threat_protection: adding import time validation of the re…

…source id
hashicorp · Jan 14, 2021 · 55826a2 · 55826a2
1 parent 09bd111
commit 55826a2
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/azurerm/internal/services/securitycenter/advanced_threat_protection_resource.go b/azurerm/internal/services/securitycenter/advanced_threat_protection_resource.go
@@ -11,6 +11,7 @@ import (
 	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/helpers/tf"
 	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/clients"
 	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/securitycenter/parse"
+	azSchema "github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/tf/schema"
 	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/timeouts"
 	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/utils"
 )
@@ -22,9 +23,10 @@ func resourceAdvancedThreatProtection() *schema.Resource {
 		Update: resourceAdvancedThreatProtectionCreateUpdate,
 		Delete: resourceAdvancedThreatProtectionDelete,
 
-		Importer: &schema.ResourceImporter{
-			State: schema.ImportStatePassthrough,
-		},
+		Importer: azSchema.ValidateResourceIDPriorToImport(func(id string) error {
+			_, err := parse.AdvancedThreatProtectionID(id)
+			return err
+		}),
 
 		Timeouts: &schema.ResourceTimeout{
 			Create: schema.DefaultTimeout(30 * time.Minute),