File size: 2,191 Bytes
e36aeda | 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 94 95 96 | // 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 main
import (
"flag"
"internal/runtime/exithook"
"os"
"time"
_ "unsafe"
)
var modeflag = flag.String("mode", "", "mode to run in")
func main() {
flag.Parse()
switch *modeflag {
case "simple":
testSimple()
case "goodexit":
testGoodExit()
case "badexit":
testBadExit()
case "panics":
testPanics()
case "callsexit":
testHookCallsExit()
case "exit2":
testExit2()
default:
panic("unknown mode")
}
}
func testSimple() {
f1 := func() { println("foo") }
f2 := func() { println("bar") }
exithook.Add(exithook.Hook{F: f1})
exithook.Add(exithook.Hook{F: f2})
// no explicit call to os.Exit
}
func testGoodExit() {
f1 := func() { println("apple") }
f2 := func() { println("orange") }
exithook.Add(exithook.Hook{F: f1})
exithook.Add(exithook.Hook{F: f2})
// explicit call to os.Exit
os.Exit(0)
}
func testBadExit() {
f1 := func() { println("blog") }
f2 := func() { println("blix") }
f3 := func() { println("blek") }
f4 := func() { println("blub") }
f5 := func() { println("blat") }
exithook.Add(exithook.Hook{F: f1})
exithook.Add(exithook.Hook{F: f2, RunOnFailure: true})
exithook.Add(exithook.Hook{F: f3})
exithook.Add(exithook.Hook{F: f4, RunOnFailure: true})
exithook.Add(exithook.Hook{F: f5})
os.Exit(1)
}
func testPanics() {
f1 := func() { println("ok") }
f2 := func() { panic("BADBADBAD") }
f3 := func() { println("good") }
exithook.Add(exithook.Hook{F: f1, RunOnFailure: true})
exithook.Add(exithook.Hook{F: f2, RunOnFailure: true})
exithook.Add(exithook.Hook{F: f3, RunOnFailure: true})
os.Exit(0)
}
func testHookCallsExit() {
f1 := func() { println("ok") }
f2 := func() { os.Exit(1) }
f3 := func() { println("good") }
exithook.Add(exithook.Hook{F: f1, RunOnFailure: true})
exithook.Add(exithook.Hook{F: f2, RunOnFailure: true})
exithook.Add(exithook.Hook{F: f3, RunOnFailure: true})
os.Exit(1)
}
func testExit2() {
f1 := func() { time.Sleep(100 * time.Millisecond) }
exithook.Add(exithook.Hook{F: f1})
for range 10 {
go os.Exit(0)
}
os.Exit(0)
}
|