render_test.go 2.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  1. // Copyright 2014 Manu Martinez-Almeida. All rights reserved.
  2. // Use of this source code is governed by a MIT style
  3. // license that can be found in the LICENSE file.
  4. package render
  5. import (
  6. "html/template"
  7. "net/http/httptest"
  8. "testing"
  9. "github.com/stretchr/testify/assert"
  10. )
  11. func TestRenderJSON(t *testing.T) {
  12. w := httptest.NewRecorder()
  13. err := JSON.Render(w, 201, map[string]interface{}{
  14. "foo": "bar",
  15. })
  16. assert.NoError(t, err)
  17. assert.Equal(t, w.Code, 201)
  18. assert.Equal(t, w.Body.String(), "{\"foo\":\"bar\"}\n")
  19. assert.Equal(t, w.Header().Get("Content-Type"), "application/json; charset=utf-8")
  20. }
  21. func TestRenderIndentedJSON(t *testing.T) {
  22. w := httptest.NewRecorder()
  23. err := IndentedJSON.Render(w, 202, map[string]interface{}{
  24. "foo": "bar",
  25. "bar": "foo",
  26. })
  27. assert.NoError(t, err)
  28. assert.Equal(t, w.Code, 202)
  29. assert.Equal(t, w.Body.String(), "{\n \"bar\": \"foo\",\n \"foo\": \"bar\"\n}")
  30. assert.Equal(t, w.Header().Get("Content-Type"), "application/json; charset=utf-8")
  31. }
  32. func TestRenderPlain(t *testing.T) {
  33. w := httptest.NewRecorder()
  34. err := Plain.Render(w, 400, "hola %s %d", []interface{}{"manu", 2})
  35. assert.NoError(t, err)
  36. assert.Equal(t, w.Code, 400)
  37. assert.Equal(t, w.Body.String(), "hola manu 2")
  38. assert.Equal(t, w.Header().Get("Content-Type"), "text/plain; charset=utf-8")
  39. }
  40. func TestRenderPlainHTML(t *testing.T) {
  41. w := httptest.NewRecorder()
  42. err := HTMLPlain.Render(w, 401, "hola %s %d", []interface{}{"manu", 2})
  43. assert.NoError(t, err)
  44. assert.Equal(t, w.Code, 401)
  45. assert.Equal(t, w.Body.String(), "hola manu 2")
  46. assert.Equal(t, w.Header().Get("Content-Type"), "text/html; charset=utf-8")
  47. }
  48. func TestRenderHTMLTemplate(t *testing.T) {
  49. w := httptest.NewRecorder()
  50. templ := template.Must(template.New("t").Parse(`Hello {{.name}}`))
  51. htmlRender := HTMLRender{Template: templ}
  52. err := htmlRender.Render(w, 402, "t", map[string]interface{}{
  53. "name": "alexandernyquist",
  54. })
  55. assert.NoError(t, err)
  56. assert.Equal(t, w.Code, 402)
  57. assert.Equal(t, w.Body.String(), "Hello alexandernyquist")
  58. assert.Equal(t, w.Header().Get("Content-Type"), "text/html; charset=utf-8")
  59. }
  60. func TestRenderJoinStrings(t *testing.T) {
  61. assert.Equal(t, joinStrings("a", "BB", "c"), "aBBc")
  62. assert.Equal(t, joinStrings("a", "", "c"), "ac")
  63. assert.Equal(t, joinStrings("text/html", "; charset=utf-8"), "text/html; charset=utf-8")
  64. }