Fix Missing Resource Identity After Update error for Terraform < 1.12 (#1024)

Author: ms-henglu

CHANGELOG.md

@@ -1,4 +1,9 @@
-## v2.8.0 (Unreleased)
+## v2.9.0 (Unreleased)
+
+BUG FIXES:
+- Fix `Missing Resource Identity After Update` error for Terraform versions below 1.12 (GH-1023).
+
+## v2.8.0
 
 FEATURES:
 - **New Provider Function**: snake2camel

internal/services/azapi_resource.go

@@ -723,6 +723,12 @@ func (r *AzapiResource) Update(ctx context.Context, request resource.UpdateReque
 	if response.Diagnostics.Append(request.State.Get(ctx, &state)...); response.Diagnostics.HasError() {
 		return
 	}
+	// This is a workaround for the issue: https://github.com/Azure/terraform-provider-azapi/issues/1023
+	// The issue is that the identity is not set in the response when the resource is updated.
+	// This issue only happens when the terraform version is below 1.12.
+	if !state.ID.IsNull() {
+		response.Diagnostics.Append(response.Identity.SetAttribute(ctx, path.Root("id"), state.ID)...)
+	}
 	if skip.CanSkipExternalRequest(plan, state, "update") {
 		response.Diagnostics.Append(response.State.Set(ctx, plan)...)
 		tflog.Debug(ctx, "azapi_resource.CreateUpdate skipping external request as no unskippable changes were detected")