cmd_test.go

package cmd

import (
	"bytes"
	"os"
	"strings"
	"testing"

	"github.com/sebdah/goldie/v2"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestPlain(t *testing.T) {
	os.Args = strings.Fields("tfvar testdata")

	var actual bytes.Buffer
	cmd, sync := New(&actual, "dev")
	defer sync()

	require.NoError(t, cmd.Execute())
	assert.Equal(t, `availability_zone_names = ["us-west-1a"]
docker_ports = [{
  external = 8300
  internal = 8300
  protocol = "tcp"
}]
image_id = null
password = null
`, actual.String())
}

func TestEnvVar(t *testing.T) {
	os.Args = strings.Fields("tfvar testdata -e")

	var actual bytes.Buffer
	cmd, sync := New(&actual, "dev")
	defer sync()

	require.NoError(t, cmd.Execute())
	assert.Equal(t, `export TF_VAR_availability_zone_names='["us-west-1a"]'
export TF_VAR_docker_ports='[{ external = 8300, internal = 8300, protocol = "tcp" }]'
export TF_VAR_image_id=''
export TF_VAR_password=''
`, actual.String())
}

func TestWFlag(t *testing.T) {
	os.Args = strings.Fields("tfvar testdata -w")

	var actual bytes.Buffer
	cmd, sync := New(&actual, "dev")
	defer sync()

	require.NoError(t, cmd.Execute())

	g := goldie.New(
		t,
		goldie.WithNameSuffix(".golden.json"),
		goldie.WithDiffEngine(goldie.ColoredDiff),
	)

	g.Assert(t, "w_flag", actual.Bytes())
}

func TestRFlag(t *testing.T) {
	os.Args = strings.Fields("tfvar testdata -r")

	var actual bytes.Buffer
	cmd, sync := New(&actual, "dev")
	defer sync()

	require.NoError(t, cmd.Execute())

	g := goldie.New(
		t,
		goldie.WithNameSuffix(".golden.tf"),
		goldie.WithDiffEngine(goldie.ColoredDiff),
	)

	g.Assert(t, "r_flag", actual.Bytes())
}

func TestIgnoreDefault(t *testing.T) {
	os.Args = strings.Fields("tfvar testdata --ignore-default")

	var actual bytes.Buffer
	cmd, sync := New(&actual, "dev")
	defer sync()

	require.NoError(t, cmd.Execute())
	assert.Equal(t, `availability_zone_names = null
docker_ports            = null
image_id                = null
password                = null
`, actual.String())
}

func TestAutoAssign(t *testing.T) {
	os.Args = strings.Fields("tfvar testdata -a")
	unsetImageId := setenv(t, "TF_VAR_image_id", "abc123")
	defer unsetImageId()

	unsetPassword := setenv(t, "TF_VAR_password", "secret")
	defer unsetPassword()

	var actual bytes.Buffer
	cmd, sync := New(&actual, "dev")
	defer sync()

	require.NoError(t, cmd.Execute())
	assert.Equal(t, `availability_zone_names = ["my-zone"]
docker_ports = [{
  external = 80
  internal = 80
  protocol = "tcp"
}]
image_id = "abc123"
password = "secret"
`, actual.String())
}

func setenv(t *testing.T, key, value string) func() {
	require.NoError(t, os.Setenv(key, value))

	return func() {
		require.NoError(t, os.Unsetenv(key))
	}
}

func TestVar(t *testing.T) {
	os.Args = strings.Fields("tfvar testdata -a --var=image_id=ignore_me --var=unknown=xxx --var=image_id=abc123 --var=password=secret")

	var actual bytes.Buffer
	cmd, sync := New(&actual, "dev")
	defer sync()

	require.NoError(t, cmd.Execute())
	assert.Equal(t, `availability_zone_names = ["my-zone"]
docker_ports = [{
  external = 80
  internal = 80
  protocol = "tcp"
}]
image_id = "abc123"
password = "secret"
`, actual.String())
}

func TestVarError(t *testing.T) {
	os.Args = strings.Fields("tfvar testdata -a --var=unknown")

	var actual bytes.Buffer
	cmd, sync := New(&actual, "dev")
	defer sync()

	assert.Error(t, cmd.Execute())
	assert.Contains(t, actual.String(), `Error: tfvar: bad var string 'unknown'`)
}

func TestVarFile(t *testing.T) {
	os.Args = strings.Fields("tfvar testdata --var-file testdata/my.tfvars")

	var actual bytes.Buffer
	cmd, sync := New(&actual, "dev")
	defer sync()

	require.NoError(t, cmd.Execute())
	assert.Equal(t, `availability_zone_names = ["us-west-1a"]
docker_ports = [{
  external = 8300
  internal = 8300
  protocol = "tcp"
}]
image_id = "xyz"
password = null
`, actual.String())
}

func TestMultipleVarFiles(t *testing.T) {
	os.Args = strings.Fields("tfvar testdata --var-file testdata/my.tfvars --var-file testdata/other.tfvars")

	var actual bytes.Buffer
	cmd, sync := New(&actual, "dev")
	defer sync()

	require.NoError(t, cmd.Execute())
	assert.Equal(t, `availability_zone_names = ["us-west-1a"]
docker_ports = [{
  external = 8300
  internal = 8300
  protocol = "tcp"
}]
image_id = "abc"
password = "secret"
`, actual.String())
}

func TestVarFileError(t *testing.T) {
	os.Args = strings.Fields("tfvar testdata --var-file testdata/bad.tfvars")

	var actual bytes.Buffer
	cmd, sync := New(&actual, "dev")
	defer sync()

	assert.Error(t, cmd.Execute())
	assert.Contains(t, actual.String(), `Error: tfvar: failed to parse 'testdata/bad.tfvars'`)
}