go.uber.org/yarpc@v1.72.1/encoding/protobuf/v2/error_test.go

// Copyright (c) 2022 Uber Technologies, Inc.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

package v2

import (
	"errors"
	"fmt"
	"strings"
	"testing"

	"github.com/golang/protobuf/ptypes/any"
	"github.com/golang/protobuf/ptypes/wrappers"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"go.uber.org/yarpc/api/transport/transporttest"
	"go.uber.org/yarpc/yarpcerrors"
	rpc_status "google.golang.org/genproto/googleapis/rpc/status"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
	"google.golang.org/protobuf/proto"
	anypb "google.golang.org/protobuf/types/known/anypb"
)

func TestNewOK(t *testing.T) {
	err := NewError(yarpcerrors.CodeOK, "okay")
	assert.Nil(t, err)

	assert.Equal(t, yarpcerrors.FromError(err).Code(), yarpcerrors.CodeOK)
	assert.Equal(t, yarpcerrors.FromError(err).Message(), "")
}

func TestNew(t *testing.T) {
	err := NewError(yarpcerrors.CodeNotFound, "unfounded accusation")
	assert.Equal(t, yarpcerrors.FromError(err).Code(), yarpcerrors.CodeNotFound)
	assert.Equal(t, yarpcerrors.FromError(err).Message(), "unfounded accusation")
	assert.Contains(t, err.Error(), "unfounded accusation")
}

func TestForeignError(t *testing.T) {
	err := errors.New("to err is go")
	assert.Equal(t, yarpcerrors.FromError(err).Code(), yarpcerrors.CodeUnknown)
	assert.Equal(t, yarpcerrors.FromError(err).Message(), "to err is go")
}

func TestConvertToYARPCErrorWithWrappedError(t *testing.T) {
	errDetail := &wrappers.BytesValue{Value: []byte("err detail bytes")}

	pbErr := NewError(
		yarpcerrors.CodeAborted,
		"aborted",
		WithErrorDetails(errDetail))

	wrappedErr := fmt.Errorf("wrapped err 2: %w", fmt.Errorf("wrapped err 1: %w", pbErr))

	err := convertToYARPCError(Encoding, wrappedErr, &codec{}, nil /* resw */)
	require.True(t, yarpcerrors.IsStatus(err), "unexpected error")
	assert.Equal(t, yarpcerrors.FromError(err).Code(), yarpcerrors.CodeAborted, "unexpected err code")
	assert.Equal(t, yarpcerrors.FromError(err).Message(), "aborted", "unexpected error message")

	gotDetails := yarpcerrors.FromError(err).Details()
	assert.NotEmpty(t, gotDetails, "no details marshaled")
}

func TestConvertToYARPCErrorApplicationErrorMeta(t *testing.T) {
	errDetails := []proto.Message{
		&wrappers.StringValue{Value: "detail message"},
		&wrappers.Int32Value{Value: 42},
		&wrappers.BytesValue{Value: []byte("detail bytes")},
	}

	pbErr := NewError(
		yarpcerrors.CodeAborted,
		"aborted",
		WithErrorDetails(errDetails...))

	resw := &transporttest.FakeResponseWriter{}
	err := convertToYARPCError(Encoding, pbErr, &codec{}, resw)
	require.Error(t, err)

	require.NotNil(t, resw.ApplicationErrorMeta)
	assert.Equal(t, "StringValue", resw.ApplicationErrorMeta.Name, "expected first error detail name")
	assert.Equal(t,
		`[]{ StringValue{value:"detail message"} , Int32Value{value:42} , BytesValue{value:"detail bytes"} }`,
		resw.ApplicationErrorMeta.Details,
		"unexpected string of error details")
	assert.Nil(t, resw.ApplicationErrorMeta.Code, "code should nil")
}

func TestMessageNameWithoutPackage(t *testing.T) {
	tests := []struct {
		name string
		give string
		want string
	}{
		{
			name: "fqn",
			give: "uber.foo.bar.baz.MessageName",
			want: "MessageName",
		},
		{
			name: "not fully qualified",
			give: "MyMessage",
			want: "MyMessage",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			assert.Equal(t, tt.want, messageNameWithoutPackage(tt.give), "unexpected trim")
		})
	}
}

type yarpcError interface{ YARPCError() *yarpcerrors.Status }

func TestPbErrorToYARPCError(t *testing.T) {
	tests := []struct {
		name             string
		code             yarpcerrors.Code
		message          string
		details          []proto.Message
		expectedGRPCCode codes.Code
	}{
		{
			name:             "pbError without details",
			code:             yarpcerrors.CodeAborted,
			message:          "simple test",
			expectedGRPCCode: codes.Aborted,
		},
		{
			name:             "pbError with single detail",
			code:             yarpcerrors.CodeInternal,
			message:          "internal error",
			expectedGRPCCode: codes.Internal,
			details: []proto.Message{
				&wrappers.StringValue{Value: "test value"},
			},
		},
		{
			name:             "pbError with multiple details",
			code:             yarpcerrors.CodeNotFound,
			message:          "not found error",
			expectedGRPCCode: codes.NotFound,
			details: []proto.Message{
				&wrappers.StringValue{Value: "test value"},
				&wrappers.Int32Value{Value: 45},
				&any.Any{Value: []byte{1, 2, 3, 4, 5}},
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			var errOpts []ErrorOption
			if len(tt.details) > 0 {
				errOpts = append(errOpts, WithErrorDetails(tt.details...))
			}
			pberror := NewError(tt.code, tt.message, errOpts...)
			st := pberror.(yarpcError).YARPCError()
			assert.Equal(t, st.Code(), tt.code)
			assert.Equal(t, st.Message(), tt.message)

			statusPb := rpc_status.Status{}
			err := proto.Unmarshal(st.Details(), &statusPb)
			assert.NoError(t, err, "unexpected unmarshal error")

			status := status.FromProto(&statusPb)
			assert.Equal(t, tt.expectedGRPCCode, status.Code(), "unexpected grpc status code")
			assert.Equal(t, tt.message, status.Message(), "unexpected grpc status message")
			assert.Len(t, status.Details(), len(tt.details), "unexpected details length")
			for i, detail := range tt.details {
				assert.True(t, proto.Equal(detail, status.Details()[i].(proto.Message)))
			}
		})
	}
}

