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
|
//go:build darwin || linux
package signal
import (
"os"
"syscall"
"testing"
"time"
)
func TestCatchAll(t *testing.T) {
sigs := make(chan os.Signal, 1)
CatchAll(sigs)
defer StopCatch(sigs)
listOfSignals := map[string]string{
"CONT": syscall.SIGCONT.String(),
"HUP": syscall.SIGHUP.String(),
"CHLD": syscall.SIGCHLD.String(),
"ILL": syscall.SIGILL.String(),
"FPE": syscall.SIGFPE.String(),
"CLD": syscall.SIGCLD.String(),
}
for sigStr := range listOfSignals {
if signal, ok := SignalMap[sigStr]; ok {
_ = syscall.Kill(syscall.Getpid(), signal)
s := <-sigs
if s.String() != signal.String() {
t.Errorf("expected: %q, got: %q", signal, s)
}
}
}
}
func TestCatchAllIgnoreSigUrg(t *testing.T) {
sigs := make(chan os.Signal, 1)
CatchAll(sigs)
defer StopCatch(sigs)
err := syscall.Kill(syscall.Getpid(), syscall.SIGURG)
if err != nil {
t.Fatal(err)
}
timer := time.NewTimer(1 * time.Second)
defer timer.Stop()
select {
case <-timer.C:
case s := <-sigs:
t.Fatalf("expected no signals to be handled, but received %q", s.String())
}
}
func TestStopCatch(t *testing.T) {
signal := SignalMap["HUP"]
channel := make(chan os.Signal, 1)
CatchAll(channel)
_ = syscall.Kill(syscall.Getpid(), signal)
signalString := <-channel
if signalString.String() != signal.String() {
t.Errorf("expected: %q, got: %q", signal, signalString)
}
StopCatch(channel)
_, ok := <-channel
if ok {
t.Error("expected: !ok, got: ok")
}
}
|
