larksuite__cli/shortcuts/im/im_chat_search_test.go

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

package im

import (
	"strings"
	"testing"

	"github.com/larksuite/cli/shortcuts/common"
)

func newSearchTestRT(t *testing.T, stringFlags map[string]string) *common.RuntimeContext {
	t.Helper()
	if stringFlags == nil {
		stringFlags = map[string]string{}
	}
	if _, ok := stringFlags["query"]; !ok {
		stringFlags["query"] = "team"
	}
	return newChatListTestRuntimeContext(t, stringFlags, nil)
}

func TestChatSearch_SortMapping(t *testing.T) {
	cases := []struct{ sort, want string }{
		{"create_time", "create_time_desc"},
		{"update_time", "update_time_desc"},
		{"member_count", "member_count_desc"},
	}
	for _, c := range cases {
		t.Run(c.sort, func(t *testing.T) {
			rt := newSearchTestRT(t, map[string]string{"sort": c.sort})
			body := buildSearchChatBody(rt)
			if body["sorter"] != c.want {
				t.Fatalf("sort=%s -> sorter=%v, want %s", c.sort, body["sorter"], c.want)
			}
		})
	}
}

// TestChatSearch_SortOmittedWhenUnset: no --sort and no --sort-by -> sorter omitted.
func TestChatSearch_SortOmittedWhenUnset(t *testing.T) {
	rt := newSearchTestRT(t, nil)
	body := buildSearchChatBody(rt)
	if _, present := body["sorter"]; present {
		t.Fatalf("sorter should be omitted when neither --sort nor --sort-by set")
	}
}

// TestChatSearch_SortAliasParity: hidden --sort-by value is already the upstream
// sorter (pass-through), so it must equal the mapped new --sort body.
func TestChatSearch_SortAliasParity(t *testing.T) {
	pairs := []struct{ newVal, oldVal string }{
		{"create_time", "create_time_desc"},
		{"update_time", "update_time_desc"},
		{"member_count", "member_count_desc"},
	}
	for _, p := range pairs {
		t.Run(p.newVal, func(t *testing.T) {
			newBody := buildSearchChatBody(newSearchTestRT(t, map[string]string{"sort": p.newVal}))
			oldBody := buildSearchChatBody(newSearchTestRT(t, map[string]string{"sort-by": p.oldVal}))
			if newBody["sorter"] != oldBody["sorter"] {
				t.Fatalf("alias parity: new sorter=%v, old sorter=%v", newBody["sorter"], oldBody["sorter"])
			}
		})
	}
}

func TestChatSearch_SortNewWins(t *testing.T) {
	rt := newSearchTestRT(t, map[string]string{"sort": "member_count", "sort-by": "create_time_desc"})
	body := buildSearchChatBody(rt)
	if body["sorter"] != "member_count_desc" {
		t.Fatalf("new should win: sorter=%v, want member_count_desc", body["sorter"])
	}
}

func TestChatSearch_SortFlagSurface(t *testing.T) {
	var sortFlag, aliasFlag *common.Flag
	for i := range ImChatSearch.Flags {
		switch ImChatSearch.Flags[i].Name {
		case "sort":
			sortFlag = &ImChatSearch.Flags[i]
		case "sort-by":
			aliasFlag = &ImChatSearch.Flags[i]
		}
	}
	if sortFlag == nil || aliasFlag == nil {
		t.Fatalf("expected both --sort and --sort-by flags declared")
	}
	if sortFlag.Default != "" {
		t.Errorf("--sort must have no default (sorter omitted when unset), got %q", sortFlag.Default)
	}
	if got := strings.Join(sortFlag.Enum, ","); got != "create_time,update_time,member_count" {
		t.Errorf("--sort Enum = %q", got)
	}
	if !aliasFlag.Hidden {
		t.Errorf("--sort-by must be Hidden")
	}
	if got := strings.Join(aliasFlag.Enum, ","); got != "create_time_desc,update_time_desc,member_count_desc" {
		t.Errorf("--sort-by Enum = %q", got)
	}
}