helmfile/helmexec/exec_test.go

package helmexec

import (
	"bytes"
	"io"
	"reflect"
	"testing"
)

// Mocking the command-line runner

type MockRunner struct {
	output []byte
	err error
}

func (mock *MockRunner) Execute(cmd string, args []string) ([]byte, error) {
	return []byte{}, nil
}

func NewMockExec(writer io.Writer, kubeContext string) Interface {
	nhe := NewHelmExec(writer, kubeContext)
	runner := &MockRunner{}
	nhe.setRunner(runner)
	return nhe
}

// Test methods

func TestNewHelmExec(t *testing.T) {
	buffer := bytes.NewBufferString("something")
	helm := NewHelmExec(buffer, "dev")
	if helm.getKubeContent() != "dev" {
		t.Error("helmexec.NewHelmExec() - kubeContext")
	}
	if buffer.String() != "something" {
		t.Error("helmexec.NewHelmExec() - changed buffer")
	}
	if len(helm.getExtra()) != 0 {
		t.Error("helmexec.NewHelmExec() - extra args not empty")
	}
}

func Test_SetExtraArgs(t *testing.T) {
	helm := NewHelmExec(new(bytes.Buffer), "dev")
	helm.SetExtraArgs()
	if len(helm.getExtra()) != 0 {
		t.Error("helmexec.SetExtraArgs() - passing no arguments should not change extra field")
	}
	helm.SetExtraArgs("foo")
	if !reflect.DeepEqual(helm.getExtra(), []string{"foo"}) {
		t.Error("helmexec.SetExtraArgs() - one extra argument missing")
	}
	helm.SetExtraArgs("alpha", "beta")
	if !reflect.DeepEqual(helm.getExtra(), []string{"alpha", "beta"}) {
		t.Error("helmexec.SetExtraArgs() - two extra arguments missing (overwriting the previous value)")
	}
}

func Test_AddRepo(t *testing.T) {
	var buffer bytes.Buffer
	helm := NewMockExec(&buffer, "dev")
	helm.AddRepo("myRepo", "https://repo.example.com/", "cert.pem", "key.pem")
	expected := "exec: helm repo add myRepo https://repo.example.com/ --cert-file cert.pem --key-file key.pem --kube-context dev\n"
	if buffer.String() != expected {
		t.Errorf("helmexec.AddRepo()\nactual = %v\nexpect = %v", buffer.String(), expected)
	}

	buffer.Reset()
	helm.AddRepo("myRepo", "https://repo.example.com/", "", "")
	expected = "exec: helm repo add myRepo https://repo.example.com/ --kube-context dev\n"
	if buffer.String() != expected {
		t.Errorf("helmexec.AddRepo()\nactual = %v\nexpect = %v", buffer.String(), expected)
	}
}

func Test_UpdateRepo(t *testing.T) {
	var buffer bytes.Buffer
	helm := NewMockExec(&buffer, "dev")
	helm.UpdateRepo()
	expected := "exec: helm repo update --kube-context dev\n"
	if buffer.String() != expected {
		t.Errorf("helmexec.UpdateRepo()\nactual = %v\nexpect = %v", buffer.String(), expected)
	}
}

func Test_SyncRelease(t *testing.T) {
	var buffer bytes.Buffer
	helm := NewMockExec(&buffer, "dev")
	helm.SyncRelease("release", "chart", "--timeout 10", "--wait")
	expected := "exec: helm upgrade --install release chart --timeout 10 --wait --kube-context dev\n"
	if buffer.String() != expected {
		t.Errorf("helmexec.SyncRelease()\nactual = %v\nexpect = %v", buffer.String(), expected)
	}

	buffer.Reset()
	helm.SyncRelease("release", "chart")
	expected = "exec: helm upgrade --install release chart --kube-context dev\n"
	if buffer.String() != expected {
		t.Errorf("helmexec.AddRepo()\nactual = %v\nexpect = %v", buffer.String(), expected)
	}
}

func Test_DecryptSecret(t *testing.T) {
	var buffer bytes.Buffer
	helm := NewMockExec(&buffer, "dev")
	helm.DecryptSecret("secretName")
	expected := "exec: helm secrets dec secretName --kube-context dev\n"
	if buffer.String() != expected {
		t.Errorf("helmexec.DecryptSecret()\nactual = %v\nexpect = %v", buffer.String(), expected)
	}
}

func Test_DiffRelease(t *testing.T) {
	var buffer bytes.Buffer
	helm := NewMockExec(&buffer, "dev")
	helm.DiffRelease("release", "chart", "--timeout 10", "--wait")
	expected := "exec: helm diff release chart --timeout 10 --wait --kube-context dev\n"
	if buffer.String() != expected {
		t.Errorf("helmexec.DiffRelease()\nactual = %v\nexpect = %v", buffer.String(), expected)
	}

	buffer.Reset()
	helm.DiffRelease("release", "chart")
	expected = "exec: helm diff release chart --kube-context dev\n"
	if buffer.String() != expected {
		t.Errorf("helmexec.DiffRelease()\nactual = %v\nexpect = %v", buffer.String(), expected)
	}
}

func Test_DeleteRelease(t *testing.T) {
	var buffer bytes.Buffer
	helm := NewMockExec(&buffer, "dev")
	helm.DeleteRelease("release")
	expected := "exec: helm delete --purge release --kube-context dev\n"
	if buffer.String() != expected {
		t.Errorf("helmexec.DeleteRelease()\nactual = %v\nexpect = %v", buffer.String(), expected)
	}
}

func Test_exec(t *testing.T) {
	var buffer bytes.Buffer
	helm := NewMockExec(&buffer, "")
	helm.exec("version")
	expected := "exec: helm version\n"
	if buffer.String() != expected {
		t.Errorf("helmexec.exec()\nactual = %v\nexpect = %v", buffer.String(), expected)
	}

	helm = NewMockExec(nil, "dev")
	ret, _ := helm.exec("diff")
	if len(ret) != 0 {
		t.Error("helmexec.exec() - expected empty return value")
	}

	buffer.Reset()
	helm = NewMockExec(&buffer, "dev")
	helm.exec("diff", "release", "chart", "--timeout 10", "--wait")
	expected = "exec: helm diff release chart --timeout 10 --wait --kube-context dev\n"
	if buffer.String() != expected {
		t.Errorf("helmexec.exec()\nactual = %v\nexpect = %v", buffer.String(), expected)
	}

	buffer.Reset()
	helm.exec("version")
	expected = "exec: helm version --kube-context dev\n"
	if buffer.String() != expected {
		t.Errorf("helmexec.exec()\nactual = %v\nexpect = %v", buffer.String(), expected)
	}

	buffer.Reset()
	helm.SetExtraArgs("foo")
	helm.exec("version")
	expected = "exec: helm version foo --kube-context dev\n"
	if buffer.String() != expected {
		t.Errorf("helmexec.exec()\nactual = %v\nexpect = %v", buffer.String(), expected)
	}
}