114 lines
2.7 KiB
Go
114 lines
2.7 KiB
Go
package command
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/nomad/ci"
|
|
"github.com/mitchellh/cli"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCommand_Ui(t *testing.T) {
|
|
ci.Parallel(t)
|
|
|
|
type testCaseSetupFn func(*testing.T)
|
|
|
|
cases := []struct {
|
|
Name string
|
|
SetupFn testCaseSetupFn
|
|
Args []string
|
|
ExpectedURL string
|
|
}{
|
|
{
|
|
Name: "default values",
|
|
ExpectedURL: "http://127.0.0.1:4646",
|
|
},
|
|
{
|
|
Name: "set namespace via flag",
|
|
Args: []string{"-namespace=dev"},
|
|
ExpectedURL: "http://127.0.0.1:4646?namespace=dev",
|
|
},
|
|
{
|
|
Name: "set region via flag",
|
|
Args: []string{"-region=earth"},
|
|
ExpectedURL: "http://127.0.0.1:4646?region=earth",
|
|
},
|
|
{
|
|
Name: "set region and namespace via flag",
|
|
Args: []string{"-region=earth", "-namespace=dev"},
|
|
ExpectedURL: "http://127.0.0.1:4646?namespace=dev®ion=earth",
|
|
},
|
|
{
|
|
Name: "set namespace via env var",
|
|
SetupFn: func(t *testing.T) {
|
|
setEnv(t, "NOMAD_NAMESPACE", "dev")
|
|
},
|
|
ExpectedURL: "http://127.0.0.1:4646?namespace=dev",
|
|
},
|
|
{
|
|
Name: "set region via env var",
|
|
SetupFn: func(t *testing.T) {
|
|
setEnv(t, "NOMAD_REGION", "earth")
|
|
},
|
|
ExpectedURL: "http://127.0.0.1:4646?region=earth",
|
|
},
|
|
{
|
|
Name: "set region and namespace via env var",
|
|
SetupFn: func(t *testing.T) {
|
|
setEnv(t, "NOMAD_REGION", "earth")
|
|
setEnv(t, "NOMAD_NAMESPACE", "dev")
|
|
},
|
|
ExpectedURL: "http://127.0.0.1:4646?namespace=dev®ion=earth",
|
|
},
|
|
{
|
|
Name: "set region and namespace via env var",
|
|
SetupFn: func(t *testing.T) {
|
|
setEnv(t, "NOMAD_REGION", "earth")
|
|
setEnv(t, "NOMAD_NAMESPACE", "dev")
|
|
},
|
|
ExpectedURL: "http://127.0.0.1:4646?namespace=dev®ion=earth",
|
|
},
|
|
{
|
|
Name: "flags have higher precedence",
|
|
SetupFn: func(t *testing.T) {
|
|
setEnv(t, "NOMAD_REGION", "earth")
|
|
setEnv(t, "NOMAD_NAMESPACE", "dev")
|
|
},
|
|
Args: []string{
|
|
"-region=mars",
|
|
"-namespace=prod",
|
|
},
|
|
ExpectedURL: "http://127.0.0.1:4646?namespace=prod®ion=mars",
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
t.Run(tc.Name, func(t *testing.T) {
|
|
// Make sure environment variables are clean.
|
|
setEnv(t, "NOMAD_NAMESPACE", "")
|
|
setEnv(t, "NOMAD_REGION", "")
|
|
|
|
// Setup fake CLI UI and test case
|
|
ui := cli.NewMockUi()
|
|
cmd := &UiCommand{Meta: Meta{Ui: ui}}
|
|
|
|
if tc.SetupFn != nil {
|
|
tc.SetupFn(t)
|
|
}
|
|
|
|
// Don't try to open a browser.
|
|
args := append(tc.Args, "-show-url")
|
|
|
|
if code := cmd.Run(args); code != 0 {
|
|
require.Equal(t, 0, code, "expected exit code 0, got %d", code)
|
|
}
|
|
|
|
got := ui.OutputWriter.String()
|
|
expected := fmt.Sprintf("URL for web UI: %s", tc.ExpectedURL)
|
|
require.Equal(t, expected, strings.TrimSpace(got))
|
|
})
|
|
}
|
|
}
|