Enterprise-Knowledge-Base/cli/cmd/root_test.go

package cmd

import (
	"bytes"
	"errors"
	"strings"
	"testing"

	"github.com/spf13/cobra"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"github.com/Tencent/WeKnora/cli/internal/cmdutil"
)

func TestRoot_Help(t *testing.T) {
	var out bytes.Buffer
	root := NewRootCmd(cmdutil.New())
	root.SetArgs([]string{"--help"})
	root.SetOut(&out)
	require.NoError(t, root.Execute())
	got := out.String()
	assert.Contains(t, got, "weknora")
	assert.Contains(t, got, "version")
}

func TestVersion_JSON(t *testing.T) {
	var out bytes.Buffer
	root := NewRootCmd(cmdutil.New())
	root.SetArgs([]string{"version", "--json"})
	root.SetOut(&out)
	require.NoError(t, root.Execute())
	got := out.String()
	assert.True(t, strings.HasPrefix(got, `{"ok":true`), "got: %q", got)
	assert.Contains(t, got, "version")
}

// Smoke test for cmdutil.ExitCode wiring; full coverage lives in
// cli/internal/cmdutil/exit_test.go.
func TestExecute_ExitCodeSurface(t *testing.T) {
	assert.Equal(t, 0, cmdutil.ExitCode(nil))
	assert.Equal(t, 1, cmdutil.ExitCode(assert.AnError))
}

// TestMapCobraError_PinnedPrefixes guards against silent breakage if cobra
// changes the message format of unknown-command / required-flag / arg-count
// errors. Cobra v1.10 emits these via fmt.Errorf in args.go and command.go;
// if a future bump alters the wording, this test fails loudly so we update
// cobraFlagErrorPrefixes (or migrate to typed sentinels if cobra ever
// provides them).
func TestMapCobraError_PinnedPrefixes(t *testing.T) {
	t.Run("unknown command", func(t *testing.T) {
		root := NewRootCmd(cmdutil.New())
		root.SetArgs([]string{"bogus"})
		root.SetErr(&bytes.Buffer{})
		root.SetOut(&bytes.Buffer{})
		err := root.Execute()
		require.Error(t, err)
		assert.True(t, strings.HasPrefix(err.Error(), "unknown command "),
			"cobra unknown-command prefix changed; update cobraFlagErrorPrefixes. got: %q", err.Error())
	})

	t.Run("required flag(s)", func(t *testing.T) {
		// Self-contained probe — the pin must hold even before resource commands
		// register their own required flags. RunE is required: without it cobra
		// treats the command as a parent and skips ValidateRequiredFlags.
		probe := &cobra.Command{Use: "probe", RunE: func(*cobra.Command, []string) error { return nil }}
		probe.Flags().String("host", "", "")
		require.NoError(t, probe.MarkFlagRequired("host"))
		probe.SetErr(&bytes.Buffer{})
		probe.SetOut(&bytes.Buffer{})
		err := probe.Execute()
		require.Error(t, err)
		assert.True(t, strings.HasPrefix(err.Error(), "required flag(s)"),
			"cobra required-flag prefix changed; update cobraFlagErrorPrefixes. got: %q", err.Error())
	})

	t.Run("accepts N arg(s) — ExactArgs", func(t *testing.T) {
		probe := &cobra.Command{
			Use:  "probe",
			Args: cobra.ExactArgs(1),
			RunE: func(*cobra.Command, []string) error { return nil },
		}
		probe.SetArgs([]string{}) // no args, but ExactArgs(1) wants 1
		probe.SetErr(&bytes.Buffer{})
		probe.SetOut(&bytes.Buffer{})
		err := probe.Execute()
		require.Error(t, err)
		assert.True(t, strings.HasPrefix(err.Error(), "accepts "),
			"cobra ExactArgs prefix changed; update cobraFlagErrorPrefixes. got: %q", err.Error())
	})
}

func TestMapCobraError(t *testing.T) {
	t.Run("nil passes through", func(t *testing.T) {
		assert.Nil(t, mapCobraError(nil))
	})
	t.Run("non-matching error passes through", func(t *testing.T) {
		err := mapCobraError(assert.AnError)
		assert.Equal(t, assert.AnError, err)
	})
	t.Run("unknown command wraps as FlagError", func(t *testing.T) {
		err := mapCobraError(errors.New(`unknown command "bogus" for "weknora"`))
		var fe *cmdutil.FlagError
		assert.True(t, errors.As(err, &fe))
	})
	t.Run("required flag wraps as FlagError", func(t *testing.T) {
		err := mapCobraError(errors.New(`required flag(s) "host" not set`))
		var fe *cmdutil.FlagError
		assert.True(t, errors.As(err, &fe))
	})
}

func TestArgsRequestJSON(t *testing.T) {
	cases := []struct {
		name string
		args []string
		want bool
	}{
		{"empty", nil, false},
		{"--json bare", []string{"version", "--json"}, true},
		{"--json=true", []string{"version", "--json=true"}, true},
		{"--json=1", []string{"version", "--json=1"}, true},
		{"--json=TRUE", []string{"version", "--json=TRUE"}, true},
		{"--json=false", []string{"version", "--json=false"}, false},
		{"unrelated", []string{"bogus", "--kb", "x"}, false},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			assert.Equal(t, tc.want, argsRequestJSON(tc.args))
		})
	}
}

func TestWantsJSONOutput(t *testing.T) {
	// Build a minimal *cobra.Command with the json flag directly so we test
	// the helper without going through cobra's parse pipeline. wantsJSONOutput
	// reads cmd.Flags() which on a fresh command equals LocalFlags().
	c := &cobra.Command{Use: "x"}
	c.Flags().Bool("json", false, "")
	assert.False(t, wantsJSONOutput(c), "default: --json unset")

	require.NoError(t, c.Flags().Set("json", "true"))
	assert.True(t, wantsJSONOutput(c), "--json=true honored")
}