-
-
Notifications
You must be signed in to change notification settings - Fork 54
/
ctl_test.go
114 lines (105 loc) · 2.49 KB
/
ctl_test.go
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
package immortal
import (
"io/ioutil"
"net"
"os"
"path/filepath"
"testing"
)
func TestGetStatus(t *testing.T) {
ctl := &Controller{}
if _, err := ctl.GetStatus("/dev/null"); err == nil {
t.Errorf("Expecting an error")
}
}
func TestSendSignal(t *testing.T) {
ctl := &Controller{}
if _, err := ctl.SendSignal("/dev/null", "test"); err == nil {
t.Errorf("Expecting an error")
}
}
func TestFindServices(t *testing.T) {
ctl := &Controller{}
dir, err := ioutil.TempDir("", "FindServices")
if err != nil {
t.Error(err)
}
defer os.RemoveAll(dir)
tdir := filepath.Join(dir, "test")
os.Mkdir(tdir, 0700)
_, err = net.Listen("unix", filepath.Join(tdir, "immortal.sock"))
if err != nil {
t.Error(err)
}
s, err := ctl.FindServices(dir)
if err != nil {
t.Error(err)
}
expect(t, 1, len(s))
}
func TestFindServicesNonexistent(t *testing.T) {
ctl := &Controller{}
_, err := ctl.FindServices("/dev/null/non-existent")
if err == nil {
t.Errorf("Expecting an error")
}
}
func TestPurgeServices(t *testing.T) {
ctl := &Controller{}
dir, err := ioutil.TempDir("", "PurgeServices")
if err != nil {
t.Error(err)
}
defer os.RemoveAll(dir)
err = ctl.PurgeServices(dir)
if err == nil {
t.Error("Expecting an error")
}
tdir := filepath.Join(dir, "test")
os.Mkdir(tdir, 0700)
os.Create(filepath.Join(tdir, "lock"))
os.Create(filepath.Join(tdir, "immortal.sock"))
os.Create(filepath.Join(tdir, "f3"))
files, _ := ioutil.ReadDir(tdir)
expect(t, 3, len(files))
err = ctl.PurgeServices(tdir)
if err == nil {
t.Error("Expecting and error")
}
files, _ = ioutil.ReadDir(tdir)
expect(t, 3, len(files))
err = ctl.PurgeServices(filepath.Join(tdir, "f3"))
if err == nil {
t.Error("Expecting and error")
}
os.Remove(filepath.Join(tdir, "f3"))
err = ctl.PurgeServices(filepath.Join(tdir, "f3"))
if err != nil {
t.Error(err)
}
files, _ = ioutil.ReadDir(tdir)
expect(t, 0, len(files))
tdir = filepath.Join(dir, "test", "root")
os.MkdirAll(tdir, 0700)
os.Create(filepath.Join(dir, "test", "lock"))
err = ctl.PurgeServices(filepath.Join(dir, "test", "immortal.sock"))
if err == nil {
t.Error("Expecting and error")
}
err = ctl.PurgeServices("/dev/null/non-existent")
if err == nil {
t.Error("Expecting and error")
}
}
func TestRun(t *testing.T) {
ctl := &Controller{}
out, err := ctl.Run("echo -n immortal")
if err != nil {
t.Fatal(err)
}
expect(t, "immortal", string(out))
_, err = ctl.Run("/dev/null/non-existent -n immortal")
if err == nil {
t.Fatal("Expecting an errro")
}
}