threshold/internal/servers/controller/handlers/errors_test.go

package handlers

import (
	"context"
	stderrors "errors"
	"fmt"
	"io/ioutil"
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/google/go-cmp/cmp"
	"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
	"github.com/hashicorp/boundary/internal/errors"
	pb "github.com/hashicorp/boundary/internal/gen/controller/api"
	"github.com/hashicorp/go-hclog"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
	"google.golang.org/protobuf/testing/protocmp"
)

func TestApiErrorHandler(t *testing.T) {
	ctx := context.Background()
	req, err := http.NewRequest("GET", "madeup/for/the/test", nil)
	require.NoError(t, err)
	mux := runtime.NewServeMux()
	inMarsh, outMarsh := runtime.MarshalerForRequest(mux, req)

	tested := ErrorHandler(hclog.L())

	testCases := []struct {
		name     string
		err      error
		expected apiError
	}{
		{
			name: "Not Found",
			err:  NotFoundErrorf("Test"),
			expected: apiError{
				status: http.StatusNotFound,
				inner: &pb.Error{
					Kind:    "NotFound",
					Message: "Test",
				},
			},
		},
		{
			name: "Invalid Fields",
			err: InvalidArgumentErrorf("Test", map[string]string{
				"k1": "v1",
				"k2": "v2",
			}),
			expected: apiError{
				status: http.StatusBadRequest,
				inner: &pb.Error{
					Kind:    "InvalidArgument",
					Message: "Test",
					Details: &pb.ErrorDetails{
						RequestFields: []*pb.FieldError{
							{
								Name:        "k1",
								Description: "v1",
							},
							{
								Name:        "k2",
								Description: "v2",
							},
						},
					},
				},
			},
		},
		{
			name: "GrpcGateway Routing Error",
			err:  runtime.ErrNotMatch,
			expected: apiError{
				status: http.StatusNotFound,
				inner: &pb.Error{
					Kind:    "NotFound",
					Message: http.StatusText(http.StatusNotFound),
				},
			},
		},
		{
			name: "Unimplemented error",
			err:  status.Error(codes.Unimplemented, "Test"),
			expected: apiError{
				status: http.StatusMethodNotAllowed,
				inner: &pb.Error{
					Kind:    "Unimplemented",
					Message: "Test",
				},
			},
		},
		{
			name: "Unknown error",
			err:  stderrors.New("Some random error"),
			expected: apiError{
				status: http.StatusInternalServerError,
				inner: &pb.Error{
					Kind: "Internal",
					Message: "Some random error",
				},
			},
		},
		{
			name: "Db invalid public id error",
			err:  fmt.Errorf("test error: %w", errors.ErrInvalidPublicId),
			expected: apiError{
				status: http.StatusInternalServerError,
				inner: &pb.Error{
					Kind: "Internal",
					Message: fmt.Sprintf("test error: %s", errors.ErrInvalidPublicId),
				},
			},
		},
		{
			name: "Db invalid parameter",
			err:  fmt.Errorf("test error: %w", errors.ErrInvalidParameter),
			expected: apiError{
				status: http.StatusInternalServerError,
				inner: &pb.Error{
					Kind: "Internal",
					Message: fmt.Sprintf("test error: %s", errors.ErrInvalidParameter),
				},
			},
		},
		{
			name: "Db invalid field mask",
			err:  fmt.Errorf("test error: %w", errors.ErrInvalidFieldMask),
			expected: apiError{
				status: http.StatusBadRequest,
				inner: &pb.Error{
					Kind:    "InvalidArgument",
					Message: "Error in provided request",
					Details: &pb.ErrorDetails{RequestFields: []*pb.FieldError{{Name: "update_mask", Description: "Invalid update mask provided."}}},
				},
			},
		},
		{
			name: "Db empty field mask",
			err:  fmt.Errorf("test error: %w", errors.ErrEmptyFieldMask),
			expected: apiError{
				status: http.StatusBadRequest,
				inner: &pb.Error{
					Kind:    "InvalidArgument",
					Message: "Error in provided request",
					Details: &pb.ErrorDetails{RequestFields: []*pb.FieldError{{Name: "update_mask", Description: "Invalid update mask provided."}}},
				},
			},
		},
		{
			name: "Db not unique",
			err:  fmt.Errorf("test error: %w", errors.ErrNotUnique),
			expected: apiError{
				status: http.StatusBadRequest,
				inner: &pb.Error{
					Kind:    "InvalidArgument",
					Message: genericUniquenessMsg,
				},
			},
		},
		{
			name: "Db record not found",
			err:  fmt.Errorf("test error: %w", errors.ErrRecordNotFound),
			expected: apiError{
				status: http.StatusNotFound,
				inner: &pb.Error{
					Kind:    "NotFound",
					Message: genericNotFoundMsg,
				},
			},
		},
		{
			name: "Db multiple records",
			err:  fmt.Errorf("test error: %w", errors.ErrMultipleRecords),
			expected: apiError{
				status: http.StatusInternalServerError,
				inner: &pb.Error{
					Kind: "Internal",
					Message: fmt.Sprintf("test error: %s", errors.ErrMultipleRecords),
				},
			},
		},
	}
	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			assert, require := assert.New(t), require.New(t)
			w := httptest.NewRecorder()
			tested(ctx, mux, outMarsh, w, req, tc.err)
			resp := w.Result()
			assert.EqualValues(tc.expected.status, resp.StatusCode)

			got, err := ioutil.ReadAll(resp.Body)
			require.NoError(err)

			gotErr := &pb.Error{}
			err = inMarsh.Unmarshal(got, gotErr)
			require.NoError(err)

			assert.Equal(tc.expected.status, int32(resp.StatusCode))
			assert.Empty(cmp.Diff(tc.expected.inner, gotErr, protocmp.Transform()))
		})
	}
}