Moves conversion of nil slices up to HTTP layer for prepared queries.

This commit is contained in:
James Phillips 2015-11-14 21:59:23 -08:00
parent c032dee945
commit a1e02996e5
4 changed files with 81 additions and 3 deletions

View File

@ -54,6 +54,11 @@ func (s *HTTPServer) PreparedQueryGeneral(resp http.ResponseWriter, req *http.Re
if err := s.agent.RPC(endpoint+".List", &args, &reply); err != nil { if err := s.agent.RPC(endpoint+".List", &args, &reply); err != nil {
return nil, err return nil, err
} }
// Use empty list instead of nil.
if reply.Queries == nil {
reply.Queries = make(structs.PreparedQueries, 0)
}
return reply.Queries, nil return reply.Queries, nil
default: default:
@ -111,6 +116,11 @@ func (s *HTTPServer) PreparedQuerySpecific(resp http.ResponseWriter, req *http.R
} }
return nil, err return nil, err
} }
// Use empty list instead of nil.
if reply.Nodes == nil {
reply.Nodes = make(structs.CheckServiceNodes, 0)
}
return reply, nil return reply, nil
} else { } else {
args := structs.PreparedQuerySpecificRequest{ args := structs.PreparedQuerySpecificRequest{

View File

@ -144,6 +144,40 @@ func TestPreparedQuery_Create(t *testing.T) {
} }
func TestPreparedQuery_List(t *testing.T) { func TestPreparedQuery_List(t *testing.T) {
httpTest(t, func(srv *HTTPServer) {
m := MockPreparedQuery{}
if err := srv.agent.InjectEndpoint("PreparedQuery", &m); err != nil {
t.Fatalf("err: %v", err)
}
m.listFn = func(args *structs.DCSpecificRequest, reply *structs.IndexedPreparedQueries) error {
// Return an empty response.
return nil
}
body := bytes.NewBuffer(nil)
req, err := http.NewRequest("GET", "/v1/query", body)
if err != nil {
t.Fatalf("err: %v", err)
}
resp := httptest.NewRecorder()
obj, err := srv.PreparedQueryGeneral(resp, req)
if err != nil {
t.Fatalf("err: %v", err)
}
if resp.Code != 200 {
t.Fatalf("bad code: %d", resp.Code)
}
r, ok := obj.(structs.PreparedQueries)
if !ok {
t.Fatalf("unexpected: %T", obj)
}
if r == nil || len(r) != 0 {
t.Fatalf("bad: %v", r)
}
})
httpTest(t, func(srv *HTTPServer) { httpTest(t, func(srv *HTTPServer) {
m := MockPreparedQuery{} m := MockPreparedQuery{}
if err := srv.agent.InjectEndpoint("PreparedQuery", &m); err != nil { if err := srv.agent.InjectEndpoint("PreparedQuery", &m); err != nil {
@ -194,6 +228,40 @@ func TestPreparedQuery_List(t *testing.T) {
} }
func TestPreparedQuery_Execute(t *testing.T) { func TestPreparedQuery_Execute(t *testing.T) {
httpTest(t, func(srv *HTTPServer) {
m := MockPreparedQuery{}
if err := srv.agent.InjectEndpoint("PreparedQuery", &m); err != nil {
t.Fatalf("err: %v", err)
}
m.executeFn = func(args *structs.PreparedQueryExecuteRequest, reply *structs.PreparedQueryExecuteResponse) error {
// Just return an empty response.
return nil
}
body := bytes.NewBuffer(nil)
req, err := http.NewRequest("GET", "/v1/query/my-id/execute", body)
if err != nil {
t.Fatalf("err: %v", err)
}
resp := httptest.NewRecorder()
obj, err := srv.PreparedQuerySpecific(resp, req)
if err != nil {
t.Fatalf("err: %v", err)
}
if resp.Code != 200 {
t.Fatalf("bad code: %d", resp.Code)
}
r, ok := obj.(structs.PreparedQueryExecuteResponse)
if !ok {
t.Fatalf("unexpected: %T", obj)
}
if r.Nodes == nil || len(r.Nodes) != 0 {
t.Fatalf("bad: %v", r)
}
})
httpTest(t, func(srv *HTTPServer) { httpTest(t, func(srv *HTTPServer) {
m := MockPreparedQuery{} m := MockPreparedQuery{}
if err := srv.agent.InjectEndpoint("PreparedQuery", &m); err != nil { if err := srv.agent.InjectEndpoint("PreparedQuery", &m); err != nil {

View File

@ -243,7 +243,7 @@ func (s *StateStore) PreparedQueryList() (uint64, structs.PreparedQueries, error
} }
// Go over all of the queries and build the response. // Go over all of the queries and build the response.
result := make(structs.PreparedQueries, 0) var result structs.PreparedQueries
for query := queries.Next(); query != nil; query = queries.Next() { for query := queries.Next(); query != nil; query = queries.Next() {
result = append(result, query.(*structs.PreparedQuery)) result = append(result, query.(*structs.PreparedQuery))
} }

View File

@ -379,7 +379,7 @@ func TestStateStore_PreparedQueryLookup(t *testing.T) {
func TestStateStore_PreparedQueryList(t *testing.T) { func TestStateStore_PreparedQueryList(t *testing.T) {
s := testStateStore(t) s := testStateStore(t)
// Make sure an empty (non-nil) slice is returned if there are no queries. // Make sure nothing is returned for an empty query
idx, actual, err := s.PreparedQueryList() idx, actual, err := s.PreparedQueryList()
if err != nil { if err != nil {
t.Fatalf("err: %s", err) t.Fatalf("err: %s", err)
@ -387,7 +387,7 @@ func TestStateStore_PreparedQueryList(t *testing.T) {
if idx != 0 { if idx != 0 {
t.Fatalf("bad index: %d", idx) t.Fatalf("bad index: %d", idx)
} }
if actual == nil || len(actual) != 0 { if len(actual) != 0 {
t.Fatalf("bad: %v", actual) t.Fatalf("bad: %v", actual)
} }