Adding Taint/Untaint command support

CHANGELOG.md

@@ -14,6 +14,7 @@ FEATURES:
  - Add support for `workspace delete` command [GH-212]
  - Add support for `workspace show` command [GH-245]
  - Add support for `force-unlock` command [GH-223]
+ - Add support for `taint`/`untaint` command [GH-251]
 
 # 0.15.0 (October 05, 2021)
 

tfexec/internal/e2etest/taint_test.go

@@ -0,0 +1,29 @@
+package e2etest
+
+import (
+	"context"
+	"testing"
+
+	"github.com/hashicorp/go-version"
+
+	"github.com/hashicorp/terraform-exec/tfexec"
+)
+
+func TestTaint(t *testing.T) {
+	runTest(t, "basic", func(t *testing.T, tfv *version.Version, tf *tfexec.Terraform) {
+		err := tf.Init(context.Background())
+		if err != nil {
+			t.Fatalf("error running Init in test directory: %s", err)
+		}
+
+		err = tf.Apply(context.Background())
+		if err != nil {
+			t.Fatalf("error running Apply: %s", err)
+		}
+
+		err = tf.Taint(context.Background(), "null_resource.foo")
+		if err != nil {
+			t.Fatalf("error running Taint: %s", err)
+		}
+	})
+}

tfexec/internal/e2etest/untaint_test.go

@@ -0,0 +1,34 @@
+package e2etest
+
+import (
+	"context"
+	"testing"
+
+	"github.com/hashicorp/go-version"
+
+	"github.com/hashicorp/terraform-exec/tfexec"
+)
+
+func TestUntaint(t *testing.T) {
+	runTest(t, "basic", func(t *testing.T, tfv *version.Version, tf *tfexec.Terraform) {
+		err := tf.Init(context.Background())
+		if err != nil {
+			t.Fatalf("error running Init in test directory: %s", err)
+		}
+
+		err = tf.Apply(context.Background())
+		if err != nil {
+			t.Fatalf("error running Apply: %s", err)
+		}
+
+		err = tf.Taint(context.Background(), "null_resource.foo")
+		if err != nil {
+			t.Fatalf("error running Taint: %s", err)
+		}
+
+		err = tf.Untaint(context.Background(), "null_resource.foo")
+		if err != nil {
+			t.Fatalf("error running UnTaint: %s", err)
+		}
+	})
+}

tfexec/output.go

