1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
package shell_test
import (
"embed"
"errors"
"io"
"os"
"path/filepath"
"testing"
"time"
"src.elv.sh/pkg/daemon"
"src.elv.sh/pkg/daemon/daemondefs"
"src.elv.sh/pkg/eval"
"src.elv.sh/pkg/eval/evaltest"
"src.elv.sh/pkg/eval/vars"
"src.elv.sh/pkg/prog/progtest"
"src.elv.sh/pkg/shell"
"src.elv.sh/pkg/testutil"
)
//go:embed *.elvts
var transcripts embed.FS
var sigCHLDName = ""
func TestTranscripts(t *testing.T) {
evaltest.TestTranscriptsInFS(t, transcripts,
"elvish-in-global", progtest.ElvishInGlobal(&shell.Program{}),
"elvish-with-activate-daemon-in-global", progtest.ElvishInGlobal(
&shell.Program{ActivateDaemon: inProcessActivateFunc(t)}),
"elvish-with-bad-activate-daemon-in-global", progtest.ElvishInGlobal(
&shell.Program{
ActivateDaemon: func(io.Writer, *daemondefs.SpawnConfig) (daemondefs.Client, error) {
return nil, errors.New("fake error")
},
}),
"kill-wait-in-global", addGlobal("kill-wait",
testutil.Scaled(10*time.Millisecond).String()),
"sigchld-name-in-global", addGlobal("sigchld-name", sigCHLDName),
"in-temp-home", func(t *testing.T) { testutil.InTempHome(t) },
)
}
func inProcessActivateFunc(t *testing.T) daemondefs.ActivateFunc {
return func(stderr io.Writer, cfg *daemondefs.SpawnConfig) (daemondefs.Client, error) {
// Start an in-process daemon.
//
// Create the socket in a temporary directory. This is necessary because
// we don't do enough mocking in the tests yet, and cfg.SockPath will
// point to the socket used by real Elvish sessions.
dir := testutil.TempDir(t)
sockPath := filepath.Join(dir, "sock")
sigCh := make(chan os.Signal)
readyCh := make(chan struct{})
daemonDone := make(chan struct{})
go func() {
// Unlike the socket path, we do honor cfg.DBPath; this is because
// we run tests in a temporary HOME, so there's no risk of using the
// DB of real Elvish sessions.
daemon.Serve(sockPath, cfg.DbPath,
daemon.ServeOpts{Ready: readyCh, Signals: sigCh})
close(daemonDone)
}()
t.Cleanup(func() {
close(sigCh)
select {
case <-daemonDone:
case <-time.After(testutil.Scaled(2 * time.Second)):
t.Errorf("timed out waiting for daemon to quit")
}
})
select {
case <-readyCh:
// Do nothing
case <-time.After(testutil.Scaled(2 * time.Second)):
t.Fatalf("timed out waiting for daemon to start")
}
// Connect to it.
return daemon.NewClient(sockPath), nil
}
}
func addGlobal(name string, value any) func(ev *eval.Evaler) {
return func(ev *eval.Evaler) {
ev.ExtendGlobal(eval.BuildNs().AddVar(name, vars.NewReadOnly(value)))
}
}
|
