feat(state_upgarder) refactor state upgrader for dns_record, page_rule

and load_balancer_pool

Author: vaishakdinesh

.github/workflows/migration-tests.yml

@@ -28,6 +28,7 @@ env:
   CLOUDFLARE_HYPERDRIVE_DATABASE_HOSTNAME: ${{ secrets.CLOUDFLARE_HYPERDRIVE_DATABASE_HOSTNAME }}
   TF_ACC: 1
   TF_MIGRATE_BINARY_PATH: ${{ github.workspace }}/tf-migrate
+  TF_MIG_TEST: true
 
 jobs:
   migration-tests:

internal/acctest/acctest.go

@@ -917,24 +917,6 @@ func RunMigrationV2Command(t *testing.T, v4Config string, tmpDir string, sourceV
 		t.Fatalf("tf-migrate binary not found at %s. Please set TF_MIGRATE_BINARY_PATH or ensure the binary is built.", migratorPath)
 	}
 
-	// Find state file in tmpDir
-	entries, err := os.ReadDir(tmpDir)
-	var stateDir string
-	if err != nil {
-		t.Logf("Failed to read test directory: %v", err)
-	} else {
-		for _, entry := range entries {
-			if entry.IsDir() {
-				inner_entries, _ := os.ReadDir(filepath.Join(tmpDir, entry.Name()))
-				for _, inner_entry := range inner_entries {
-					if inner_entry.Name() == "terraform.tfstate" {
-						stateDir = filepath.Join(tmpDir, entry.Name())
-					}
-				}
-			}
-		}
-	}
-
 	// Build the command
 	args := []string{
 		"migrate",
@@ -943,11 +925,6 @@ func RunMigrationV2Command(t *testing.T, v4Config string, tmpDir string, sourceV
 		"--target-version", targetVersion,
 	}
 
-	// Add state file argument if found
-	if stateDir != "" {
-		args = append(args, "--state-file", filepath.Join(stateDir, "terraform.tfstate"))
-	}
-
 	// Add debug logging if TF_LOG is set
 	if strings.ToLower(os.Getenv("TF_LOG")) == "debug" {
 		args = append(args, "--log-level", "debug")
@@ -1099,6 +1076,15 @@ func MigrationV2TestStepWithPlan(t *testing.T, v4Config string, tmpDir string, e
 	return []resource.TestStep{migrationStep, planStep, validationStep}
 }
 
+// InferMigrationVersions determines source and target versions from test provider version.
+// Returns ("v4", "v5") for v4.x versions, ("v5", "v5") for v5.x versions.
+func InferMigrationVersions(testVersion string) (source, target string) {
+	if strings.HasPrefix(testVersion, "5.") {
+		return "v5", "v5"
+	}
+	return "v4", "v5"
+}
+
 // MigrationTestStep creates a test step that runs the migration command and validates with v5 provider
 func MigrationTestStep(t *testing.T, v4Config string, tmpDir string, exactVersion string, stateChecks []statecheck.StateCheck) resource.TestStep {
 	// Choose the appropriate plan check based on the version
@@ -1166,8 +1152,14 @@ func MigrationV2TestStep(t *testing.T, v4Config string, tmpDir string, exactVers
 	return resource.TestStep{
 		PreConfig: func() {
 			WriteOutConfig(t, v4Config, tmpDir)
-			debugLogf(t, "Running migration command for version: %s (%s -> %s)", exactVersion, sourceVersion, targetVersion)
-			RunMigrationV2Command(t, v4Config, tmpDir, sourceVersion, targetVersion)
+			// Only run migration command if source version is v4
+			// v5→v5 "migrations" only test StateUpgrader idempotency, no config changes needed
+			if sourceVersion == "v4" {
+				debugLogf(t, "Running migration command for version: %s (%s -> %s)", exactVersion, sourceVersion, targetVersion)
+				RunMigrationV2Command(t, v4Config, tmpDir, sourceVersion, targetVersion)
+			} else {
+				debugLogf(t, "Skipping migration command for version: %s (%s -> %s) - testing StateUpgrader only", exactVersion, sourceVersion, targetVersion)
+			}
 		},
 		ProtoV6ProviderFactories: TestAccProtoV6ProviderFactories,
 		ConfigDirectory:          config.StaticDirectory(tmpDir),

internal/services/dns_record/migration/v500/handler.go

@@ -0,0 +1,66 @@
+package v500
+
+import (
+	"context"
+
+	"github.com/hashicorp/terraform-plugin-framework/resource"
+	"github.com/hashicorp/terraform-plugin-log/tflog"
+)
+
+// UpgradeFromV0 handles state upgrades from earlier v500 versions (schema_version=0) to current v500.
+// This is a no-op upgrade since the schema is compatible - just copy state through.
+//
+// Why this exists: Terraform requires explicit upgraders to be defined for version tracking,
+// even when the schema is identical. This ensures the schema_version is updated in the statefile.
+func UpgradeFromV0(
+	ctx context.Context,
+	req resource.UpgradeStateRequest,
+	resp *resource.UpgradeStateResponse,
+) {
+	// No-op upgrade: schema is compatible, just copy raw state through
+	// We use the raw state value directly to avoid issues with custom field type serialization
+	resp.State.Raw = req.State.Raw
+}
+
+// UpgradeFromV4 handles state upgrades from v5 releases with schema_version=4 to current v500.
+// This is a no-op upgrade since we only changed the version number from 4 to 500.
+// All v5.x.x releases up to 5.16.0 used schema_version=4 for cloudflare_dns_record.
+func UpgradeFromV4(
+	ctx context.Context,
+	req resource.UpgradeStateRequest,
+	resp *resource.UpgradeStateResponse,
+) {
+	tflog.Info(ctx, "Upgrading DNS record state from schema_version=4 to schema_version=500")
+	// No-op upgrade: schema is identical, just copy raw state through
+	// We use the raw state value directly to avoid issues with custom field type serialization
+	resp.State.Raw = req.State.Raw
+}
+
+// UpgradeFromLegacyV3 handles state upgrades from the legacy cloudflare_record resource to cloudflare_dns_record.
+// This is triggered when users manually run `terraform state mv cloudflare_record.x cloudflare_dns_record.x`
+// (Terraform < 1.8), which preserves the source schema_version=3 from the legacy provider.
+//
+// Note: schema_version=3 was the final schema version of cloudflare_record in the legacy (SDKv2) provider
+// before it was deprecated. The state structure matches SourceCloudflareRecordModel.
+func UpgradeFromLegacyV3(ctx context.Context, req resource.UpgradeStateRequest, resp *resource.UpgradeStateResponse) {
+	tflog.Info(ctx, "Upgrading DNS record state from legacy cloudflare_record (schema_version=3)")
+
+	// Parse the state (schema_version=3, source resource type)
+	var sourceState SourceCloudflareRecordModel
+	resp.Diagnostics.Append(req.State.Get(ctx, &sourceState)...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	// Transform to target
+	targetState, diags := Transform(ctx, sourceState)
+	resp.Diagnostics.Append(diags...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	// Set the upgraded state
+	resp.Diagnostics.Append(resp.State.Set(ctx, targetState)...)
+
+	tflog.Info(ctx, "State upgrade from legacy cloudflare_record completed successfully")
+}