github.com/haraldLmueller/buffalo@v0.11.1/render/auto_test.go

package render_test

import (
	"os"
	"path/filepath"
	"strings"
	"testing"

	"github.com/gobuffalo/buffalo"
	"github.com/gobuffalo/buffalo/render"
	"github.com/gobuffalo/packr"
	"github.com/markbates/willie"
	"github.com/stretchr/testify/require"
)

type Car struct {
	ID   int
	Name string
}

type Cars []Car

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

	app := buffalo.New(buffalo.Options{})
	app.GET("/cars", func(c buffalo.Context) error {
		return c.Render(200, render.Auto(c, []string{"Honda", "Toyota", "Ford", "Chevy"}))
	})

	w := willie.New(app)

	res := w.JSON("/cars").Get()
	r.Equal(`["Honda","Toyota","Ford","Chevy"]`, strings.TrimSpace(res.Body.String()))
}

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

	app := buffalo.New(buffalo.Options{})
	app.GET("/cars", func(c buffalo.Context) error {
		return c.Render(200, render.Auto(c, []string{"Honda", "Toyota", "Ford", "Chevy"}))
	})

	w := willie.New(app)

	res := w.XML("/cars").Get()
	r.Equal("<string>Honda</string>\n<string>Toyota</string>\n<string>Ford</string>\n<string>Chevy</string>", strings.TrimSpace(res.Body.String()))
}

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

	err := withHTMLFile("cars/index.html", "INDEX: <%= len(cars) %>", func(e *render.Engine) {
		app := buffalo.New(buffalo.Options{})
		app.GET("/cars", func(c buffalo.Context) error {
			return c.Render(200, e.Auto(c, Cars{
				{Name: "Ford"},
				{Name: "Chevy"},
			}))
		})

		w := willie.New(app)
		res := w.HTML("/cars").Get()

		r.Contains(res.Body.String(), "INDEX: 2")
	})
	r.NoError(err)
}

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

	err := withHTMLFile("cars/show.html", "Show: <%= car.Name %>", func(e *render.Engine) {
		app := buffalo.New(buffalo.Options{})
		app.GET("/cars/{id}", func(c buffalo.Context) error {
			return c.Render(200, e.Auto(c, Car{Name: "Honda"}))
		})

		w := willie.New(app)
		res := w.HTML("/cars/1").Get()
		r.Contains(res.Body.String(), "Show: Honda")
	})
	r.NoError(err)
}

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

	err := withHTMLFile("cars/new.html", "New: <%= car.Name %>", func(e *render.Engine) {
		app := buffalo.New(buffalo.Options{})
		app.GET("/cars/new", func(c buffalo.Context) error {
			return c.Render(200, e.Auto(c, Car{Name: "Honda"}))
		})

		w := willie.New(app)
		res := w.HTML("/cars/new").Get()
		r.Contains(res.Body.String(), "New: Honda")
	})
	r.NoError(err)
}

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

	err := withHTMLFile("cars/new.html", "New: <%= car.Name %>", func(e *render.Engine) {
		app := buffalo.New(buffalo.Options{})
		app.POST("/cars", func(c buffalo.Context) error {
			return c.Render(201, e.Auto(c, Car{Name: "Honda"}))
		})

		w := willie.New(app)
		res := w.HTML("/cars").Post(nil)
		r.Contains(res.Body.String(), "New: Honda")
	})
	r.NoError(err)
}

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

	app := buffalo.New(buffalo.Options{})
	app.POST("/cars", func(c buffalo.Context) error {
		return c.Render(201, render.Auto(c, Car{
			ID:   1,
			Name: "Honda",
		}))
	})

	w := willie.New(app)
	res := w.HTML("/cars").Post(nil)
	r.Equal("/cars/1", res.Location())
	r.Equal(302, res.Code)
}

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

	err := withHTMLFile("cars/new.html", "Create: <%= car.Name %>", func(e *render.Engine) {
		app := buffalo.New(buffalo.Options{})
		app.POST("/cars", func(c buffalo.Context) error {
			b := Car{
				Name: "Honda",
			}
			return c.Render(422, e.Auto(c, b))
		})

		w := willie.New(app)
		res := w.HTML("/cars").Post(nil)
		r.Equal(422, res.Code)
		r.Contains(res.Body.String(), "Create: Honda")
	})
	r.NoError(err)
}

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

	err := withHTMLFile("cars/edit.html", "Edit: <%= car.Name %>", func(e *render.Engine) {
		app := buffalo.New(buffalo.Options{})
		app.GET("/cars/{id}/edit", func(c buffalo.Context) error {
			return c.Render(200, e.Auto(c, Car{Name: "Honda"}))
		})

		w := willie.New(app)
		res := w.HTML("/cars/1/edit").Get()
		r.Contains(res.Body.String(), "Edit: Honda")
	})
	r.NoError(err)
}

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

	err := withHTMLFile("cars/edit.html", "Update: <%= car.Name %>", func(e *render.Engine) {
		app := buffalo.New(buffalo.Options{})
		app.PUT("/cars/{id}", func(c buffalo.Context) error {
			return c.Render(200, e.Auto(c, Car{Name: "Honda"}))
		})

		w := willie.New(app)
		res := w.HTML("/cars/1").Put(nil)

		r.Contains(res.Body.String(), "Update: Honda")
	})
	r.NoError(err)
}

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

	app := buffalo.New(buffalo.Options{})
	app.PUT("/cars/{id}", func(c buffalo.Context) error {
		b := Car{
			ID:   1,
			Name: "Honda",
		}
		return c.Render(200, render.Auto(c, b))
	})

	w := willie.New(app)
	res := w.HTML("/cars/1").Put(nil)
	r.Equal("/cars/1", res.Location())
	r.Equal(302, res.Code)
}

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

	err := withHTMLFile("cars/edit.html", "Update: <%= car.Name %>", func(e *render.Engine) {
		app := buffalo.New(buffalo.Options{})
		app.PUT("/cars/{id}", func(c buffalo.Context) error {
			b := Car{
				ID:   1,
				Name: "Honda",
			}
			return c.Render(422, e.Auto(c, b))
		})

		w := willie.New(app)
		res := w.HTML("/cars/1").Put(nil)
		r.Equal(422, res.Code)
		r.Contains(res.Body.String(), "Update: Honda")
	})
	r.NoError(err)
}

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

	app := buffalo.New(buffalo.Options{})
	app.DELETE("/cars/{id}", func(c buffalo.Context) error {
		b := Car{
			ID:   1,
			Name: "Honda",
		}
		return c.Render(200, render.Auto(c, b))
	})

	w := willie.New(app)
	res := w.HTML("/cars/1").Delete()
	r.Equal("/cars", res.Location())
	r.Equal(302, res.Code)
}

func withHTMLFile(name string, contents string, fn func(*render.Engine)) error {
	tmpDir := filepath.Join(os.TempDir(), filepath.Dir(name))
	err := os.MkdirAll(tmpDir, 0766)
	if err != nil {
		return err
	}
	defer os.Remove(tmpDir)

	tmpFile, err := os.Create(filepath.Join(tmpDir, filepath.Base(name)))
	if err != nil {
		return err
	}
	defer os.Remove(tmpFile.Name())

	_, err = tmpFile.Write([]byte(contents))
	if err != nil {
		return err
	}

	e := render.New(render.Options{
		TemplatesBox: packr.NewBox(os.TempDir()),
	})

	fn(e)
	return nil
}