Function TestServerHandleReturn4XXErrors

httpgrpc/server/server_test.go:190–256

(t *testing.T)

}

func TestServerHandleReturn4XXErrors(t *testing.T) {
	testCases := map[string]struct {
		errorCode       int
		return4xxErrors bool
		expectedError   bool
	}{
		"HTTPResponse with code 5xx should return an error when server creates with Return4XXErrorsOption": {
			errorCode:       http.StatusInternalServerError,
			return4xxErrors: true,
			expectedError:   true,
		},
		"HTTPResponse with code 5xx should return an error when server creates without Return4XXErrorsOption": {
			errorCode:       http.StatusInternalServerError,
			return4xxErrors: false,
			expectedError:   true,
		},
		"HTTPResponse with code 4xx should return an error when server creates with Return4XXErrorsOption": {
			errorCode:       http.StatusBadRequest,
			return4xxErrors: true,
			expectedError:   true,
		},
		"HTTPResponse with code 4xx should not return an error when server creates without Return4XXErrorsOption": {
			errorCode:       http.StatusBadRequest,
			return4xxErrors: false,
			expectedError:   false,
		},
		"HTTPResponse with code different from 5xx and 4xx should not return an error when server creates with Return4XXErrorsOption": {
			errorCode:       http.StatusNoContent,
			return4xxErrors: true,
			expectedError:   false,
		},
		"HTTPResponse with code different from 5xx and 4xx should not return an error when server creates without Return4XXErrorsOption": {
			errorCode:       http.StatusNoContent,
			return4xxErrors: false,
			expectedError:   false,
		},
	}
	errMsg := "this is an error"
	for testName, testData := range testCases {
		t.Run(testName, func(t *testing.T) {
			h := http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
				http.Error(w, errMsg, testData.errorCode)
			})

			var serverOptions []Option
			if testData.return4xxErrors {
				serverOptions = []Option{WithReturn4XXErrors}
			}
			s := NewServer(h, serverOptions...)

			req := &httpgrpc.HTTPRequest{
				Method: "GET",
				Url:    "/test",
			}
			resp, err := s.Handle(context.Background(), req)
			if testData.expectedError {
				require.Error(t, err)
				require.Nil(t, resp)