package argparser

import (
	"github.com/roboll/helmfile/pkg/state"
	"strings"
	"testing"
)

func TestGetArgs(t *testing.T) {
	args := "--timeout=3600 --set app1.bootstrap=true --set app2.bootstrap=false --tiller-namespace ns"
	defaultArgs := []string{"--recreate-pods", "--force"}
	Helmdefaults := state.HelmSpec{KubeContext: "test", TillerNamespace: "test-namespace", Args: defaultArgs}
	testState := &state.HelmState{
		ReleaseSetSpec: state.ReleaseSetSpec{
			HelmDefaults: Helmdefaults,
		},
	}
	receivedArgs := GetArgs(args, testState)

	expectedOutput := "--timeout=3600 --set app1.bootstrap=true --set app2.bootstrap=false --tiller-namespace ns --recreate-pods --force"

	if compareArgs(expectedOutput, receivedArgs) == false {
		t.Errorf("expected %s, got %s", expectedOutput, strings.Join(receivedArgs, " "))
	}
}

func Test2(t *testing.T) {
	args := "--timeout=3600 --set app1.bootstrap=true --set app2.bootstrap=false,app3.bootstrap=true --tiller-namespace ns"
	defaultArgs := []string{"--recreate-pods", "--force"}
	Helmdefaults := state.HelmSpec{KubeContext: "test", TillerNamespace: "test-namespace", Args: defaultArgs}
	testState := &state.HelmState{
		ReleaseSetSpec: state.ReleaseSetSpec{
			HelmDefaults: Helmdefaults,
		},
	}
	receivedArgs := GetArgs(args, testState)

	expectedOutput := "--timeout=3600 --set app1.bootstrap=true --set app2.bootstrap=false,app3.bootstrap=true --tiller-namespace ns --recreate-pods --force"

	if compareArgs(expectedOutput, receivedArgs) == false {
		t.Errorf("expected %s, got %s", expectedOutput, strings.Join(receivedArgs, " "))
	}

}

func compareArgs(expectedArgs string, args []string) bool {

	if strings.Compare(strings.Join(args, " "), expectedArgs) != 0 {
		return false
	}
	return true

}