31 lines
1.0 KiB
Go
31 lines
1.0 KiB
Go
// Copyright 2022 PingCAP, Inc. Licensed under Apache-2.0.
|
|
|
|
package errors_test
|
|
|
|
import (
|
|
"context"
|
|
"net/url"
|
|
"testing"
|
|
|
|
"github.com/pingcap/errors"
|
|
berrors "github.com/pingcap/tidb/br/pkg/errors"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestIsContextCanceled(t *testing.T) {
|
|
require.False(t, berrors.IsContextCanceled(nil))
|
|
require.False(t, berrors.IsContextCanceled(errors.New("connection closed")))
|
|
require.True(t, berrors.IsContextCanceled(context.Canceled))
|
|
require.True(t, berrors.IsContextCanceled(context.DeadlineExceeded))
|
|
require.True(t, berrors.IsContextCanceled(errors.Trace(context.Canceled)))
|
|
require.True(t, berrors.IsContextCanceled(errors.Trace(context.DeadlineExceeded)))
|
|
require.True(t, berrors.IsContextCanceled(&url.Error{Err: context.Canceled}))
|
|
require.True(t, berrors.IsContextCanceled(&url.Error{Err: context.DeadlineExceeded}))
|
|
}
|
|
|
|
func TestEqual(t *testing.T) {
|
|
err := errors.Annotate(berrors.ErrPDBatchScanRegion, "test error equla")
|
|
r := berrors.ErrPDBatchScanRegion.Equal(err)
|
|
require.True(t, r)
|
|
}
|