48 lines
889 B
Go
48 lines
889 B
Go
|
// Copyright (C) 2019 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package rpcstatus
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"google.golang.org/grpc/status"
|
||
|
|
|
||
|
|
"storj.io/drpc/drpcerr"
|
||
|
|
)
|
||
|
|
|
||
|
|
var allCodes = []StatusCode{
|
||
|
|
Unknown,
|
||
|
|
OK,
|
||
|
|
Canceled,
|
||
|
|
InvalidArgument,
|
||
|
|
DeadlineExceeded,
|
||
|
|
NotFound,
|
||
|
|
AlreadyExists,
|
||
|
|
PermissionDenied,
|
||
|
|
ResourceExhausted,
|
||
|
|
FailedPrecondition,
|
||
|
|
Aborted,
|
||
|
|
OutOfRange,
|
||
|
|
Unimplemented,
|
||
|
|
Internal,
|
||
|
|
Unavailable,
|
||
|
|
DataLoss,
|
||
|
|
Unauthenticated,
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestStatus(t *testing.T) {
|
||
|
|
for _, code := range allCodes {
|
||
|
|
err := Error(code, "")
|
||
|
|
assert.Equal(t, Code(err), code)
|
||
|
|
assert.Equal(t, status.Code(err), code.toGRPC())
|
||
|
|
assert.Equal(t, drpcerr.Code(err), uint64(code))
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.Equal(t, Code(nil), OK)
|
||
|
|
assert.Equal(t, Code(context.Canceled), Canceled)
|
||
|
|
assert.Equal(t, Code(context.DeadlineExceeded), DeadlineExceeded)
|
||
|
|
}
|