Function TestCustomHTTPErrorHandler

pkg/middleware/http/handler_test.go:20–47

(t *testing.T)

)

func TestCustomHTTPErrorHandler(t *testing.T) {
	req := httptest.NewRequest(http.MethodGet, "/upper?word=abc", nil)
	w := httptest.NewRecorder()
	e := errors.NewEncodedError(int32(openfgav1.ErrorCode_assertions_too_many_items), "some error")
	metaData := runtime.ServerMetadata{
		HeaderMD: metadata.New(map[string]string{
			"foo": "boo",
		}),
		TrailerMD: metadata.New(map[string]string{}),
	}
	ctx := runtime.NewServerMetadataContext(context.Background(), metaData)
	CustomHTTPErrorHandler(ctx, w, req, e)
	res := w.Result()
	defer res.Body.Close()
	require.Equal(t, http.StatusBadRequest, res.StatusCode)

	header := res.Header.Get("Foo")
	require.Equal(t, "boo", header)

	contentType := res.Header.Get("Content-Type")
	require.Equal(t, "application/json", contentType)

	data, err := io.ReadAll(res.Body)
	require.NoError(t, err)

	expectedData := "{\"code\":\"assertions_too_many_items\",\"message\":\"some error\"}"
	require.Equal(t, expectedData, strings.TrimSpace(string(data)))
}

func TestCustomHTTPErrorHandlerSpecialEncoding(t *testing.T) {
	req := httptest.NewRequest(http.MethodGet, "/upper?word=abc", nil)