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
|
//go:build linux
// +build linux
package signal // import "github.com/docker/docker/pkg/signal"
import (
"os"
"os/exec"
"syscall"
"testing"
"gotest.tools/v3/assert"
is "gotest.tools/v3/assert/cmp"
)
func buildTestBinary(t *testing.T, tmpdir string, prefix string) (string, string) {
t.Helper()
tmpDir, err := os.MkdirTemp(tmpdir, prefix)
assert.NilError(t, err)
exePath := tmpDir + "/" + prefix
wd, _ := os.Getwd()
testHelperCode := wd + "/testfiles/main.go"
cmd := exec.Command("go", "build", "-o", exePath, testHelperCode)
err = cmd.Run()
assert.NilError(t, err)
return exePath, tmpDir
}
func TestTrap(t *testing.T) {
var sigmap = []struct {
name string
signal os.Signal
multiple bool
}{
{"TERM", syscall.SIGTERM, false},
{"QUIT", syscall.SIGQUIT, true},
{"INT", os.Interrupt, false},
{"TERM", syscall.SIGTERM, true},
{"INT", os.Interrupt, true},
}
exePath, tmpDir := buildTestBinary(t, "", "main")
defer os.RemoveAll(tmpDir)
for _, v := range sigmap {
t.Run(v.name, func(t *testing.T) {
cmd := exec.Command(exePath)
cmd.Env = append(os.Environ(), "SIGNAL_TYPE="+v.name)
if v.multiple {
cmd.Env = append(cmd.Env, "IF_MULTIPLE=1")
}
err := cmd.Start()
assert.NilError(t, err)
err = cmd.Wait()
e, ok := err.(*exec.ExitError)
assert.Assert(t, ok, "expected exec.ExitError, got %T", e)
code := e.Sys().(syscall.WaitStatus).ExitStatus()
if v.multiple {
assert.Check(t, is.DeepEqual(128+int(v.signal.(syscall.Signal)), code))
} else {
assert.Check(t, is.Equal(99, code))
}
})
}
}
func TestDumpStacks(t *testing.T) {
directory, err := os.MkdirTemp("", "test-dump-tasks")
assert.Check(t, err)
defer os.RemoveAll(directory)
dumpPath, err := DumpStacks(directory)
assert.Check(t, err)
readFile, _ := os.ReadFile(dumpPath)
fileData := string(readFile)
assert.Check(t, is.Contains(fileData, "goroutine"))
}
func TestDumpStacksWithEmptyInput(t *testing.T) {
path, err := DumpStacks("")
assert.Check(t, err)
assert.Check(t, is.Equal(os.Stderr.Name(), path))
}
|
