mirror of
https://github.com/larksuite/cli.git
synced 2026-07-09 02:14:02 +08:00
8abd39b18e
Add LARKSUITE_CLI_PROFILE env var and make BootstrapInvocationContext fall back to it when --profile is empty, so downstream credential resolution sees the correct profile. Also track whether the resolved profile came from the flag or the env fallback via a new InvocationContext.ProfileFromFlag field, needed by a later task to report the correct credential source.
119 lines
3.5 KiB
Go
119 lines
3.5 KiB
Go
// Copyright (c) 2026 Lark Technologies Pte. Ltd.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/larksuite/cli/internal/envvars"
|
|
)
|
|
|
|
func TestBootstrapInvocationContext_ProfileFlag(t *testing.T) {
|
|
inv, err := BootstrapInvocationContext([]string{"--profile", "target", "auth", "status"})
|
|
if err != nil {
|
|
t.Fatalf("BootstrapInvocationContext() error = %v", err)
|
|
}
|
|
if inv.Profile != "target" {
|
|
t.Fatalf("BootstrapInvocationContext() profile = %q, want %q", inv.Profile, "target")
|
|
}
|
|
}
|
|
|
|
func TestBootstrapInvocationContext_ProfileEquals(t *testing.T) {
|
|
inv, err := BootstrapInvocationContext([]string{"auth", "status", "--profile=target"})
|
|
if err != nil {
|
|
t.Fatalf("BootstrapInvocationContext() error = %v", err)
|
|
}
|
|
if inv.Profile != "target" {
|
|
t.Fatalf("BootstrapInvocationContext() profile = %q, want %q", inv.Profile, "target")
|
|
}
|
|
}
|
|
|
|
func TestBootstrapInvocationContext_IgnoresUnknownFlags(t *testing.T) {
|
|
inv, err := BootstrapInvocationContext([]string{"auth", "status", "--verify", "--profile", "target"})
|
|
if err != nil {
|
|
t.Fatalf("BootstrapInvocationContext() error = %v", err)
|
|
}
|
|
if inv.Profile != "target" {
|
|
t.Fatalf("BootstrapInvocationContext() profile = %q, want %q", inv.Profile, "target")
|
|
}
|
|
}
|
|
|
|
func TestBootstrapInvocationContext_MissingProfileValue(t *testing.T) {
|
|
if _, err := BootstrapInvocationContext([]string{"auth", "status", "--profile"}); err == nil {
|
|
t.Fatal("BootstrapInvocationContext() error = nil, want non-nil")
|
|
}
|
|
}
|
|
|
|
func TestBootstrapInvocationContext_HelpFlag(t *testing.T) {
|
|
inv, err := BootstrapInvocationContext([]string{"--help"})
|
|
if err != nil {
|
|
t.Fatalf("--help should not error, got: %v", err)
|
|
}
|
|
if inv.Profile != "" {
|
|
t.Fatalf("profile = %q, want empty", inv.Profile)
|
|
}
|
|
}
|
|
|
|
func TestBootstrapInvocationContext_ShortHelp(t *testing.T) {
|
|
inv, err := BootstrapInvocationContext([]string{"-h"})
|
|
if err != nil {
|
|
t.Fatalf("-h should not error, got: %v", err)
|
|
}
|
|
if inv.Profile != "" {
|
|
t.Fatalf("profile = %q, want empty", inv.Profile)
|
|
}
|
|
}
|
|
|
|
func TestBootstrapInvocationContext_HelpWithProfile(t *testing.T) {
|
|
inv, err := BootstrapInvocationContext([]string{"--profile", "target", "--help"})
|
|
if err != nil {
|
|
t.Fatalf("--profile + --help should not error, got: %v", err)
|
|
}
|
|
if inv.Profile != "target" {
|
|
t.Fatalf("profile = %q, want %q", inv.Profile, "target")
|
|
}
|
|
}
|
|
|
|
func TestBootstrapProfileEnvFallback(t *testing.T) {
|
|
t.Run("flag wins over env", func(t *testing.T) {
|
|
t.Setenv(envvars.CliProfile, "tenant_env")
|
|
inv, err := BootstrapInvocationContext([]string{"--profile", "tenant_flag", "whoami"})
|
|
if err != nil {
|
|
t.Fatalf("unexpected err: %v", err)
|
|
}
|
|
if inv.Profile != "tenant_flag" {
|
|
t.Errorf("got %q, want tenant_flag", inv.Profile)
|
|
}
|
|
if !inv.ProfileFromFlag {
|
|
t.Errorf("ProfileFromFlag = false, want true")
|
|
}
|
|
})
|
|
t.Run("env used when flag absent", func(t *testing.T) {
|
|
t.Setenv(envvars.CliProfile, "tenant_env")
|
|
inv, err := BootstrapInvocationContext([]string{"whoami"})
|
|
if err != nil {
|
|
t.Fatalf("unexpected err: %v", err)
|
|
}
|
|
if inv.Profile != "tenant_env" {
|
|
t.Errorf("got %q, want tenant_env", inv.Profile)
|
|
}
|
|
if inv.ProfileFromFlag {
|
|
t.Errorf("ProfileFromFlag = true, want false")
|
|
}
|
|
})
|
|
t.Run("empty when neither set", func(t *testing.T) {
|
|
t.Setenv(envvars.CliProfile, "")
|
|
inv, err := BootstrapInvocationContext([]string{"whoami"})
|
|
if err != nil {
|
|
t.Fatalf("unexpected err: %v", err)
|
|
}
|
|
if inv.Profile != "" {
|
|
t.Errorf("got %q, want empty", inv.Profile)
|
|
}
|
|
if inv.ProfileFromFlag {
|
|
t.Errorf("ProfileFromFlag = true, want false")
|
|
}
|
|
})
|
|
}
|