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  }