package main import ( "os" "os/exec" "reflect" "testing" . "github.com/franela/goblin" ) func Test_destroyCommand(t *testing.T) { type args struct { config Config } tests := []struct { name string args args want *exec.Cmd }{ { "default", args{config: Config{}}, exec.Command("terraform", "destroy", "-force"), }, { "with targets", args{config: Config{Targets: []string{"target1", "target2"}}}, exec.Command("terraform", "destroy", "-target=target1", "-target=target2", "-force"), }, { "with parallelism", args{config: Config{Parallelism: 5}}, exec.Command("terraform", "destroy", "-parallelism=5", "-force"), }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { if got := destroyCommand(tt.args.config); !reflect.DeepEqual(got, tt.want) { t.Errorf("destroyCommand() = %v, want %v", got, tt.want) } }) } } func Test_applyCommand(t *testing.T) { type args struct { config Config } tests := []struct { name string args args want *exec.Cmd }{ { "default", args{config: Config{}}, exec.Command("terraform", "apply", "plan.tfout"), }, { "with targets", args{config: Config{Targets: []string{"target1", "target2"}}}, exec.Command("terraform", "apply", "--target", "target1", "--target", "target2", "plan.tfout"), }, { "with parallelism", args{config: Config{Parallelism: 5}}, exec.Command("terraform", "apply", "-parallelism=5", "plan.tfout"), }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { if got := applyCommand(tt.args.config); !reflect.DeepEqual(got, tt.want) { t.Errorf("applyCommand() = %v, want %v", got, tt.want) } }) } } func Test_terraformCommand(t *testing.T) { type args struct { config Config } tests := []struct { name string args args want *exec.Cmd }{ { "default", args{config: Config{}}, exec.Command("terraform", "apply", "plan.tfout"), }, { "destroy", args{config: Config{Destroy: true}}, exec.Command("terraform", "destroy", "-force"), }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { if got := terraformCommand(tt.args.config); !reflect.DeepEqual(got, tt.want) { t.Errorf("terraformCommand() = %v, want %v", got, tt.want) } }) } } func Test_planCommand(t *testing.T) { type args struct { config Config } tests := []struct { name string args args want *exec.Cmd }{ { "default", args{config: Config{}}, exec.Command("terraform", "plan", "-out=plan.tfout"), }, { "destroy", args{config: Config{Destroy: true}}, exec.Command("terraform", "plan", "-destroy"), }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { if got := planCommand(tt.args.config); !reflect.DeepEqual(got, tt.want) { t.Errorf("planCommand() = %v, want %v", got, tt.want) } }) } } func TestPlugin(t *testing.T) { g := Goblin(t) g.Describe("CopyTfEnv", func() { g.It("Should create copies of TF_VAR_ to lowercase", func() { // Set some initial TF_VAR_ that are uppercase os.Setenv("TF_VAR_SOMETHING", "some value") os.Setenv("TF_VAR_SOMETHING_ELSE", "some other value") CopyTfEnv() // Make sure new env vars exist with proper values g.Assert(os.Getenv("TF_VAR_something")).Equal("some value") g.Assert(os.Getenv("TF_VAR_something_else")).Equal("some other value") }) }) }