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.
118 lines
3.0 KiB
Go
118 lines
3.0 KiB
Go
package bugbox
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
func newGETEnclosureRequest(id uint64) *http.Request {
|
|
request, _ := http.NewRequest(http.MethodGet, fmt.Sprintf("/enclosure/%d", id), nil)
|
|
return request
|
|
}
|
|
|
|
func newPOSTEventRequest(id uint64, value string) *http.Request {
|
|
body := []byte(value)
|
|
request, _ := http.NewRequest(http.MethodPost, fmt.Sprintf("/enclosure/%d/events", id), bytes.NewBuffer(body))
|
|
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[uint64]*Enclosure
|
|
}
|
|
|
|
func (s *MockEnclosureStore) GetEnclosure(id uint64) (enclosure *Enclosure, ok bool) {
|
|
enclosure, ok = s.enclosures[id]
|
|
return
|
|
}
|
|
|
|
func (s *MockEnclosureStore) RecordEvent(id uint64, event string) error {
|
|
enc, ok := s.enclosures[id]
|
|
if !ok {
|
|
return fmt.Errorf("Failed to find enclosure %d", id)
|
|
}
|
|
enc.RecordEvent(event)
|
|
return nil
|
|
}
|
|
|
|
func TestGETEnclosures(t *testing.T) {
|
|
mockEnclosureStore := MockEnclosureStore{
|
|
map[uint64]*Enclosure{
|
|
1337: &Enclosure{Id: 1337},
|
|
7331: &Enclosure{Id: 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)
|
|
})
|
|
}
|
|
|
|
func TestStoreEnclosure(t *testing.T) {
|
|
testEnclosure := &Enclosure{Id: 2057}
|
|
store := MockEnclosureStore{
|
|
map[uint64]*Enclosure{2057: testEnclosure},
|
|
}
|
|
|
|
server := &BugBoxServer{EnclosureStore: &store}
|
|
|
|
t.Run("it accepts a POST request for an event", func(t *testing.T) {
|
|
request := newPOSTEventRequest(2057, "dirp")
|
|
response := httptest.NewRecorder()
|
|
|
|
server.ServeHTTP(response, request)
|
|
assertStatus(t, response.Code, http.StatusAccepted)
|
|
|
|
if len(testEnclosure.Events) != 1 {
|
|
t.Errorf("Got %d calls to RecordEvent, want %d", len(testEnclosure.Events), 1)
|
|
}
|
|
})
|
|
}
|