mirror of
https://github.com/qdm12/gluetun.git
synced 2026-07-20 17:36:28 +02:00
Quentin McGaw
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package cleanup
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"go.uber.org/mock/gomock"
|
|
)
|
|
|
|
func Test_Cleanups(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ctrl := gomock.NewController(t)
|
|
|
|
var ACloseCalled, BCloseCalled, CCloseCalled bool
|
|
var (
|
|
AErr error
|
|
BErr = errors.New("B failed")
|
|
CErr = errors.New("C failed")
|
|
)
|
|
|
|
var cleanups Cleanups
|
|
cleanups.Add("cleaning up A", 5, func() error {
|
|
ACloseCalled = true
|
|
return AErr
|
|
})
|
|
|
|
cleanups.Add("cleaning up B", 3, func() error {
|
|
BCloseCalled = true
|
|
return BErr
|
|
})
|
|
|
|
cleanups.Add("cleaning up C", 2, func() error {
|
|
CCloseCalled = true
|
|
return CErr
|
|
})
|
|
|
|
logger := NewMockLogger(ctrl)
|
|
prevCall := logger.EXPECT().Debug("cleaning up C...")
|
|
prevCall = logger.EXPECT().Error("failed cleaning up C: C failed").After(prevCall)
|
|
prevCall = logger.EXPECT().Debug("cleaning up B...").After(prevCall)
|
|
prevCall = logger.EXPECT().Error("failed cleaning up B: B failed").After(prevCall)
|
|
logger.EXPECT().Debug("cleaning up A...").After(prevCall)
|
|
|
|
cleanups.Cleanup(logger)
|
|
|
|
cleanups.Cleanup(logger) // run twice should not close already closed
|
|
|
|
for _, cleanup := range cleanups {
|
|
assert.True(t, cleanup.done)
|
|
}
|
|
|
|
assert.True(t, ACloseCalled)
|
|
assert.True(t, BCloseCalled)
|
|
assert.True(t, CCloseCalled)
|
|
}
|