Asai Neko
210b8b08ce
All checks were successful
Server Check Build (NixCN CMS) TeamCity build finished
Signed-off-by: Asai Neko <sugar@sne.moe>
104 lines
2.6 KiB
Go
104 lines
2.6 KiB
Go
package exception
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestContextWithServiceAndEndpoint(t *testing.T) {
|
|
ctx := context.Background()
|
|
ctx = ContextWithService(ctx, "my_service")
|
|
ctx = ContextWithEndpoint(ctx, "my_endpoint")
|
|
|
|
svc := extractFromCtx(ctx, ctxKeyService, "fallback")
|
|
ep := extractFromCtx(ctx, ctxKeyEndpoint, "fallback")
|
|
|
|
assert.Equal(t, "my_service", svc)
|
|
assert.Equal(t, "my_endpoint", ep)
|
|
}
|
|
|
|
func TestExtractFromCtxFallback(t *testing.T) {
|
|
ctx := context.Background()
|
|
val := extractFromCtx(ctx, ctxKeyService, "default")
|
|
assert.Equal(t, "default", val)
|
|
}
|
|
|
|
func TestBuilderString(t *testing.T) {
|
|
// Use proper 3-char endpoint and service to form a 13-char code
|
|
ctx := ContextWithService(context.Background(), ServiceAuthToken)
|
|
ctx = ContextWithEndpoint(ctx, EndpointAuthToken)
|
|
|
|
b := New(
|
|
WithStatus(StatusSuccess),
|
|
WithType(TypeCommon),
|
|
WithOriginal(CommonSuccess),
|
|
).Throw(ctx)
|
|
|
|
s := b.String()
|
|
require.Len(t, s, 13, "error code must be 13 chars")
|
|
assert.Equal(t, StatusSuccess, string(s[0]))
|
|
}
|
|
|
|
func TestBuilderThrowSuccess(t *testing.T) {
|
|
ctx := ContextWithService(context.Background(), ServiceAuthToken)
|
|
ctx = ContextWithEndpoint(ctx, EndpointAuthToken)
|
|
|
|
b := New(
|
|
WithStatus(StatusSuccess),
|
|
WithType(TypeCommon),
|
|
WithOriginal(CommonSuccess),
|
|
).Throw(ctx)
|
|
|
|
assert.Equal(t, CommonSuccess, b.Original)
|
|
assert.NotEmpty(t, b.ErrorCode)
|
|
}
|
|
|
|
func TestBuilderThrowWithError(t *testing.T) {
|
|
ctx := ContextWithService(context.Background(), ServiceAuthToken)
|
|
ctx = ContextWithEndpoint(ctx, EndpointAuthToken)
|
|
|
|
origErr := errors.New("something went wrong")
|
|
b := New(
|
|
WithStatus(StatusUser),
|
|
WithType(TypeSpecific),
|
|
WithOriginal(CommonErrorUnauthorized),
|
|
WithError(origErr),
|
|
).Throw(ctx)
|
|
|
|
assert.Equal(t, origErr, b.Error)
|
|
assert.Equal(t, StatusUser, b.Status)
|
|
}
|
|
|
|
func TestBuilderWithOptionsApplied(t *testing.T) {
|
|
err := errors.New("test")
|
|
b := New(
|
|
WithStatus(StatusServer),
|
|
WithType(TypeSpecific),
|
|
WithOriginal(CommonErrorInternal),
|
|
WithError(err),
|
|
)
|
|
|
|
assert.Equal(t, StatusServer, b.Status)
|
|
assert.Equal(t, TypeSpecific, b.Type)
|
|
assert.Equal(t, CommonErrorInternal, b.Original)
|
|
assert.Equal(t, err, b.Error)
|
|
}
|
|
|
|
func TestBuilderErrorCodeLength(t *testing.T) {
|
|
ctx := ContextWithService(context.Background(), "svc")
|
|
ctx = ContextWithEndpoint(ctx, "ep_")
|
|
|
|
b := New(
|
|
WithStatus(StatusSuccess),
|
|
WithType(TypeCommon),
|
|
WithOriginal(CommonSuccess),
|
|
).Throw(ctx)
|
|
|
|
// Status(1) + Endpoint(3) + Service(3) + Type(1) + Original(5) = 13
|
|
assert.Len(t, b.ErrorCode, 13)
|
|
}
|