larksuite--cli/internal/cmdutil/identity_test.go

// Copyright (c) 2026 Lark Technologies Pte. Ltd.
// SPDX-License-Identifier: MIT

package cmdutil

import (
	"bytes"
	"strings"
	"testing"

	"github.com/larksuite/cli/internal/core"
)

func TestPrintIdentity_BotExplicit(t *testing.T) {
	var buf bytes.Buffer
	PrintIdentity(&buf, core.AsBot, nil, false)
	if !strings.Contains(buf.String(), "[identity: bot]") {
		t.Errorf("unexpected output: %s", buf.String())
	}
}

func TestPrintIdentity_BotAutoDetected(t *testing.T) {
	var buf bytes.Buffer
	PrintIdentity(&buf, core.AsBot, nil, true)
	if !strings.Contains(buf.String(), "auto") {
		t.Errorf("expected auto hint, got: %s", buf.String())
	}
}

func TestPrintIdentity_UserWithOpenId(t *testing.T) {
	var buf bytes.Buffer
	cfg := &core.CliConfig{UserOpenId: "ou_abc123"}
	PrintIdentity(&buf, core.AsUser, cfg, false)
	if !strings.Contains(buf.String(), "ou_abc123") {
		t.Errorf("expected UserOpenId in output, got: %s", buf.String())
	}
}

func TestPrintIdentity_UserWithoutOpenId(t *testing.T) {
	var buf bytes.Buffer
	PrintIdentity(&buf, core.AsUser, &core.CliConfig{}, false)
	if !strings.Contains(buf.String(), "[identity: user]") {
		t.Errorf("unexpected output: %s", buf.String())
	}
}

func TestPrintIdentity_UserNilConfig(t *testing.T) {
	var buf bytes.Buffer
	PrintIdentity(&buf, core.AsUser, nil, false)
	if !strings.Contains(buf.String(), "[identity: user]") {
		t.Errorf("unexpected output: %s", buf.String())
	}
}