You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
1.9 KiB
Go
80 lines
1.9 KiB
Go
package bugbox
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
func newGETEnclosureRequest(name string) *http.Request {
|
|
request, _ := http.NewRequest(http.MethodGet, fmt.Sprintf("/enclosure/%s", name), nil)
|
|
return request
|
|
}
|
|
|
|
func assertResponseBody(t testing.TB, got, want string) {
|
|
t.Helper()
|
|
if got != want {
|
|
// a %q is a double quoted string
|
|
t.Errorf("got %q, want %q", got, want)
|
|
}
|
|
}
|
|
|
|
func assertStatus(t testing.TB, got, want int) {
|
|
t.Helper()
|
|
if got != want {
|
|
t.Errorf("Status error, got %d want %d", got, want)
|
|
}
|
|
}
|
|
|
|
type MockEnclosureStore struct {
|
|
enclosures map[string]string
|
|
}
|
|
|
|
func (s *MockEnclosureStore) GetEnclosure(name string) string {
|
|
e := s.enclosures[name]
|
|
return e
|
|
}
|
|
|
|
func TestGETEnclosures(t *testing.T) {
|
|
mockEnclosureStore := MockEnclosureStore{
|
|
map[string]string{
|
|
"1337": "1337",
|
|
"7331": "7331"},
|
|
}
|
|
server := &BugBoxServer{EnclosureStore: &mockEnclosureStore}
|
|
|
|
t.Run("returns enclosure 1337", func(t *testing.T) {
|
|
// nil is used since we are not providing a request body
|
|
request := newGETEnclosureRequest("1337")
|
|
// response is a ResponseRecorder used for spying on response
|
|
response := httptest.NewRecorder()
|
|
|
|
server.ServeHTTP(response, request)
|
|
|
|
assertResponseBody(t, response.Body.String(), "1337")
|
|
assertStatus(t, response.Code, http.StatusOK)
|
|
|
|
})
|
|
t.Run("returns enclosure 7331", func(t *testing.T) {
|
|
// nil is used since we are not providing a request body
|
|
request := newGETEnclosureRequest("7331")
|
|
// response is a ResponseRecorder used for spying on response
|
|
response := httptest.NewRecorder()
|
|
|
|
server.ServeHTTP(response, request)
|
|
|
|
assertResponseBody(t, response.Body.String(), "7331")
|
|
assertStatus(t, response.Code, http.StatusOK)
|
|
})
|
|
|
|
t.Run("returns 404 when enclosure missing", func(t *testing.T) {
|
|
request := newGETEnclosureRequest("9000")
|
|
response := httptest.NewRecorder()
|
|
|
|
server.ServeHTTP(response, request)
|
|
|
|
assertStatus(t, response.Code, http.StatusNotFound)
|
|
})
|
|
}
|