|
@@ -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)
|
|
|
+}
|