81 lines
2.1 KiB
Go
81 lines
2.1 KiB
Go
|
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)
|
||
|
|
}
|
||
|
|
}
|