@@ -34,6 +34,7 @@ type OutputMeta struct {
 
 // Output represents the terraform output subcommand.
 func (tf *Terraform) Output(ctx context.Context, opts ...OutputOption) (map[string]OutputMeta, error) {
+
 	outputCmd := tf.outputCmd(ctx, opts...)
 
 	outputs := map[string]OutputMeta{}
@@ -51,7 +52,6 @@ func (tf *Terraform) outputCmd(ctx context.Context, opts ...OutputOption) *exec.
 	for _, o := range opts {
 		o.configureOutput(&c)
 	}
-
 	args := []string{"output", "-no-color", "-json"}
 
 	// string opts: only pass if set

tfexec/taint.go

@@ -0,0 +1,51 @@
+package tfexec
+
+import (
+	"context"
+	"fmt"
+	"os/exec"
+)
+
+type taintConfig struct {
+	state string
+}
+
+var defaultTaintOptions = taintConfig{}
+
+// TaintOption represents options used in the Taint method.
+type TaintOption interface {
+	configureTaint(*taintConfig)
+}
+
+func (opt *StateOption) configureTaint(conf *taintConfig) {
+	conf.state = opt.path
+}
+
+// Taint represents the terraform taint subcommand.
+func (tf *Terraform) Taint(ctx context.Context, address string, opts ...TaintOption) error {
+	err := tf.compatible(ctx, tf0_4_1, nil)
+	if err != nil {
+		return fmt.Errorf("taint was first introduced in Terraform 0.4.1: %w", err)
+	}
+	taintCmd := tf.taintCmd(ctx, address, opts...)
+	return tf.runTerraformCmd(ctx, taintCmd)
+}
+
+func (tf *Terraform) taintCmd(ctx context.Context, address string, opts ...TaintOption) *exec.Cmd {
+	c := defaultTaintOptions
+
+	for _, o := range opts {
+		o.configureTaint(&c)
+	}
+
+	args := []string{"taint", "-no-color"}
+
+	// string opts: only pass if set
+	if c.state != "" {
+		args = append(args, "-state="+c.state)
+	}
+
+	args = append(args, address)
+
+	return tf.buildTerraformCmd(ctx, nil, args...)
+}

tfexec/taint_test.go

@@ -0,0 +1,41 @@
+package tfexec
+
+import (
+	"context"
+	"testing"
+
+	"github.com/hashicorp/terraform-exec/tfexec/internal/testutil"
+)
+
+func TestTaintCmd(t *testing.T) {
+	td := t.TempDir()
+
+	tf, err := NewTerraform(td, tfVersion(t, testutil.Latest013))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// empty env, to avoid environ mismatch in testing
+	tf.SetEnv(map[string]string{})
+
+	t.Run("defaults", func(t *testing.T) {
+		taintCmd := tf.taintCmd(context.Background(), "aws_instance.foo")
+
+		assertCmd(t, []string{
+			"taint",
+			"-no-color",
+			"aws_instance.foo",
+		}, nil, taintCmd)
+	})
+
+	t.Run("override all defaults", func(t *testing.T) {
+		taintCmd := tf.taintCmd(context.Background(), "aws_instance.foo", State("teststate"))
+
+		assertCmd(t, []string{
+			"taint",
+			"-no-color",
+			"-state=teststate",
+			"aws_instance.foo",
+		}, nil, taintCmd)
+	})
+}

tfexec/untaint.go

@@ -0,0 +1,51 @@
+package tfexec
+
+import (
+	"context"
+	"fmt"
+	"os/exec"
+)
+
+type untaintConfig struct {
+	state string
+}
+
+var defaultUnTaintOptions = untaintConfig{}
+
+// OutputOption represents options used in the Output method.
+type UnTaintOption interface {
+	configureUnTaint(*untaintConfig)
+}
+
+func (opt *StateOption) configureUnTaint(conf *untaintConfig) {
+	conf.state = opt.path
+}
+
+// Untaint represents the terraform untaint subcommand.
+func (tf *Terraform) Untaint(ctx context.Context, address string, opts ...UnTaintOption) error {
+	err := tf.compatible(ctx, tf0_6_13, nil)
+	if err != nil {
+		return fmt.Errorf("untaint was first introduced in Terraform 0.6.13: %w", err)
+	}
+	unTaintCmd := tf.unTaintCmd(ctx, address, opts...)
+	return tf.runTerraformCmd(ctx, unTaintCmd)
+}
+
+func (tf *Terraform) unTaintCmd(ctx context.Context, address string, opts ...UnTaintOption) *exec.Cmd {
+	c := defaultUnTaintOptions
+
+	for _, o := range opts {
+		o.configureUnTaint(&c)
+	}
+
+	args := []string{"untaint", "-no-color"}
+
+	// string opts: only pass if set
+	if c.state != "" {
+		args = append(args, "-state="+c.state)
+	}
+
+	args = append(args, address)
+
+	return tf.buildTerraformCmd(ctx, nil, args...)
+}

tfexec/untaint_test.go

@@ -0,0 +1,41 @@
+package tfexec
+
+import (
+	"context"
+	"testing"
+
+	"github.com/hashicorp/terraform-exec/tfexec/internal/testutil"
+)
+
+func TestUnTaintCmd(t *testing.T) {
+	td := t.TempDir()
+
+	tf, err := NewTerraform(td, tfVersion(t, testutil.Latest013))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// empty env, to avoid environ mismatch in testing
+	tf.SetEnv(map[string]string{})
+
+	t.Run("defaults", func(t *testing.T) {
+		unTaintCmd := tf.unTaintCmd(context.Background(), "aws_instance.foo")
+
+		assertCmd(t, []string{
+			"untaint",
+			"-no-color",
+			"aws_instance.foo",
+		}, nil, unTaintCmd)
+	})
+
+	t.Run("override all defaults", func(t *testing.T) {
+		unTaintCmd := tf.unTaintCmd(context.Background(), "aws_instance.foo", State("teststate"))
+
+		assertCmd(t, []string{
+			"untaint",
+			"-no-color",
+			"-state=teststate",
+			"aws_instance.foo",
+		}, nil, unTaintCmd)
+	})
+}

tfexec/version.go

@@ -14,6 +14,8 @@ import (
 )
 
 var (
+	tf0_4_1  = version.Must(version.NewVersion("0.4.1"))
+	tf0_6_13 = version.Must(version.NewVersion("0.6.13"))
 	tf0_7_7  = version.Must(version.NewVersion("0.7.7"))
 	tf0_10_0 = version.Must(version.NewVersion("0.10.0"))
 	tf0_12_0 = version.Must(version.NewVersion("0.12.0"))