func TestAddValidPositionCreatesCommand(t *testing.T) {
var (
request *http.Request
recorder *httptest.ResponseRecorder
)
dispatcher := fakes.NewFakeQueueDispatcher()
server := MakeTestServer(dispatcher)
recorder = httptest.NewRecorder()
body := []byte("{\"drone_id\":\"positiondrone1\", \"latitude\": 81.231, \"longitude\": 43.1231, \"altitude\": 2301.1, \"current_speed\": 41.3, \"heading_cardinal\": 1}")
reader := bytes.NewReader(body)
request, _ = http.NewRequest("POST", "/api/cmds/positions", reader)
server.ServeHTTP(recorder, request)
if recorder.Code != http.StatusCreated {
t.Errorf("Expected creation of new position item to return 201, got %d/%s", recorder.Code, string(recorder.Body.Bytes()))
}
if len(dispatcher.Messages) != 1 {
t.Errorf("Expected queue dispatch count of 1, got %d", len(dispatcher.Messages))
}
var positionResponse dronescommon.PositionChangedEvent
payload := recorder.Body.Bytes()
err := json.Unmarshal(payload, &positionResponse)
if err != nil {
t.Errorf("Could not unmarshal payload into position response object")
}
if positionResponse.DroneID != "positiondrone1" {
t.Errorf("Expected drone ID of 'positiondrone1' got %s", positionResponse.DroneID)
}
if positionResponse.CurrentSpeed != 41.3 {
t.Errorf("Expected drone speed of 41.3, got %f", positionResponse.CurrentSpeed)
}
}
func TestAddValidAlertCreatesCommand(t *testing.T) {
var (
request *http.Request
recorder *httptest.ResponseRecorder
)
dispatcher := fakes.NewFakeQueueDispatcher()
server := MakeTestServer(dispatcher)
recorder = httptest.NewRecorder()
body := []byte("{\"drone_id\":\"alertingdrone4\", \"fault_code\": 12, \"description\": \"all the things are failing\"}")
reader := bytes.NewReader(body)
request, _ = http.NewRequest("POST", "/api/cmds/alerts", reader)
server.ServeHTTP(recorder, request)
if recorder.Code != http.StatusCreated {
t.Errorf("Expected creation of new alert item to return 201, got %d/%s", recorder.Code, string(recorder.Body.Bytes()))
}
if len(dispatcher.Messages) != 1 {
t.Errorf("Expected queue dispatch count of 1, got %d", len(dispatcher.Messages))
}
var alertResponse dronescommon.AlertSignalledEvent
payload := recorder.Body.Bytes()
err := json.Unmarshal(payload, &alertResponse)
if err != nil {
t.Errorf("Could not unmarshal payload into alertResponse object")
}
if alertResponse.DroneID != "alertingdrone4" {
t.Errorf("Expected drone ID of 'alertingdrone4' got %s", alertResponse.DroneID)
}
if alertResponse.FaultCode != 12 {
t.Errorf("Expected drone fault code of 12, got %d", alertResponse.FaultCode)
}
}
func TestAddValidTelemetryCreatesCommand(t *testing.T) {
var (
request *http.Request
recorder *httptest.ResponseRecorder
)
dispatcher := fakes.NewFakeQueueDispatcher()
server := MakeTestServer(dispatcher)
recorder = httptest.NewRecorder()
body := []byte("{\"drone_id\":\"drone666\", \"battery\": 72, \"uptime\": 6941, \"core_temp\": 21 }")
reader := bytes.NewReader(body)
request, _ = http.NewRequest("POST", "/api/cmds/telemetry", reader)
server.ServeHTTP(recorder, request)
if recorder.Code != http.StatusCreated {
t.Errorf("Expected creation of new telemetry item to return 201, got %d", recorder.Code)
}
if len(dispatcher.Messages) != 1 {
t.Errorf("Expected queue dispatch count of 1, got %d", len(dispatcher.Messages))
}
var telemetryResponse dronescommon.TelemetryUpdatedEvent
payload := recorder.Body.Bytes()
err := json.Unmarshal(payload, &telemetryResponse)
if err != nil {
t.Errorf("Could not unmarshal payload into telemetry response object")
}
if telemetryResponse.DroneID != "drone666" {
t.Errorf("Expected drone ID of 'drone666' got %s", telemetryResponse.DroneID)
}
if telemetryResponse.Uptime != 6941 {
t.Errorf("Expected drone uptime of 6941, got %d", telemetryResponse.Uptime)
}
}
func buildDispatcher(queueName string, appEnv *cfenv.App) queueDispatcher {
url := resolveAMQPURL(appEnv)
if strings.Compare(url, "fake://foo") == 0 {
fmt.Printf("Building fake dispatcher for queue '%s'", queueName)
return fakes.NewFakeQueueDispatcher()
}
return createAMQPDispatcher(queueName, url)
}
func TestAddInvalidTelemetryReturnsBadRequest(t *testing.T) {
var (
request *http.Request
recorder *httptest.ResponseRecorder
)
dispatcher := fakes.NewFakeQueueDispatcher()
server := MakeTestServer(dispatcher)
recorder = httptest.NewRecorder()
body := []byte("{\"foo\":\"bar\"}")
reader := bytes.NewReader(body)
request, _ = http.NewRequest("POST", "/api/cmds/telemetry", reader)
server.ServeHTTP(recorder, request)
if recorder.Code != http.StatusBadRequest {
t.Errorf("Expected creation of invalid/unparseable new telemetry item to return bad request, got %d", recorder.Code)
}
if len(dispatcher.Messages) != 0 {
t.Errorf("Expected dispatcher to dispatch 0 messages, got %d", len(dispatcher.Messages))
}
}