func TestConvertToYARPCErrorWithIncorrectEncoding(t *testing.T) {
	pberr := &pberror{code: yarpcerrors.CodeAborted, message: "test"}
	err := convertToYARPCError("thrift", pberr, &codec{}, nil)
	assert.Error(t, err, "unexpected empty error")
	assert.Equal(t, err.Error(),
		"code:internal message:encoding.Expect should have handled encoding \"thrift\" but did not")
}

func TestConvertFromYARPCError(t *testing.T) {
	t.Run("incorrect encoding", func(t *testing.T) {
		yerr := yarpcerrors.Newf(yarpcerrors.CodeAborted, "test").WithDetails([]byte{1, 2})
		err := convertFromYARPCError("thrift", yerr, &codec{})
		assert.Equal(t, err.Error(),
			`code:internal message:encoding.Expect should have handled encoding "thrift" but did not`)
	})
	t.Run("empty details", func(t *testing.T) {
		yerr := yarpcerrors.Newf(yarpcerrors.CodeAborted, "test")
		err := convertFromYARPCError(Encoding, yerr, &codec{})
		assert.Equal(t, err.Error(), "code:aborted message:test")
	})
}

func TestCreateStatusWithDetailErrors(t *testing.T) {
	t.Run("code ok", func(t *testing.T) {
		pberr := &pberror{code: yarpcerrors.CodeOK, message: "test"}
		_, err := createStatusWithDetail(pberr, Encoding, &codec{})
		assert.Error(t, err, "unexpected empty error")
		assert.Equal(t, err.Error(), "no status error for error with code OK")
	})

	t.Run("unsupported code", func(t *testing.T) {
		pberr := &pberror{code: 99, message: "test"}
		_, err := createStatusWithDetail(pberr, Encoding, &codec{})
		assert.Error(t, err, "unexpected empty error")
		assert.Equal(t, err.Error(), "no status error for error with code 99")
	})

	t.Run("unsupported encoding", func(t *testing.T) {
		pberr := &pberror{code: yarpcerrors.CodeAborted}
		_, err := createStatusWithDetail(pberr, "thrift", &codec{})
		assert.Error(t, err, "unexpected empty error")
		assert.Equal(t, err.Error(),
			"code:internal message:encoding.Expect should have handled encoding \"thrift\" but did not")
	})
}

func TestErrorHandling(t *testing.T) {
	t.Run("GetErrorDetail empty error handling", func(t *testing.T) {
		assert.Nil(t, GetErrorDetails(nil), "unexpected details")
	})
	t.Run("GetErrorDetail non pberror", func(t *testing.T) {
		assert.Nil(t, GetErrorDetails(errors.New("test")), "unexpected details")
	})
	t.Run("GetErrorDetail with no error detail", func(t *testing.T) {
		err := NewError(
			yarpcerrors.CodeAborted,
			"aborted")
		assert.Nil(t, GetErrorDetails(err), "unexpected details")
	})
	t.Run("PbError empty error handling", func(t *testing.T) {
		var pbErr *pberror
		assert.Nil(t, pbErr.YARPCError(), "unexpected yarpcerror")
	})
	t.Run("PbError with nil detail message", func(t *testing.T) {
		pbErr := &pberror{
			details: []*any.Any{nil},
		}
		require.Len(t, GetErrorDetails(pbErr), 1)
		assert.Equal(t, strings.ReplaceAll(GetErrorDetails(pbErr)[0].(error).Error(), "\u00a0", " "),
			fmt.Errorf("proto: invalid empty type URL").Error())
	})
	t.Run("NewError with nil error detail", func(t *testing.T) {
		err := NewError(
			yarpcerrors.CodeAborted,
			"aborted",
			WithErrorDetails([]proto.Message{nil}...))
		assert.Equal(t, strings.ReplaceAll(err.Error(), "\u00a0", " "),
			fmt.Errorf("proto: invalid nil source message").Error())
	})
}

func TestSetApplicationErrorMeta(t *testing.T) {
	respErr := NewError(
		yarpcerrors.CodeAborted,
		"aborted",
	)

	anyString, err := anypb.New(&wrappers.StringValue{Value: "baz"})
	require.NoError(t, err)

	pbErr := respErr.(*pberror)
	pbErr.details = append(pbErr.details, &any.Any{TypeUrl: "foo", Value: []byte("bar")})
	pbErr.details = append(pbErr.details, anyString)

	resw := &transporttest.FakeResponseWriter{}
	setApplicationErrorMeta(pbErr, resw)

	assert.Equal(t, "StringValue", resw.ApplicationErrorMeta.Name)
	assert.Equal(t, `[]{ StringValue{value:"baz"} }`, resw.ApplicationErrorMeta.Details)
}