1
0
Fork 0
mirror of https://github.com/moby/moby.git synced 2022-11-09 12:21:53 -05:00
moby--moby/pkg/signal/trap_linux_test.go
Daniel Nephin 4f0d95fa6e Add canonical import comment
Signed-off-by: Daniel Nephin <dnephin@docker.com>
2018-02-05 16:51:57 -05:00

82 lines
2 KiB
Go

// +build linux
package signal // import "github.com/docker/docker/pkg/signal"
import (
"fmt"
"io/ioutil"
"os"
"os/exec"
"syscall"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func buildTestBinary(t *testing.T, tmpdir string, prefix string) (string, string) {
tmpDir, err := ioutil.TempDir(tmpdir, prefix)
require.NoError(t, err)
exePath := tmpDir + "/" + prefix
wd, _ := os.Getwd()
testHelperCode := wd + "/testfiles/main.go"
cmd := exec.Command("go", "build", "-o", exePath, testHelperCode)
err = cmd.Run()
require.NoError(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()
require.NoError(t, err)
err = cmd.Wait()
if e, ok := err.(*exec.ExitError); ok {
code := e.Sys().(syscall.WaitStatus).ExitStatus()
if v.multiple {
assert.Equal(t, 128+int(v.signal.(syscall.Signal)), code)
} else {
assert.Equal(t, 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.NoError(t, err)
defer os.RemoveAll(directory)
dumpPath, err := DumpStacks(directory)
assert.NoError(t, err)
readFile, _ := ioutil.ReadFile(dumpPath)
fileData := string(readFile)
assert.Contains(t, fileData, "goroutine")
}
func TestDumpStacksWithEmptyInput(t *testing.T) {
path, err := DumpStacks("")
assert.NoError(t, err)
assert.Equal(t, os.Stderr.Name(), path)
}