elvish/pkg/eval/eval_test.go
Qi Xiao 53a1eca9af pkg/eval: Remove New*Source functions.
All call sites of such functions are replaced with simple &Source{...}
expressions.
2020-04-25 13:22:07 +01:00

108 lines
2.3 KiB
Go

package eval
import (
"reflect"
"strconv"
"sync"
"syscall"
"testing"
"github.com/elves/elvish/pkg/eval/vals"
)
func TestBuiltinPid(t *testing.T) {
pid := strconv.Itoa(syscall.Getpid())
builtinPid := vals.ToString(builtinNs["pid"].Get())
if builtinPid != pid {
t.Errorf(`ev.builtin["pid"] = %v, want %v`, builtinPid, pid)
}
}
func TestNumBgJobs(t *testing.T) {
Test(t,
That("put $num-bg-jobs").Puts("0"),
// TODO(xiaq): Test cases where $num-bg-jobs > 0. This cannot be done
// with { put $num-bg-jobs }& because the output channel may have
// already been closed when the closure is run.
)
}
func TestMiscEval(t *testing.T) {
Test(t,
// Pseudo-namespace E:
That("E:FOO=lorem; put $E:FOO").Puts("lorem"),
That("del E:FOO; put $E:FOO").Puts(""),
)
}
func TestMultipleEval(t *testing.T) {
texts := []string{"x=hello", "put $x"}
r := evalAndCollect(t, NewEvaler(), texts)
wantOuts := []interface{}{"hello"}
if r.exception != nil {
t.Errorf("eval %s => %v, want nil", texts, r.exception)
}
if !reflect.DeepEqual(r.valueOut, wantOuts) {
t.Errorf("eval %s outputs %v, want %v", texts, r.valueOut, wantOuts)
}
}
func TestConcurrentEval(t *testing.T) {
// Run this test with "go test -race".
ev := NewEvaler()
src := &Source{Name: "[test]"}
op, err := ev.ParseAndCompile(src)
if err != nil {
panic(err)
}
var wg sync.WaitGroup
wg.Add(2)
go func() {
ev.Eval(op, EvalCfg{})
wg.Done()
}()
go func() {
ev.Eval(op, EvalCfg{})
wg.Done()
}()
wg.Wait()
}
func BenchmarkOutputCaptureOverhead(b *testing.B) {
benchmarkOutputCapture(b.N, func(fm *Frame) {})
}
func BenchmarkOutputCaptureValues(b *testing.B) {
benchmarkOutputCapture(b.N, func(fm *Frame) {
fm.ports[1].Chan <- "test"
})
}
func BenchmarkOutputCaptureBytes(b *testing.B) {
bytesToWrite := []byte("test")
benchmarkOutputCapture(b.N, func(fm *Frame) {
fm.ports[1].File.Write(bytesToWrite)
})
}
func BenchmarkOutputCaptureMixed(b *testing.B) {
bytesToWrite := []byte("test")
benchmarkOutputCapture(b.N, func(fm *Frame) {
fm.ports[1].Chan <- false
fm.ports[1].File.Write(bytesToWrite)
})
}
func benchmarkOutputCapture(n int, f func(*Frame)) {
ev := NewEvaler()
defer ev.Close()
fm := NewTopFrame(ev, &Source{Name: "[benchmark]"}, []*Port{{}, {}, {}})
for i := 0; i < n; i++ {
captureOutput(fm, func(fm *Frame) error {
f(fm)
return nil
})
}
}