chore: refactor errors to use errors.Is (#3654)

2023-10-21 00:00:57 +08:00
parent 81ae7d36b5
commit 42e0a6f90c
21 changed files with 90 additions and 53 deletions
--- a/zrpc/internal/serverinterceptors/sheddinginterceptor_test.go
+++ b/zrpc/internal/serverinterceptors/sheddinginterceptor_test.go
@@ -8,6 +8,8 @@ import (
 	"github.com/zeromicro/go-zero/core/load"
 	"github.com/zeromicro/go-zero/core/stat"
 	"google.golang.org/grpc"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
 )

 func TestUnarySheddingInterceptor(t *testing.T) {
@@ -33,7 +35,7 @@ func TestUnarySheddingInterceptor(t *testing.T) {
 			name:      "reject",
 			allow:     false,
 			handleErr: nil,
-			expect:    load.ErrServiceOverloaded,
+			expect:    status.Error(codes.ResourceExhausted, load.ErrServiceOverloaded.Error()),
 		},
 	}