fix test

Author: kota65535

internal/provider/branch_resource_migration.go

@@ -178,6 +178,7 @@ type branchResourceModelV0 struct {
 	ShardCount                  types.Float64 `tfsdk:"shard_count"`
 	Sharded                     types.Bool    `tfsdk:"sharded"`
 	UpdatedAt                   types.String  `tfsdk:"updated_at"`
+	SeedData                    types.String  `tfsdk:"seed_data"`
 }
 
 // branchSchemaV0 defines the schema for version 0.
@@ -279,6 +280,13 @@ func branchSchemaV0() *schema.Schema {
 				Description: "When the branch was last updated.",
 				Computed:    true,
 			},
+			"seed_data": schema.StringAttribute{
+				Description: "Seed data using the Data Branching® feature. Valid value is `last_successful_backup`",
+				Optional:    true,
+				PlanModifiers: []planmodifier.String{
+					stringplanmodifier.RequiresReplace(),
+				},
+			},
 		},
 	}
 }
@@ -312,6 +320,7 @@ func upgradeBranchStateV0toCurrent(ctx context.Context, req resource.UpgradeStat
 		ShardCount:                  priorStateData.ShardCount,
 		Sharded:                     priorStateData.Sharded,
 		UpdatedAt:                   priorStateData.UpdatedAt,
+		SeedData:                    priorStateData.SeedData,
 	}
 
 	resp.Diagnostics.Append(resp.State.Set(ctx, upgradedStateData)...)

internal/provider/branch_resource_migration_test.go

@@ -42,6 +42,7 @@ func TestUpgradeBranchStateV0toCurrent(t *testing.T) {
 					"shard_count":                    tftypes.NewValue(tftypes.Number, 1),
 					"sharded":                        tftypes.NewValue(tftypes.Bool, false),
 					"updated_at":                     tftypes.NewValue(tftypes.String, "2023-01-01T03:00:00Z"),
+					"seed_data":                      tftypes.NewValue(tftypes.String, "last_successful_backup"),
 				}),
 			Schema: branchSchemaV0(),
 		},
@@ -78,6 +79,7 @@ func TestUpgradeBranchStateV0toCurrent(t *testing.T) {
 					"shard_count":                    tftypes.NewValue(tftypes.Number, 1),
 					"sharded":                        tftypes.NewValue(tftypes.Bool, false),
 					"updated_at":                     tftypes.NewValue(tftypes.String, "2023-01-01T03:00:00Z"),
+					"seed_data":                      tftypes.NewValue(tftypes.String, "last_successful_backup"),
 				}),
 			Schema: branchSchemaV1(),
 		},