package args

import (
	"fmt"
	"reflect"
	"testing"
)

const (
	ipKey   = "NOMAD_IP"
	ipVal   = "127.0.0.1"
	portKey = "NOMAD_PORT_WEB"
	portVal = ":80"
)

var (
	envVars = map[string]string{
		ipKey:   ipVal,
		portKey: portVal,
	}
)

func TestDriverArgs_ParseAndReplaceInvalidEnv(t *testing.T) {
	input := "invalid $FOO"
	exp := []string{"invalid", "$FOO"}
	act, err := ParseAndReplace(input, envVars)
	if err != nil {
		t.Fatalf("Failed to parse valid input args %v: %v", input, err)
	}

	if !reflect.DeepEqual(act, exp) {
		t.Fatalf("ParseAndReplace(%v, %v) returned %#v; want %#v", input, envVars, act, exp)
	}
}

func TestDriverArgs_ParseAndReplaceValidEnv(t *testing.T) {
	input := fmt.Sprintf("nomad_ip \\\"$%v\\\"!", ipKey)
	exp := []string{"nomad_ip", fmt.Sprintf("\"%s\"!", ipVal)}
	act, err := ParseAndReplace(input, envVars)
	if err != nil {
		t.Fatalf("Failed to parse valid input args %v: %v", input, err)
	}

	if !reflect.DeepEqual(act, exp) {
		t.Fatalf("ParseAndReplace(%v, %v) returned %#v; want %#v", input, envVars, act, exp)
	}
}

func TestDriverArgs_ParseAndReplaceChainedEnv(t *testing.T) {
	input := fmt.Sprintf("-foo $%s$%s", ipKey, portKey)
	exp := []string{"-foo", fmt.Sprintf("%s%s", ipVal, portVal)}
	act, err := ParseAndReplace(input, envVars)
	if err != nil {
		t.Fatalf("Failed to parse valid input args %v: %v", input, err)
	}

	if !reflect.DeepEqual(act, exp) {
		t.Fatalf("ParseAndReplace(%v, %v) returned %#v; want %#v", input, envVars, act, exp)
	}
}

func TestDriverArgs_ParseAndReplaceInvalidArgEscape(t *testing.T) {
	input := "-c \"echo \"foo\\\" > bar.txt\""
	if _, err := ParseAndReplace(input, envVars); err == nil {
		t.Fatalf("ParseAndReplace(%v, %v) should have failed", input, envVars)
	}
}

func TestDriverArgs_ParseAndReplaceValidArgEscape(t *testing.T) {
	input := "-c \"echo \\\"foo\\\" > bar.txt\""
	exp := []string{"-c", "echo \"foo\" > bar.txt"}
	act, err := ParseAndReplace(input, envVars)
	if err != nil {
		t.Fatalf("Failed to parse valid input args %v: %v", input, err)
	}

	if !reflect.DeepEqual(act, exp) {
		t.Fatalf("ParseAndReplace(%v, %v) returned %#v; want %#v", input, envVars, act, exp)
	}
}