fleet/server/service/transport_targets_test.go
Zach Wasserman 8ee47f8b23
Add support for Teams in targets and live queries (#880)
- Accept Teams as a searchable target type for the target selection API.
- Accept Teams for targets in running live queries.
- Refactoring to support these changes.
- Update API documentation.
2021-05-27 13:18:00 -07:00

47 lines
907 B
Go

package service
import (
"bytes"
"context"
"net/http"
"net/http/httptest"
"testing"
"github.com/gorilla/mux"
"github.com/stretchr/testify/assert"
)
func TestDecodeSearchTargetsRequest(t *testing.T) {
router := mux.NewRouter()
router.HandleFunc("/api/v1/fleet/targets", func(writer http.ResponseWriter, request *http.Request) {
r, err := decodeSearchTargetsRequest(context.Background(), request)
assert.Nil(t, err)
params := r.(searchTargetsRequest)
assert.Equal(t, "bar", params.MatchQuery)
assert.Len(t, params.Selected.HostIDs, 3)
assert.Len(t, params.Selected.LabelIDs, 2)
}).Methods("POST")
var body bytes.Buffer
body.Write([]byte(`{
"query": "bar",
"selected": {
"hosts": [
1,
2,
3
],
"labels": [
1,
2
]
}
}`))
router.ServeHTTP(
httptest.NewRecorder(),
httptest.NewRequest("POST", "/api/v1/fleet/targets", &body),
)
}