mirror of https://github.com/hashicorp/consul
60 lines
1.1 KiB
Go
60 lines
1.1 KiB
Go
package command
|
|
|
|
import (
|
|
"flag"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
const defaultRPC = "127.0.0.1:8400"
|
|
|
|
func getParsedRPC(t *testing.T, cliRPC, envRPC string) string {
|
|
args := []string{}
|
|
|
|
if cliRPC != "" {
|
|
args = append(args, "-rpc-addr="+cliRPC)
|
|
}
|
|
|
|
os.Clearenv()
|
|
if envRPC != "" {
|
|
os.Setenv(RPCAddrEnvName, envRPC)
|
|
}
|
|
|
|
cmdFlags := flag.NewFlagSet("rpc", flag.ContinueOnError)
|
|
rpc := RPCAddrFlag(cmdFlags)
|
|
|
|
if err := cmdFlags.Parse(args); err != nil {
|
|
t.Fatal("Parse error", err)
|
|
}
|
|
|
|
return *rpc
|
|
}
|
|
|
|
func TestRPCAddrFlag_default(t *testing.T) {
|
|
rpc := getParsedRPC(t, "", "")
|
|
|
|
if rpc != defaultRPC {
|
|
t.Fatalf("Expected rpc addr: %s, got: %s", defaultRPC, rpc)
|
|
}
|
|
}
|
|
|
|
func TestRPCAddrFlag_onlyEnv(t *testing.T) {
|
|
envRPC := "4.4.4.4:8400"
|
|
rpc := getParsedRPC(t, "", envRPC)
|
|
|
|
if rpc != envRPC {
|
|
t.Fatalf("Expected rpc addr: %s, got: %s", envRPC, rpc)
|
|
}
|
|
}
|
|
|
|
func TestRPCAddrFlag_precedence(t *testing.T) {
|
|
cliRPC := "8.8.8.8:8400"
|
|
envRPC := "4.4.4.4:8400"
|
|
|
|
rpc := getParsedRPC(t, cliRPC, envRPC)
|
|
|
|
if rpc != cliRPC {
|
|
t.Fatalf("Expected rpc addr: %s, got: %s", cliRPC, rpc)
|
|
}
|
|
}
|