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
|
// Copyright 2022 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package runtime_test
import (
"internal/platform"
"internal/testenv"
"os/exec"
"runtime"
"strings"
"testing"
)
func TestExitHooks(t *testing.T) {
bmodes := []string{""}
if testing.Short() {
t.Skip("skipping due to -short")
}
// Note the HasCGO() test below; this is to prevent the test
// running if CGO_ENABLED=0 is in effect.
haverace := platform.RaceDetectorSupported(runtime.GOOS, runtime.GOARCH)
if haverace && testenv.HasCGO() {
bmodes = append(bmodes, "-race")
}
for _, bmode := range bmodes {
scenarios := []struct {
mode string
expected string
musthave []string
}{
{
mode: "simple",
expected: "bar foo",
},
{
mode: "goodexit",
expected: "orange apple",
},
{
mode: "badexit",
expected: "blub blix",
},
{
mode: "panics",
musthave: []string{
"fatal error: exit hook invoked panic",
"main.testPanics",
},
},
{
mode: "callsexit",
musthave: []string{
"fatal error: exit hook invoked exit",
},
},
{
mode: "exit2",
expected: "",
},
}
exe, err := buildTestProg(t, "testexithooks", bmode)
if err != nil {
t.Fatal(err)
}
bt := ""
if bmode != "" {
bt = " bmode: " + bmode
}
for _, s := range scenarios {
cmd := exec.Command(exe, []string{"-mode", s.mode}...)
out, _ := cmd.CombinedOutput()
outs := strings.ReplaceAll(string(out), "\n", " ")
outs = strings.TrimSpace(outs)
if s.expected != "" && s.expected != outs {
t.Fatalf("failed%s mode %s: wanted %q\noutput:\n%s", bt,
s.mode, s.expected, outs)
}
for _, need := range s.musthave {
if !strings.Contains(outs, need) {
t.Fatalf("failed mode %s: output does not contain %q\noutput:\n%s",
s.mode, need, outs)
}
}
if s.expected == "" && s.musthave == nil && outs != "" {
t.Errorf("failed mode %s: wanted no output\noutput:\n%s", s.mode, outs)
}
}
}
}
|
