rest_test.go 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134
  1. package rest
  2. import (
  3. "bytes"
  4. "fmt"
  5. "net/http"
  6. "net/http/httptest"
  7. "testing"
  8. "github.com/go-test/deep"
  9. )
  10. type Something struct {
  11. First int
  12. Second string
  13. }
  14. func TestJSONDecoder(t *testing.T) {
  15. const json = `{ "First": 15, "Second": "Some string" }`
  16. buffer := bytes.NewBuffer([]byte(json))
  17. expected := Something{First: 15, Second: "Some string"}
  18. var result Something
  19. err := JSONDecoder(buffer, &result)
  20. if err != nil {
  21. t.Error(err)
  22. }
  23. if diff := deep.Equal(expected, result); diff != nil {
  24. t.Errorf("Wrong result: %v", diff)
  25. }
  26. }
  27. func TestJSONDecoderBadJSON(t *testing.T) {
  28. const badJSON = `{ First: 15, "Second": "Some string" }`
  29. buffer := bytes.NewBuffer([]byte(badJSON))
  30. expected := Something{First: 15, Second: "Some string"}
  31. var result Something
  32. err := JSONDecoder(buffer, &result)
  33. if err == nil {
  34. t.Error("There should be an error")
  35. }
  36. if diff := deep.Equal(expected, result); diff == nil {
  37. t.Errorf("Structures shouldn't match, %v", diff)
  38. }
  39. }
  40. func fakeServer(statusCode int, data string) *httptest.Server {
  41. function := func(w http.ResponseWriter, r *http.Request) {
  42. w.WriteHeader(statusCode)
  43. w.Header().Set("Content-Type", "text")
  44. fmt.Fprint(w, data)
  45. }
  46. return httptest.NewServer(http.HandlerFunc(function))
  47. }
  48. func TestGET(t *testing.T) {
  49. expected := "Some text"
  50. server := fakeServer(http.StatusOK, expected)
  51. defer server.Close()
  52. data, err := GET(server.URL)
  53. if data == nil {
  54. t.Error("Data shouldn't be nil")
  55. }
  56. if err != nil {
  57. t.Error(err)
  58. }
  59. result := data.String()
  60. if expected != result {
  61. t.Errorf("Wrong result, %v", result)
  62. }
  63. }
  64. func TestGET206(t *testing.T) {
  65. expected := "Some text"
  66. server := fakeServer(http.StatusPartialContent, expected)
  67. defer server.Close()
  68. data, err := GET(server.URL)
  69. if data == nil {
  70. t.Error("Data shouldn't be nil")
  71. }
  72. if err != nil {
  73. t.Error(err)
  74. }
  75. result := data.String()
  76. if expected != result {
  77. t.Errorf("Wrong result, %v", result)
  78. }
  79. }
  80. func TestGET404(t *testing.T) {
  81. expected := "Some text"
  82. server := fakeServer(http.StatusNotFound, expected)
  83. defer server.Close()
  84. data, err := GET(server.URL)
  85. if data == nil {
  86. t.Error("Data shouldn't be nil")
  87. }
  88. if err == nil {
  89. t.Error("There should be an error!")
  90. }
  91. result := data.String()
  92. if expected != result {
  93. t.Errorf("Wrong result, %v", result)
  94. }
  95. }
  96. func TestGETNoServer(t *testing.T) {
  97. data, err := GET("/")
  98. if data != nil {
  99. t.Error("Data should be nil!")
  100. }
  101. if err == nil {
  102. t.Error("There should be an error!")
  103. }
  104. }