test: move ImportState scheduled rebalancing test cases to the end

Author: vladklokun

castai/resource_rebalancing_schedule_test.go

@@ -23,19 +23,6 @@ func TestAccResourceRebalancingSchedule_basic(t *testing.T) {
 					resource.TestCheckResourceAttr("castai_rebalancing_schedule.test", "schedule.0.cron", "5 4 * * *"),
 				),
 			},
-			{
-				// import by ID
-				ImportState:       true,
-				ResourceName:      "castai_rebalancing_schedule.test",
-				ImportStateVerify: true,
-			},
-			{
-				// import by name
-				ImportState:       true,
-				ResourceName:      "castai_rebalancing_schedule.test",
-				ImportStateId:     rName,
-				ImportStateVerify: true,
-			},
 			{
 				// test edits
 				Config: makeUpdatedRebalancingScheduleConfig(rName + " renamed"),
@@ -56,6 +43,24 @@ func TestAccResourceRebalancingSchedule_basic(t *testing.T) {
 					resource.TestCheckResourceAttr("castai_rebalancing_schedule.test", "launch_configuration.0.rebalancing_min_nodes", "0"),
 				),
 			},
+			// We keep the ImportState test cases at the end so they will test any newly added fields.
+			// This way it will also verify that after importing the state the output of `tf plan` is empty.
+			{
+				// Import state by ID
+				ImportState:       true,
+				ResourceName:      "castai_rebalancing_schedule.test",
+				ImportStateVerify: true,
+			},
+			{
+				// Import state by name
+				ImportState:  true,
+				ResourceName: "castai_rebalancing_schedule.test",
+				// Make sure to use the resource (rebalancing schedule) name from the latest prior step that changes the resource,
+				// otherwise this step will not to see the changes made after the State ID you provided,
+				// and this step will fail.
+				ImportStateId:     rName + " min_nodes_zero",
+				ImportStateVerify: true,
+			},
 		},
 	})
 }