add more tests for proc (#439)
parent
b88ba14597
commit
d04b54243d
@ -0,0 +1,28 @@
|
|||||||
|
package proc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestShutdown(t *testing.T) {
|
||||||
|
SetTimeToForceQuit(time.Hour)
|
||||||
|
assert.Equal(t, time.Hour, delayTimeBeforeForceQuit)
|
||||||
|
|
||||||
|
var val int
|
||||||
|
called := AddWrapUpListener(func() {
|
||||||
|
val++
|
||||||
|
})
|
||||||
|
wrapUpListeners.notifyListeners()
|
||||||
|
called()
|
||||||
|
assert.Equal(t, 1, val)
|
||||||
|
|
||||||
|
called = AddShutdownListener(func() {
|
||||||
|
val += 2
|
||||||
|
})
|
||||||
|
shutdownListeners.notifyListeners()
|
||||||
|
called()
|
||||||
|
assert.Equal(t, 3, val)
|
||||||
|
}
|
@ -0,0 +1,8 @@
|
|||||||
|
package proc
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
func TestNopStopper(t *testing.T) {
|
||||||
|
// no panic
|
||||||
|
noopStopper.Stop()
|
||||||
|
}
|
Loading…
Reference in New Issue