110 lines
2.4 KiB
Go
110 lines
2.4 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package args
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
const (
|
|
ipKey = "NOMAD_IP"
|
|
ipVal = "127.0.0.1"
|
|
portKey = "NOMAD_PORT_WEB"
|
|
portVal = ":80"
|
|
periodKey = "NOMAD.PERIOD"
|
|
periodVal = "period"
|
|
dashKey = "NOMAD-DASH"
|
|
dashVal = "dash"
|
|
)
|
|
|
|
var (
|
|
envVars = map[string]string{
|
|
ipKey: ipVal,
|
|
portKey: portVal,
|
|
periodKey: periodVal,
|
|
dashKey: dashVal,
|
|
}
|
|
)
|
|
|
|
func TestArgs_ReplaceEnv_Invalid(t *testing.T) {
|
|
input := "${FOO}"
|
|
exp := input
|
|
act := ReplaceEnv(input, envVars)
|
|
|
|
if !reflect.DeepEqual(act, exp) {
|
|
t.Fatalf("ReplaceEnv(%v, %v) returned %#v; want %#v", input, envVars, act, exp)
|
|
}
|
|
}
|
|
|
|
func TestArgs_ReplaceEnv_Valid(t *testing.T) {
|
|
input := fmt.Sprintf(`"${%v}"!`, ipKey)
|
|
exp := fmt.Sprintf("\"%s\"!", ipVal)
|
|
act := ReplaceEnv(input, envVars)
|
|
|
|
if !reflect.DeepEqual(act, exp) {
|
|
t.Fatalf("ReplaceEnv(%v, %v) returned %#v; want %#v", input, envVars, act, exp)
|
|
}
|
|
}
|
|
|
|
func TestArgs_ReplaceEnv_Period(t *testing.T) {
|
|
input := fmt.Sprintf(`"${%v}"!`, periodKey)
|
|
exp := fmt.Sprintf("\"%s\"!", periodVal)
|
|
act := ReplaceEnv(input, envVars)
|
|
|
|
if !reflect.DeepEqual(act, exp) {
|
|
t.Fatalf("ReplaceEnv(%v, %v) returned %#v; want %#v", input, envVars, act, exp)
|
|
}
|
|
}
|
|
|
|
func TestArgs_ReplaceEnv_Dash(t *testing.T) {
|
|
input := fmt.Sprintf(`"${%v}"!`, dashKey)
|
|
exp := fmt.Sprintf("\"%s\"!", dashVal)
|
|
act := ReplaceEnv(input, envVars)
|
|
|
|
if !reflect.DeepEqual(act, exp) {
|
|
t.Fatalf("ReplaceEnv(%v, %v) returned %#v; want %#v", input, envVars, act, exp)
|
|
}
|
|
}
|
|
|
|
func TestArgs_ReplaceEnv_Chained(t *testing.T) {
|
|
input := fmt.Sprintf("${%s}${%s}", ipKey, portKey)
|
|
exp := fmt.Sprintf("%s%s", ipVal, portVal)
|
|
act := ReplaceEnv(input, envVars)
|
|
|
|
if !reflect.DeepEqual(act, exp) {
|
|
t.Fatalf("ReplaceEnv(%v, %v) returned %#v; want %#v", input, envVars, act, exp)
|
|
}
|
|
}
|
|
|
|
func TestArgs_ContainsEnv(t *testing.T) {
|
|
positiveCases := []string{
|
|
"test-${env_var}",
|
|
}
|
|
for _, c := range positiveCases {
|
|
t.Run(fmt.Sprintf("positive case: %v", c), func(t *testing.T) {
|
|
if !ContainsEnv(c) {
|
|
t.Fatalf("ContainsEnv(%v) returned false; want true", c)
|
|
}
|
|
})
|
|
}
|
|
|
|
negativeCases := []string{
|
|
"test",
|
|
"test-$",
|
|
"test-${asdf",
|
|
"test-{asdf}",
|
|
"$test",
|
|
}
|
|
for _, c := range negativeCases {
|
|
t.Run(fmt.Sprintf("negative case: %v", c), func(t *testing.T) {
|
|
if ContainsEnv(c) {
|
|
t.Fatalf("ContainsEnv(%v) returned true; want false", c)
|
|
}
|
|
})
|
|
}
|
|
|
|
}
|