elvish/pkg/edit/completion_test.go
Qi Xiao a1b30deb86 pkg/eval/evaltest: Rename TestWithFullSetup -> TestWithSetup.
The old TestWithSetup is now more accurately called TestWithEvalerSetup.
2023-07-22 23:04:18 +01:00

240 lines
6.6 KiB
Go

package edit
import (
"testing"
"src.elv.sh/pkg/cli/term"
"src.elv.sh/pkg/eval"
"src.elv.sh/pkg/eval/errs"
. "src.elv.sh/pkg/eval/evaltest"
"src.elv.sh/pkg/eval/vals"
"src.elv.sh/pkg/testutil"
"src.elv.sh/pkg/ui"
)
func TestCompletionAddon(t *testing.T) {
f := setup(t)
testutil.ApplyDir(testutil.Dir{"a": "", "b": ""})
feedInput(f.TTYCtrl, "echo \t")
f.TestTTY(t,
"~> echo a \n", Styles,
" vvvv __",
" COMPLETING argument ", Styles,
"********************* ", term.DotHere, "\n",
"a b", Styles,
"+ ",
)
}
func TestCompletionAddon_CompletesLongestCommonPrefix(t *testing.T) {
f := setup(t)
testutil.ApplyDir(testutil.Dir{"foo1": "", "foo2": "", "foo": "", "fox": ""})
feedInput(f.TTYCtrl, "echo \t")
f.TestTTY(t,
"~> echo fo", Styles,
" vvvv", term.DotHere,
)
feedInput(f.TTYCtrl, "\t")
f.TestTTY(t,
"~> echo foo \n", Styles,
" vvvv ____",
" COMPLETING argument ", Styles,
"********************* ", term.DotHere, "\n",
"foo foo1 foo2 fox", Styles,
"+++ ",
)
}
func TestCompletionAddon_AppliesAutofix(t *testing.T) {
f := setup(t)
fooNs := eval.BuildNs().AddGoFn("a", func() {}).AddGoFn("b", func() {}).Ns()
f.Evaler.AddModule("foo", fooNs)
feedInput(f.TTYCtrl, "foo:")
f.TestTTY(t,
"~> foo:", Styles,
" !!!!", term.DotHere, "\n",
"Ctrl-A autofix: use foo Tab Enter autofix first", Styles,
"++++++ +++ +++++",
)
feedInput(f.TTYCtrl, "\t")
f.TestTTY(t,
"~> foo:a\n", Styles,
" VVVVV",
" COMPLETING command ", Styles,
"******************** ", term.DotHere, "\n",
"foo:a foo:b", Styles,
"+++++ ",
)
}
func TestCompleteFilename(t *testing.T) {
f := setup(t)
testutil.ApplyDir(testutil.Dir{"d": testutil.Dir{"a": "", "b": ""}})
evals(f.Evaler, `var @cands = (edit:complete-filename ls ./d/a)`)
testGlobal(t, f.Evaler,
"cands",
vals.MakeList(
complexItem{Stem: "./d/a", CodeSuffix: " ", Display: ui.T("./d/a")},
complexItem{Stem: "./d/b", CodeSuffix: " ", Display: ui.T("./d/b")}))
testThatOutputErrorIsBubbled(t, f, "edit:complete-filename ls ''")
}
func TestComplexCandidate(t *testing.T) {
TestWithEvalerSetup(t, func(ev *eval.Evaler) {
ev.ExtendGlobal(eval.BuildNs().AddGoFn("cc", complexCandidate))
},
That("cc a/b").Puts(complexItem{Stem: "a/b"}),
That("cc a/b &code-suffix=' '").Puts(complexItem{Stem: "a/b", CodeSuffix: " "}),
That("cc a/b &code-suffix=' ' &display=A/B").Puts(
complexItem{"a/b", " ", ui.T("A/B")}),
That("cc a/b &code-suffix=' ' &display=(styled A/B red)").Puts(
complexItem{"a/b", " ", ui.T("A/B", ui.FgRed)}),
That("cc a/b &code-suffix=' ' &display=[]").Throws(
errs.BadValue{What: "&display", Valid: "string or styled", Actual: "[]"}),
That("kind-of (cc stem)").Puts("map"),
That("keys (cc stem)").Puts("stem", "code-suffix", "display"),
That("repr (cc a/b &code-suffix=' ' &display=A/B)").Prints(
"(edit:complex-candidate a/b &code-suffix=' ' &display=(ui:text A/B))\n"),
That("eq (cc stem) (cc stem)").Puts(true),
That("eq (cc stem &code-suffix=' ') (cc stem)").Puts(false),
That("eq (cc stem &display=STEM) (cc stem)").Puts(false),
That("put [&(cc stem)=value][(cc stem)]").Puts("value"),
That("put (cc a/b &code-suffix=' ' &display=A/B)[stem code-suffix display]").
Puts("a/b", " ", ui.T("A/B")),
)
}
func TestComplexCandidate_InEditModule(t *testing.T) {
// A sanity check that the complex-candidate command is part of the edit
// module.
f := setup(t)
evals(f.Evaler, `var stem = (edit:complex-candidate stem)[stem]`)
testGlobal(t, f.Evaler, "stem", "stem")
}
func TestCompletionArgCompleter_ArgsAndValueOutput(t *testing.T) {
f := setup(t)
evals(f.Evaler,
`var foo-args = []`,
`fn foo { }`,
`set edit:completion:arg-completer[foo] = {|@args|
set foo-args = $args
put 1val
edit:complex-candidate 2val &display=2VAL
}`)
feedInput(f.TTYCtrl, "foo foo1 foo2 \t")
f.TestTTY(t,
"~> foo foo1 foo2 1val\n", Styles,
" vvv ____",
" COMPLETING argument ", Styles,
"********************* ", term.DotHere, "\n",
"1val 2VAL", Styles,
"++++ ",
)
testGlobal(t, f.Evaler,
"foo-args", vals.MakeList("foo", "foo1", "foo2", ""))
}
func TestCompletionArgCompleter_BytesOutput(t *testing.T) {
f := setup(t)
evals(f.Evaler,
`fn foo { }`,
`set edit:completion:arg-completer[foo] = {|@args|
echo 1val
echo 2val
}`)
feedInput(f.TTYCtrl, "foo foo1 foo2 \t")
f.TestTTY(t,
"~> foo foo1 foo2 1val\n", Styles,
" vvv ____",
" COMPLETING argument ", Styles,
"********************* ", term.DotHere, "\n",
"1val 2val", Styles,
"++++ ",
)
}
func TestCompleteSudo(t *testing.T) {
f := setup(t)
evals(f.Evaler,
`fn foo { }`,
`set edit:completion:arg-completer[foo] = {|@args|
echo val1
echo val2
}`,
`var @cands = (edit:complete-sudo sudo foo '')`)
testGlobal(t, f.Evaler, "cands", vals.MakeList("val1", "val2"))
}
func TestCompletionMatcher(t *testing.T) {
f := setup(t)
testutil.ApplyDir(testutil.Dir{"foo": "", "oof": ""})
evals(f.Evaler, `set edit:completion:matcher[''] = $edit:match-substr~`)
feedInput(f.TTYCtrl, "echo f\t")
f.TestTTY(t,
"~> echo foo \n", Styles,
" vvvv ____",
" COMPLETING argument ", Styles,
"********************* ", term.DotHere, "\n",
"foo oof", Styles,
"+++ ",
)
}
func TestBuiltinMatchers(t *testing.T) {
f := setup(t)
evals(f.Evaler,
`var @prefix = (edit:match-prefix ab [ab abc cab acb ba [ab] [a b] [b a]])`,
`var @substr = (edit:match-substr ab [ab abc cab acb ba [ab] [a b] [b a]])`,
`var @subseq = (edit:match-subseq ab [ab abc cab acb ba [ab] [a b] [b a]])`,
)
testGlobals(t, f.Evaler, map[string]any{
"prefix": vals.MakeList(true, true, false, false, false, false, false, false),
"substr": vals.MakeList(true, true, true, false, false, true, false, false),
"subseq": vals.MakeList(true, true, true, true, false, true, true, false),
})
testThatOutputErrorIsBubbled(t, f, "edit:match-prefix ab [ab]")
}
func TestBuiltinMatchers_Options(t *testing.T) {
f := setup(t)
// The two options work identically on all the builtin matchers, so we only
// test for match-prefix for simplicity.
evals(f.Evaler,
`var @a = (edit:match-prefix &ignore-case ab [abc aBc AbC])`,
`var @b = (edit:match-prefix &ignore-case aB [abc aBc AbC])`,
`var @c = (edit:match-prefix &smart-case ab [abc aBc Abc])`,
`var @d = (edit:match-prefix &smart-case aB [abc aBc AbC])`,
)
testGlobals(t, f.Evaler, map[string]any{
"a": vals.MakeList(true, true, true),
"b": vals.MakeList(true, true, true),
"c": vals.MakeList(true, true, true),
"d": vals.MakeList(false, true, false),
})
testThatOutputErrorIsBubbled(t, f, "edit:match-prefix &ignore-case ab [ab]")
}