route_info_test.go

package buffalo

import (
	"database/sql"
	"fmt"
	"net/http"
	"testing"

	"github.com/gobuffalo/buffalo/render"
	"github.com/gobuffalo/httptest"
	"github.com/stretchr/testify/require"
)

func Test_RouteInfo_ServeHTTP_SQL_Error(t *testing.T) {
	r := require.New(t)

	app := New(Options{})
	app.GET("/good", func(c Context) error {
		return c.Render(http.StatusOK, render.String("hi"))
	})

	app.GET("/bad", func(c Context) error {
		return sql.ErrNoRows
	})

	app.GET("/bad-2", func(c Context) error {
		return sql.ErrTxDone
	})

	app.GET("/gone-unwrap", func(c Context) error {
		return c.Error(http.StatusGone, sql.ErrTxDone)
	})

	app.GET("/gone-wrap", func(c Context) error {
		return c.Error(http.StatusGone, fmt.Errorf("some error wrapping here: %w", sql.ErrNoRows))
	})

	w := httptest.New(app)

	res := w.HTML("/good").Get()
	r.Equal(http.StatusOK, res.Code)

	res = w.HTML("/bad").Get()
	r.Equal(http.StatusNotFound, res.Code)

	res = w.HTML("/gone-wrap").Get()
	r.Equal(http.StatusGone, res.Code)

	res = w.HTML("/gone-unwrap").Get()
	r.Equal(http.StatusGone, res.Code)
}