mirror of
https://github.com/gotenberg/gotenberg.git
synced 2026-08-08 16:42:15 +01:00
43 lines
1.2 KiB
Go
43 lines
1.2 KiB
Go
package xcontext
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/thecodingmachine/gotenberg/internal/pkg/xerror"
|
|
"github.com/thecodingmachine/gotenberg/internal/pkg/xtime"
|
|
"github.com/thecodingmachine/gotenberg/test"
|
|
)
|
|
|
|
func TestMustHandleError(t *testing.T) {
|
|
previousErr := errors.New("previous error")
|
|
logger := test.DebugLogger()
|
|
// context should not have an error.
|
|
ctx, cancel := WithTimeout(logger, 5)
|
|
defer cancel()
|
|
err := MustHandleError(ctx, previousErr)
|
|
assert.Equal(t, previousErr, err)
|
|
// should panic.
|
|
ctx, cancel = WithTimeout(logger, 5)
|
|
defer cancel()
|
|
assert.Panics(t, func() {
|
|
MustHandleError(ctx, nil)
|
|
})
|
|
// context should timed out.
|
|
ctx, cancel = WithTimeout(logger, 0.5)
|
|
defer cancel()
|
|
time.Sleep(xtime.Duration(1))
|
|
err = MustHandleError(ctx, previousErr)
|
|
xerr := test.AssertError(t, err)
|
|
assert.Equal(t, xerror.TimeoutCode, xerror.Code(xerr))
|
|
// context should have an error different
|
|
// than context.DeadlineExceeded.
|
|
ctx, cancel = WithTimeout(logger, 5)
|
|
cancel()
|
|
err = MustHandleError(ctx, previousErr)
|
|
xerr = test.AssertError(t, err)
|
|
assert.Equal(t, xerror.InternalCode, xerror.Code(xerr))
|
|
}
|