github.com/segakazzz/buffalo@v0.16.22-0.20210119082501-1f52048d3feb/route_info_test.go (about) 1 package buffalo 2 3 import ( 4 "database/sql" 5 "net/http" 6 "testing" 7 8 "github.com/gobuffalo/buffalo/render" 9 "github.com/gobuffalo/httptest" 10 "github.com/stretchr/testify/require" 11 ) 12 13 func Test_RouteInfo_ServeHTTP_SQL_Error(t *testing.T) { 14 r := require.New(t) 15 16 app := New(Options{}) 17 app.GET("/good", func(c Context) error { 18 return c.Render(http.StatusOK, render.String("hi")) 19 }) 20 21 app.GET("/bad", func(c Context) error { 22 return sql.ErrNoRows 23 }) 24 25 w := httptest.New(app) 26 27 res := w.HTML("/good").Get() 28 r.Equal(http.StatusOK, res.Code) 29 30 res = w.HTML("/bad").Get() 31 r.Equal(http.StatusNotFound, res.Code) 32 }