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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
package daemon
import (
"io"
"net"
"os"
"runtime"
"testing"
"time"
"src.elv.sh/pkg/daemon/daemondefs"
"src.elv.sh/pkg/must"
"src.elv.sh/pkg/testutil"
)
func TestActivate_ConnectsToExistingServer(t *testing.T) {
setup(t)
startServer(t, cli("sock", "db"))
_, err := Activate(io.Discard,
&daemondefs.SpawnConfig{DbPath: "db", SockPath: "sock", RunDir: "."})
if err != nil {
t.Errorf("got error %v, want nil", err)
}
}
func TestActivate_SpawnsNewServer(t *testing.T) {
activated := 0
setupForActivate(t, func(name string, argv []string, attr *os.ProcAttr) error {
startServer(t, argv)
activated++
return nil
})
_, err := Activate(io.Discard,
&daemondefs.SpawnConfig{DbPath: "db", SockPath: "sock", RunDir: "."})
if err != nil {
t.Errorf("got error %v, want nil", err)
}
if activated != 1 {
t.Errorf("got activated %v times, want 1", activated)
}
}
func TestActivate_RemovesHangingSocketAndSpawnsNewServer(t *testing.T) {
activated := 0
setupForActivate(t, func(name string, argv []string, attr *os.ProcAttr) error {
startServer(t, argv)
activated++
return nil
})
makeHangingUnixSocket(t, "sock")
_, err := Activate(io.Discard,
&daemondefs.SpawnConfig{DbPath: "db", SockPath: "sock", RunDir: "."})
if err != nil {
t.Errorf("got error %v, want nil", err)
}
if activated != 1 {
t.Errorf("got activated %v times, want 1", activated)
}
}
func TestActivate_FailsIfCannotStatSock(t *testing.T) {
setup(t)
// Build a path for which Lstat will return a non-nil err such that
// os.IsNotExist(err) is false.
badSockPath := ""
if runtime.GOOS != "windows" {
// POSIX lstat(2) returns ENOTDIR instead of ENOENT if a path prefix is
// not a directory.
must.CreateEmpty("not-dir")
badSockPath = "not-dir/sock"
} else {
// Use a syntactically invalid drive letter on Windows.
badSockPath = `CD:\sock`
}
_, err := Activate(io.Discard,
&daemondefs.SpawnConfig{DbPath: "db", SockPath: badSockPath, RunDir: "."})
if err == nil {
t.Errorf("got error nil, want non-nil")
}
}
func TestActivate_FailsIfCannotDialSock(t *testing.T) {
setup(t)
must.CreateEmpty("sock")
_, err := Activate(io.Discard,
&daemondefs.SpawnConfig{DbPath: "db", SockPath: "sock", RunDir: "."})
if err == nil {
t.Errorf("got error nil, want non-nil")
}
}
func setupForActivate(t *testing.T, f func(string, []string, *os.ProcAttr) error) {
setup(t)
testutil.Set(t, &startProcess, f)
scaleDuration(t, &daemonSpawnTimeout)
scaleDuration(t, &daemonKillTimeout)
}
func scaleDuration(t *testing.T, d *time.Duration) {
testutil.Set(t, d, testutil.Scaled(*d))
}
func makeHangingUnixSocket(t *testing.T, path string) {
t.Helper()
l, err := net.Listen("unix", path)
if err != nil {
t.Fatal(err)
}
// We need to call l.Close() to make the socket hang, but that will
// helpfully remove the socket file. Work around this by renaming the socket
// file.
err = os.Rename(path, path+".save")
if err != nil {
t.Fatal(err)
}
l.Close()
err = os.Rename(path+".save", path)
if err != nil {
t.Fatal(err)
}
}
|
