mirror of
https://github.com/rocky-linux/peridot.git
synced 2024-12-30 14:00:57 +00:00
18 lines
441 B
Go
18 lines
441 B
Go
package grpcutil
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
// RequireStatus asserts that an error is a gRPC error and returns the expected
|
|
// status code.
|
|
func RequireStatus(t *testing.T, expected codes.Code, err error) {
|
|
require.Error(t, err)
|
|
errStatus, ok := status.FromError(err)
|
|
require.True(t, ok)
|
|
require.Equal(t, expected, errStatus.Code())
|
|
}
|