mirror of
https://github.com/genuinetools/reg.git
synced 2024-09-12 06:11:03 -04:00
83 lines
2 KiB
Go
83 lines
2 KiB
Go
![]() |
// +build linux
|
||
|
|
||
|
package signal // import "github.com/docker/docker/pkg/signal"
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"io/ioutil"
|
||
|
"os"
|
||
|
"os/exec"
|
||
|
"syscall"
|
||
|
"testing"
|
||
|
|
||
|
"gotest.tools/assert"
|
||
|
is "gotest.tools/assert/cmp"
|
||
|
)
|
||
|
|
||
|
func buildTestBinary(t *testing.T, tmpdir string, prefix string) (string, string) {
|
||
|
tmpDir, err := ioutil.TempDir(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 {
|
||
|
cmd := exec.Command(exePath)
|
||
|
cmd.Env = append(os.Environ(), fmt.Sprintf("SIGNAL_TYPE=%s", v.name))
|
||
|
if v.multiple {
|
||
|
cmd.Env = append(cmd.Env, "IF_MULTIPLE=1")
|
||
|
}
|
||
|
err := cmd.Start()
|
||
|
assert.NilError(t, err)
|
||
|
err = cmd.Wait()
|
||
|
if e, ok := err.(*exec.ExitError); ok {
|
||
|
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))
|
||
|
}
|
||
|
continue
|
||
|
}
|
||
|
t.Fatal("process didn't end with any error")
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|
||
|
func TestDumpStacks(t *testing.T) {
|
||
|
directory, err := ioutil.TempDir("", "test-dump-tasks")
|
||
|
assert.Check(t, err)
|
||
|
defer os.RemoveAll(directory)
|
||
|
dumpPath, err := DumpStacks(directory)
|
||
|
assert.Check(t, err)
|
||
|
readFile, _ := ioutil.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))
|
||
|
}
|