36 lines
776 B
Go
36 lines
776 B
Go
package root
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestShellArgs(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
login bool
|
|
want []string
|
|
}{
|
|
{name: "login", login: true, want: []string{"--login"}},
|
|
{name: "non-login", login: false},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := shellArgs(tt.login); !reflect.DeepEqual(got, tt.want) {
|
|
t.Fatalf("shellArgs(%v) = %v, want %v", tt.login, got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestShellLoginFlagIsRegistered(t *testing.T) {
|
|
flag := rootCmd.PersistentFlags().Lookup("shell-login")
|
|
if flag == nil {
|
|
t.Fatal("expected --shell-login flag to be registered")
|
|
}
|
|
if flag.DefValue != "false" {
|
|
t.Fatalf("expected --shell-login to default to false, got %q", flag.DefValue)
|
|
}
|
|
}
|