mirror of
https://github.com/aljazceru/kata-containers.git
synced 2025-12-26 10:34:24 +01:00
a02a8bda66
To prepare for merging into kata-containers repository. Signed-off-by: Peng Tao <bergwolf@hyper.sh>
43 lines
637 B
Go
43 lines
637 B
Go
// Copyright (c) 2017 Intel Corporation
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
package main
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
var testFoo string
|
|
|
|
func testFunc() {
|
|
testFoo = "bar"
|
|
}
|
|
|
|
func TestExit(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
var testExitStatus int
|
|
exitFunc = func(status int) {
|
|
testExitStatus = status
|
|
}
|
|
|
|
defer func() {
|
|
exitFunc = os.Exit
|
|
}()
|
|
|
|
// test with no atexit functions added.
|
|
exit(1)
|
|
assert.Equal(testExitStatus, 1)
|
|
|
|
// test with a function added to the atexit list.
|
|
atexit(testFunc)
|
|
exit(0)
|
|
assert.Equal(testFoo, "bar")
|
|
assert.Equal(testExitStatus, 0)
|
|
}
|