unit test StateReplaceProviderAction

minamijoyo · Aug 24, 2023 · d522a4b · d522a4b
1 parent 64b3a3a
commit d522a4b
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/tfmigrate/state_action_test.go b/tfmigrate/state_action_test.go
@@ -39,6 +39,33 @@ func TestNewStateActionFromString(t *testing.T) {
 			want:   nil,
 			ok:     false,
 		},
+		{
+			desc:   "replace-provider action (0 args)",
+			cmdStr: "replace-provider",
+			want:   nil,
+			ok:     false,
+		},
+		{
+			desc:   "replace-provider action (1 args)",
+			cmdStr: "replace-provider registry.terraform.io/hashicorp/null",
+			want:   nil,
+			ok:     false,
+		},
+		{
+			desc:   "replace-provider action (valid)",
+			cmdStr: "replace-provider registry.terraform.io/hashicorp/null registry.tfmigrate.io/hashicorp/null",
+			want: &StateReplaceProviderAction{
+				source:      "registry.terraform.io/hashicorp/null",
+				destination: "registry.tfmigrate.io/hashicorp/null",
+			},
+			ok: true,
+		},
+		{
+			desc:   "replace-provider action (3 args)",
+			cmdStr: "replace-provider registry.terraform.io/hashicorp/null registry.tfmigrate.io/hashicorp/null registry.tfmigrate.io/hashicorp/null",
+			want:   nil,
+			ok:     false,
+		},
 		{
 			desc:   "xmv action (valid)",
 			cmdStr: "xmv null_resource.* null_resource.$1",