Enterprise-Knowledge-Base/cli/cmd/kb/list_test.go

package kb

import (
	"context"
	"encoding/json"
	"errors"
	"fmt"
	"strings"
	"testing"
	"time"

	"github.com/Tencent/WeKnora/cli/internal/cmdutil"
	"github.com/Tencent/WeKnora/cli/internal/iostreams"
	sdk "github.com/Tencent/WeKnora/client"
)

type fakeListSvc struct {
	items []sdk.KnowledgeBase
	err   error
}

func (f *fakeListSvc) ListKnowledgeBases(ctx context.Context) ([]sdk.KnowledgeBase, error) {
	return f.items, f.err
}

func TestList_Empty_Human(t *testing.T) {
	out, _ := iostreams.SetForTest(t)
	if err := runList(context.Background(), &ListOptions{}, nil, &fakeListSvc{items: []sdk.KnowledgeBase{}}); err != nil {
		t.Fatalf("runList: %v", err)
	}
	if !strings.Contains(out.String(), "(no knowledge bases)") {
		t.Errorf("empty output expected '(no knowledge bases)', got %q", out.String())
	}
}

func TestList_Empty_JSON(t *testing.T) {
	out, _ := iostreams.SetForTest(t)
	jopts := &cmdutil.JSONOptions{}
	if err := runList(context.Background(), &ListOptions{}, jopts, &fakeListSvc{items: []sdk.KnowledgeBase{}}); err != nil {
		t.Fatalf("runList: %v", err)
	}
	got := strings.TrimSpace(out.String())
	if got != "[]" {
		t.Errorf("empty JSON should be bare `[]`, got %q", got)
	}
}

func TestList_NonEmpty_Human_RenderColumns(t *testing.T) {
	out, _ := iostreams.SetForTest(t)
	now := time.Now()
	items := []sdk.KnowledgeBase{
		{ID: "kb1", Name: "Marketing", KnowledgeCount: 5, UpdatedAt: now.Add(-3 * time.Hour)},
		{ID: "kb2", Name: "Engineering", KnowledgeCount: 1, UpdatedAt: now.Add(-2 * 24 * time.Hour)},
	}
	if err := runList(context.Background(), &ListOptions{}, nil, &fakeListSvc{items: items}); err != nil {
		t.Fatalf("runList: %v", err)
	}
	got := out.String()
	for _, want := range []string{"ID", "NAME", "DOCS", "UPDATED", "kb1", "Marketing", "5 docs", "kb2", "Engineering", "1 doc"} {
		if !strings.Contains(got, want) {
			t.Errorf("output missing %q in:\n%s", want, got)
		}
	}
}

func TestList_JSON_FieldFilter(t *testing.T) {
	out, _ := iostreams.SetForTest(t)
	now := time.Now()
	items := []sdk.KnowledgeBase{
		{ID: "kb1", Name: "Marketing", Description: "MKT desc", UpdatedAt: now},
	}
	jopts := &cmdutil.JSONOptions{Fields: []string{"id", "name"}}
	if err := runList(context.Background(), &ListOptions{}, jopts, &fakeListSvc{items: items}); err != nil {
		t.Fatalf("runList: %v", err)
	}
	var got []map[string]any
	if err := json.Unmarshal(out.Bytes(), &got); err != nil {
		t.Fatalf("parse: %v\n%s", err, out.String())
	}
	if len(got) != 1 {
		t.Fatalf("expected 1 item, got %d", len(got))
	}
	item := got[0]
	if item["id"] != "kb1" || item["name"] != "Marketing" {
		t.Errorf("kept fields wrong: %+v", item)
	}
	if _, has := item["description"]; has {
		t.Errorf("description should be dropped, got: %+v", item)
	}
}

func TestList_JSON_JQ(t *testing.T) {
	out, _ := iostreams.SetForTest(t)
	now := time.Now()
	items := []sdk.KnowledgeBase{
		{ID: "kb1", Name: "Marketing", UpdatedAt: now},
		{ID: "kb2", Name: "Engineering", UpdatedAt: now.Add(-time.Hour)},
	}
	jopts := &cmdutil.JSONOptions{JQ: ". | length"}
	if err := runList(context.Background(), &ListOptions{}, jopts, &fakeListSvc{items: items}); err != nil {
		t.Fatalf("runList: %v", err)
	}
	if got := strings.TrimSpace(out.String()); got != "2" {
		t.Errorf("expected '2', got %q", got)
	}
}

