mirror of
https://github.com/go-sylixos/elvish.git
synced 2024-12-15 03:37:52 +08:00
139 lines
3.3 KiB
Go
139 lines
3.3 KiB
Go
package edit
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/elves/elvish/pkg/cli/apptest"
|
|
"github.com/elves/elvish/pkg/cli/term"
|
|
"github.com/elves/elvish/pkg/eval"
|
|
"github.com/elves/elvish/pkg/eval/vals"
|
|
"github.com/elves/elvish/pkg/eval/vars"
|
|
"github.com/elves/elvish/pkg/parse"
|
|
"github.com/elves/elvish/pkg/store"
|
|
)
|
|
|
|
var Styles = apptest.Styles
|
|
|
|
type fixture struct {
|
|
Editor *Editor
|
|
TTYCtrl apptest.TTYCtrl
|
|
Evaler *eval.Evaler
|
|
Store store.Store
|
|
Home string
|
|
|
|
width int
|
|
codeCh <-chan string
|
|
errCh <-chan error
|
|
cleanup func()
|
|
}
|
|
|
|
func rc(codes ...string) func(*fixture) {
|
|
return func(f *fixture) { evals(f.Evaler, codes...) }
|
|
}
|
|
|
|
func assign(name string, val interface{}) func(*fixture) {
|
|
return func(f *fixture) {
|
|
f.Evaler.Global["temp"] = vars.NewReadOnly(val)
|
|
evals(f.Evaler, name+` = $temp`)
|
|
}
|
|
}
|
|
|
|
func storeOp(storeFn func(store.Store)) func(*fixture) {
|
|
return func(f *fixture) {
|
|
storeFn(f.Store)
|
|
// TODO(xiaq): Don't depend on this Elvish API.
|
|
evals(f.Evaler, "edit:history:fast-forward")
|
|
}
|
|
}
|
|
|
|
func setup(fns ...func(*fixture)) *fixture {
|
|
st, cleanupStore := store.MustGetTempStore()
|
|
home, cleanupFs := eval.InTempHome()
|
|
tty, ttyCtrl := apptest.NewFakeTTY()
|
|
ev := eval.NewEvaler()
|
|
ed := NewEditor(tty, ev, st)
|
|
ev.InstallModule("edit", ed.Ns())
|
|
evals(ev,
|
|
`use edit`,
|
|
// This is the same as the default prompt for non-root users. This makes
|
|
// sure that the tests will work when run as root.
|
|
"edit:prompt = { tilde-abbr $pwd; put '> ' }",
|
|
// This will simplify most tests against the terminal.
|
|
"edit:rprompt = { }")
|
|
f := &fixture{Editor: ed, TTYCtrl: ttyCtrl, Evaler: ev, Store: st, Home: home}
|
|
for _, fn := range fns {
|
|
fn(f)
|
|
}
|
|
_, f.width = tty.Size()
|
|
f.codeCh, f.errCh = apptest.StartReadCode(f.Editor.ReadCode)
|
|
f.cleanup = func() {
|
|
f.Editor.app.CommitEOF()
|
|
f.Wait()
|
|
cleanupFs()
|
|
cleanupStore()
|
|
}
|
|
return f
|
|
}
|
|
|
|
func (f *fixture) Wait() (string, error) {
|
|
return <-f.codeCh, <-f.errCh
|
|
}
|
|
|
|
func (f *fixture) Cleanup() {
|
|
f.cleanup()
|
|
}
|
|
|
|
func (f *fixture) MakeBuffer(args ...interface{}) *term.Buffer {
|
|
return term.NewBufferBuilder(f.width).MarkLines(args...).Buffer()
|
|
}
|
|
|
|
func (f *fixture) TestTTY(t *testing.T, args ...interface{}) {
|
|
t.Helper()
|
|
f.TTYCtrl.TestBuffer(t, f.MakeBuffer(args...))
|
|
}
|
|
|
|
func (f *fixture) TestTTYNotes(t *testing.T, args ...interface{}) {
|
|
t.Helper()
|
|
f.TTYCtrl.TestNotesBuffer(t, f.MakeBuffer(args...))
|
|
}
|
|
|
|
func feedInput(ttyCtrl apptest.TTYCtrl, s string) {
|
|
for _, r := range s {
|
|
ttyCtrl.Inject(term.K(r))
|
|
}
|
|
}
|
|
|
|
func evals(ev *eval.Evaler, codes ...string) {
|
|
for _, code := range codes {
|
|
src := parse.Source{Name: "[test]", Code: code}
|
|
op, err := ev.ParseAndCompile(src, nil)
|
|
if err != nil {
|
|
panic(fmt.Errorf("parse and compile %q: %s", code, err))
|
|
}
|
|
err = ev.Eval(op, eval.EvalCfg{})
|
|
if err != nil {
|
|
panic(fmt.Errorf("eval %q: %s", code, err))
|
|
}
|
|
}
|
|
}
|
|
|
|
func getGlobal(ev *eval.Evaler, name string) interface{} {
|
|
return ev.Global[name].Get()
|
|
}
|
|
|
|
func testGlobals(t *testing.T, ev *eval.Evaler, wantVals map[string]interface{}) {
|
|
t.Helper()
|
|
for name, wantVal := range wantVals {
|
|
testGlobal(t, ev, name, wantVal)
|
|
}
|
|
}
|
|
|
|
func testGlobal(t *testing.T, ev *eval.Evaler, name string, wantVal interface{}) {
|
|
t.Helper()
|
|
if val := getGlobal(ev, name); !vals.Equal(val, wantVal) {
|
|
t.Errorf("$%s = %s, want %s",
|
|
name, vals.Repr(val, vals.NoPretty), vals.Repr(wantVal, vals.NoPretty))
|
|
}
|
|
}
|