2017-08-08 21:16:20 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2017-05-30 14:48:34 +00:00
|
|
|
"os"
|
2017-08-08 21:16:20 +00:00
|
|
|
"os/exec"
|
2019-11-10 00:18:12 +00:00
|
|
|
"testing"
|
2017-05-30 14:48:34 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
. "github.com/franela/goblin"
|
2017-08-08 21:16:20 +00:00
|
|
|
)
|
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
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==")
|
|
|
|
})
|
|
|
|
})
|
|
|
|
|
2019-12-02 14:39:22 +00:00
|
|
|
g.Describe("credsSet", func() {
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
2020-02-09 22:55:31 +00:00
|
|
|
args map[string]string
|
2019-12-02 14:39:22 +00:00
|
|
|
want bool
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
"Should return true when all credentials were set",
|
2020-02-09 22:55:31 +00:00
|
|
|
map[string]string{"AWS_ACCESS_KEY_ID": "x", "AWS_SECRET_ACCESS_KEY": "x", "AWS_SESSION_TOKEN": "x"},
|
2019-12-02 14:39:22 +00:00
|
|
|
true,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"Should return false when access key id is missing",
|
2020-02-09 22:55:31 +00:00
|
|
|
map[string]string{"AWS_SECRET_ACCESS_KEY": "x", "AWS_SESSION_TOKEN": "x"},
|
2019-12-02 14:39:22 +00:00
|
|
|
false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"Should return false when secret access key is missing",
|
2020-02-09 22:55:31 +00:00
|
|
|
map[string]string{"AWS_ACCESS_KEY_ID": "x", "AWS_SESSION_TOKEN": "x"},
|
2019-12-02 14:39:22 +00:00
|
|
|
false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"Should return false when session token is missing",
|
2020-02-09 22:55:31 +00:00
|
|
|
map[string]string{"AWS_ACCESS_KEY_ID": "x", "AWS_SECRET_ACCESS_KEY": "x"},
|
2019-12-02 14:39:22 +00:00
|
|
|
false,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tt := range tests {
|
|
|
|
g.It(tt.name, func() {
|
2020-02-09 22:55:31 +00:00
|
|
|
for k, v := range tt.args {
|
|
|
|
os.Setenv(k, v)
|
|
|
|
}
|
2019-12-02 14:39:22 +00:00
|
|
|
g.Assert(credsSet()).Equal(tt.want)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
})
|
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
g.Describe("tfApply", func() {
|
|
|
|
g.It("Should return correct apply commands given the arguments", func() {
|
|
|
|
type args struct {
|
|
|
|
config Config
|
2019-07-17 13:48:56 +00:00
|
|
|
}
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
args args
|
|
|
|
want *exec.Cmd
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
"default",
|
|
|
|
args{config: Config{}},
|
|
|
|
exec.Command("terraform", "apply", "plan.tfout"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"with parallelism",
|
|
|
|
args{config: Config{Parallelism: 5}},
|
|
|
|
exec.Command("terraform", "apply", "-parallelism=5", "plan.tfout"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"with targets",
|
|
|
|
args{config: Config{Targets: []string{"target1", "target2"}}},
|
|
|
|
exec.Command("terraform", "apply", "--target", "target1", "--target", "target2", "plan.tfout"),
|
|
|
|
},
|
|
|
|
}
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
for _, tt := range tests {
|
2020-02-12 16:07:44 +00:00
|
|
|
g.Assert(tfApply(tt.args.config).Args).Equal(tt.want.Args)
|
2019-11-10 00:18:12 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
})
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
g.Describe("tfDestroy", func() {
|
|
|
|
g.It("Should return correct destroy commands given the arguments", func() {
|
|
|
|
type args struct {
|
|
|
|
config Config
|
|
|
|
}
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
args args
|
|
|
|
want *exec.Cmd
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
"default",
|
|
|
|
args{config: Config{}},
|
2021-06-01 14:08:05 +00:00
|
|
|
exec.Command("terraform", "destroy", "-auto-approve"),
|
2019-11-10 00:18:12 +00:00
|
|
|
},
|
|
|
|
{
|
|
|
|
"with parallelism",
|
|
|
|
args{config: Config{Parallelism: 5}},
|
2021-06-01 14:08:05 +00:00
|
|
|
exec.Command("terraform", "destroy", "-parallelism=5", "-auto-approve"),
|
2019-11-10 00:18:12 +00:00
|
|
|
},
|
|
|
|
{
|
|
|
|
"with targets",
|
|
|
|
args{config: Config{Targets: []string{"target1", "target2"}}},
|
2021-06-01 14:08:05 +00:00
|
|
|
exec.Command("terraform", "destroy", "-target=target1", "-target=target2", "-auto-approve"),
|
2019-11-10 00:18:12 +00:00
|
|
|
},
|
|
|
|
{
|
|
|
|
"with vars",
|
|
|
|
args{config: Config{Vars: map[string]string{"username": "someuser", "password": "1pass"}}},
|
2021-06-01 14:08:05 +00:00
|
|
|
exec.Command("terraform", "destroy", "-var", "username=someuser", "-var", "password=1pass", "-auto-approve"),
|
2019-11-10 00:18:12 +00:00
|
|
|
},
|
|
|
|
{
|
|
|
|
"with var-files",
|
|
|
|
args{config: Config{VarFiles: []string{"common.tfvars", "prod.tfvars"}}},
|
2021-06-01 14:08:05 +00:00
|
|
|
exec.Command("terraform", "destroy", "-var-file=common.tfvars", "-var-file=prod.tfvars", "-auto-approve"),
|
2019-11-10 00:18:12 +00:00
|
|
|
},
|
|
|
|
}
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
for _, tt := range tests {
|
|
|
|
g.Assert(tfDestroy(tt.args.config)).Equal(tt.want)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
})
|
2019-11-26 14:51:53 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
g.Describe("tfValidate", func() {
|
|
|
|
g.It("Should return correct validate command", func() {
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
want *exec.Cmd
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
"default",
|
|
|
|
exec.Command("terraform", "validate"),
|
|
|
|
},
|
|
|
|
}
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
for _, tt := range tests {
|
|
|
|
g.Assert(tfValidate()).Equal(tt.want)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
})
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
g.Describe("tfPlan", func() {
|
|
|
|
g.It("Should return correct plan commands given the arguments", func() {
|
|
|
|
type args struct {
|
|
|
|
config Config
|
|
|
|
}
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
args args
|
|
|
|
destroy bool
|
|
|
|
want *exec.Cmd
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
"default",
|
|
|
|
args{config: Config{}},
|
|
|
|
false,
|
|
|
|
exec.Command("terraform", "plan", "-out=plan.tfout"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"destroy",
|
|
|
|
args{config: Config{}},
|
|
|
|
true,
|
|
|
|
exec.Command("terraform", "plan", "-destroy"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"with vars",
|
|
|
|
args{config: Config{Vars: map[string]string{"username": "someuser", "password": "1pass"}}},
|
|
|
|
false,
|
|
|
|
exec.Command("terraform", "plan", "-out=plan.tfout", "-var", "username=someuser", "-var", "password=1pass"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"with var-files",
|
|
|
|
args{config: Config{VarFiles: []string{"common.tfvars", "prod.tfvars"}}},
|
|
|
|
false,
|
|
|
|
exec.Command("terraform", "plan", "-out=plan.tfout", "-var-file=common.tfvars", "-var-file=prod.tfvars"),
|
|
|
|
},
|
|
|
|
}
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
for _, tt := range tests {
|
|
|
|
g.Assert(tfPlan(tt.args.config, tt.destroy)).Equal(tt.want)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
})
|
|
|
|
g.Describe("tfFmt", func() {
|
|
|
|
g.It("Should return correct fmt commands given the arguments", func() {
|
|
|
|
type args struct {
|
|
|
|
config Config
|
|
|
|
}
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
affirmative := true
|
|
|
|
negative := false
|
|
|
|
|
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
args args
|
|
|
|
want *exec.Cmd
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
"default",
|
|
|
|
args{config: Config{}},
|
|
|
|
exec.Command("terraform", "fmt"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"with list",
|
|
|
|
args{config: Config{FmtOptions: FmtOptions{List: &affirmative}}},
|
|
|
|
exec.Command("terraform", "fmt", "-list=true"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"with write",
|
|
|
|
args{config: Config{FmtOptions: FmtOptions{Write: &affirmative}}},
|
|
|
|
exec.Command("terraform", "fmt", "-write=true"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"with diff",
|
|
|
|
args{config: Config{FmtOptions: FmtOptions{Diff: &affirmative}}},
|
|
|
|
exec.Command("terraform", "fmt", "-diff=true"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"with check",
|
|
|
|
args{config: Config{FmtOptions: FmtOptions{Check: &affirmative}}},
|
|
|
|
exec.Command("terraform", "fmt", "-check=true"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"with combination",
|
|
|
|
args{config: Config{FmtOptions: FmtOptions{
|
|
|
|
List: &negative,
|
|
|
|
Write: &negative,
|
|
|
|
Diff: &affirmative,
|
|
|
|
Check: &affirmative,
|
|
|
|
}}},
|
|
|
|
exec.Command("terraform", "fmt", "-list=false", "-write=false", "-diff=true", "-check=true"),
|
|
|
|
},
|
|
|
|
}
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
for _, tt := range tests {
|
|
|
|
g.Assert(tfFmt(tt.args.config)).Equal(tt.want)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
})
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
g.Describe("tfDataDir", func() {
|
|
|
|
g.It("Should override the terraform data dir environment variable when provided", func() {
|
|
|
|
type args struct {
|
|
|
|
config Config
|
|
|
|
}
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
tests := []struct {
|
|
|
|
name string
|
|
|
|
args args
|
|
|
|
want *exec.Cmd
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
"with TerraformDataDir",
|
|
|
|
args{config: Config{TerraformDataDir: ".overriden_terraform_dir"}},
|
|
|
|
exec.Command("terraform", "apply", ".overriden_terraform_dir.plan.tfout"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"with TerraformDataDir value as .terraform",
|
|
|
|
args{config: Config{TerraformDataDir: ".terraform"}},
|
|
|
|
exec.Command("terraform", "apply", "plan.tfout"),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"without TerraformDataDir",
|
|
|
|
args{config: Config{}},
|
|
|
|
exec.Command("terraform", "apply", "plan.tfout"),
|
|
|
|
},
|
|
|
|
}
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
for _, tt := range tests {
|
|
|
|
os.Setenv("TF_DATA_DIR", tt.args.config.TerraformDataDir)
|
|
|
|
applied := tfApply(tt.args.config)
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
g.Assert(applied).Equal(tt.want)
|
2019-11-10 00:15:08 +00:00
|
|
|
|
2019-11-10 00:18:12 +00:00
|
|
|
}
|
|
|
|
})
|
|
|
|
})
|
2019-11-10 00:15:08 +00:00
|
|
|
}
|