func TestList_PinnedFilter(t *testing.T) {
	out, _ := iostreams.SetForTest(t)
	now := time.Now()
	items := []sdk.KnowledgeBase{
		{ID: "kb1", Name: "Marketing", IsPinned: true, UpdatedAt: now},
		{ID: "kb2", Name: "Engineering", IsPinned: false, UpdatedAt: now.Add(-time.Hour)},
		{ID: "kb3", Name: "Finance", IsPinned: true, UpdatedAt: now.Add(-2 * time.Hour)},
	}
	if err := runList(context.Background(), &ListOptions{Pinned: true}, nil, &fakeListSvc{items: items}); err != nil {
		t.Fatalf("runList: %v", err)
	}
	got := out.String()
	if !strings.Contains(got, "kb1") || !strings.Contains(got, "kb3") {
		t.Errorf("expected pinned KBs kb1 and kb3 in output, got:\n%s", got)
	}
	if strings.Contains(got, "kb2") {
		t.Errorf("unpinned kb2 should be filtered out, got:\n%s", got)
	}
}

func TestList_PinnedFilter_NoPinned_HumanMessage(t *testing.T) {
	out, _ := iostreams.SetForTest(t)
	items := []sdk.KnowledgeBase{
		{ID: "kb1", Name: "Marketing", IsPinned: false, UpdatedAt: time.Now()},
	}
	if err := runList(context.Background(), &ListOptions{Pinned: true}, nil, &fakeListSvc{items: items}); err != nil {
		t.Fatalf("runList: %v", err)
	}
	if !strings.Contains(out.String(), "(no pinned knowledge bases)") {
		t.Errorf("expected pinned-specific empty message, got: %q", out.String())
	}
}

// makeKBs returns N KBs with distinct IDs and descending UpdatedAt.
func makeKBs(n int) []sdk.KnowledgeBase {
	base := time.Now()
	out := make([]sdk.KnowledgeBase, n)
	for i := 0; i < n; i++ {
		out[i] = sdk.KnowledgeBase{
			ID:        fmt.Sprintf("kb_%02d", i),
			Name:      fmt.Sprintf("kb-%02d", i),
			UpdatedAt: base.Add(-time.Duration(i) * time.Hour),
		}
	}
	return out
}

func TestList_Limit_CapsResults(t *testing.T) {
	out, _ := iostreams.SetForTest(t)
	svc := &fakeListSvc{items: makeKBs(20)}
	jopts := &cmdutil.JSONOptions{}
	if err := runList(context.Background(), &ListOptions{Limit: 5}, jopts, svc); err != nil {
		t.Fatalf("runList: %v", err)
	}
	got := strings.Count(out.String(), `"id":"kb_`)
	if got != 5 {
		t.Errorf("--limit 5 should slice 20 items to 5; got %d in:\n%s", got, out.String())
	}
}

func TestList_Limit_Zero_NoCap(t *testing.T) {
	out, _ := iostreams.SetForTest(t)
	svc := &fakeListSvc{items: makeKBs(7)}
	jopts := &cmdutil.JSONOptions{}
	if err := runList(context.Background(), &ListOptions{Limit: 0}, jopts, svc); err != nil {
		t.Fatalf("runList: %v", err)
	}
	got := strings.Count(out.String(), `"id":"kb_`)
	if got != 7 {
		t.Errorf("--limit 0 must not cap; got %d, want 7", got)
	}
}

func TestList_Limit_Negative_Rejected(t *testing.T) {
	_, _ = iostreams.SetForTest(t)
	err := runList(context.Background(), &ListOptions{Limit: -1}, nil, &fakeListSvc{items: makeKBs(3)})
	if err == nil {
		t.Fatal("expected error for negative --limit")
	}
	var typed *cmdutil.Error
	if !errors.As(err, &typed) {
		t.Fatalf("expected *cmdutil.Error, got %T: %v", err, err)
	}
	if typed.Code != cmdutil.CodeInputInvalidArgument {
		t.Errorf("expected CodeInputInvalidArgument, got %v", typed.Code)
	}
}