feat(pubsub/pstest)!: Add status code to error injection.

pubsub/pstest/fake.go

@@ -1171,20 +1171,22 @@ func (s *GServer) runReactor(req interface{}, funcName string, defaultObj interf
 	return false, nil, nil
 }
 
-// ErrorInjectionReactor is a reactor to inject an error message
-type ErrorInjectionReactor struct {
-	errMsg string
+// errorInjectionReactor is a reactor to inject an error message with status code.
+type errorInjectionReactor struct {
+	code codes.Code
+	msg  string
 }
 
-// React simply returns an error with defined error message.
-func (e *ErrorInjectionReactor) React(_ interface{}) (handled bool, ret interface{}, err error) {
-	return true, nil, fmt.Errorf(e.errMsg)
+// React simply returns an error with defined error message and status code.
+func (e *errorInjectionReactor) React(_ interface{}) (handled bool, ret interface{}, err error) {
+	return true, nil, status.Errorf(e.code, e.msg)
 }
 
-// WithErrorInjection creates a ServerReactorOption that injects error for a certain function.
-func WithErrorInjection(funcName string, errMsg string) ServerReactorOption {
+// WithErrorInjection creates a ServerReactorOption that injects error with defined status code and
+// message for a certain function.
+func WithErrorInjection(funcName string, code codes.Code, msg string) ServerReactorOption {
 	return ServerReactorOption{
 		FuncName: funcName,
-		Reactor:  &ErrorInjectionReactor{errMsg: errMsg},
+		Reactor:  &errorInjectionReactor{code: code, msg: msg},
 	}
 }

pubsub/pstest/fake_test.go

@@ -951,15 +951,19 @@ func TestErrorInjection(t *testing.T) {
 	testcases := []struct {
 		funcName string
 		param    interface{}
+		code     codes.Code
 	}{
 		{
 			funcName: "CreateTopic",
+			code:     codes.Internal,
 		},
 		{
 			funcName: "GetTopic",
+			code:     codes.Aborted,
 		},
 		{
 			funcName: "UpdateTopic",
+			code:     codes.DeadlineExceeded,
 		},
 		{
 			funcName: "ListTopics",
@@ -1010,8 +1014,13 @@ func TestErrorInjection(t *testing.T) {
 	for _, tc := range testcases {
 		ctx := context.TODO()
 		errMsg := "error-injection-" + tc.funcName
+		// set error code to unknown unless specified
+		ec := codes.Unknown
+		if tc.code != codes.OK {
+			ec = tc.code
+		}
 		opts := []ServerReactorOption{
-			WithErrorInjection(tc.funcName, errMsg),
+			WithErrorInjection(tc.funcName, ec, errMsg),
 		}
 		_, _, server, cleanup := newFake(ctx, t, opts...)
 		defer cleanup()
@@ -1033,7 +1042,7 @@ func TestErrorInjection(t *testing.T) {
 		ret := reflect.ValueOf(&server.GServer).MethodByName(tc.funcName).Call([]reflect.Value{reflect.ValueOf(ctx), req})
 
 		got := ret[1].Interface().(error)
-		if got == nil || !strings.Contains(got.Error(), errMsg) {
+		if got == nil || status.Code(got) != ec || !strings.Contains(got.Error(), errMsg) {
 			t.Errorf("Got error does not contain the right key %v", got)
 		}
 	}