drone-terraform/plugin_test.go

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")
			os.Setenv("TF_VAR_BASE64", "dGVzdA==")

			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")
			g.Assert(os.Getenv("TF_VAR_base64")).Equal("dGVzdA==")
		})
	})
}