70 lines
2.0 KiB
Go
70 lines
2.0 KiB
Go
package influxql_test
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/influxdata/influxdb/v2/query"
|
|
"github.com/influxdata/influxdb/v2/query/influxql"
|
|
)
|
|
|
|
func TestService(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
// Verify the parameters were passed correctly.
|
|
if want, got := "POST", r.Method; want != got {
|
|
t.Errorf("unexpected method -want/+got\n\t- %q\n\t+ %q", want, got)
|
|
}
|
|
if want, got := "SHOW DATABASES", r.FormValue("q"); want != got {
|
|
t.Errorf("unexpected query -want/+got\n\t- %q\n\t+ %q", want, got)
|
|
}
|
|
if want, got := "db0", r.FormValue("db"); want != got {
|
|
t.Errorf("unexpected database -want/+got\n\t- %q\n\t+ %q", want, got)
|
|
}
|
|
if want, got := "rp0", r.FormValue("rp"); want != got {
|
|
t.Errorf("unexpected retention policy -want/+got\n\t- %q\n\t+ %q", want, got)
|
|
}
|
|
user, pass, ok := r.BasicAuth()
|
|
if !ok {
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
return
|
|
}
|
|
if want, got := "me", user; want != got {
|
|
t.Errorf("unexpected username -want/+got\n\t- %q\n\t+ %q", want, got)
|
|
}
|
|
if want, got := "secretpassword", pass; want != got {
|
|
t.Errorf("unexpected password -want/+got\n\t- %q\n\t+ %q", want, got)
|
|
}
|
|
io.WriteString(w, `{"results":[{"statement_id":0,"series":[{"name":"databases","columns":["name"],"values":[["db0"]]}]}]}`)
|
|
}))
|
|
defer server.Close()
|
|
|
|
service := &influxql.Service{
|
|
Endpoints: map[string]influxql.Endpoint{
|
|
"myserver": {
|
|
URL: server.URL,
|
|
Username: "me",
|
|
Password: "secretpassword",
|
|
},
|
|
},
|
|
}
|
|
req := &query.Request{Compiler: &influxql.Compiler{
|
|
Cluster: "myserver",
|
|
DB: "db0",
|
|
RP: "rp0",
|
|
Query: "SHOW DATABASES",
|
|
}}
|
|
|
|
results, err := service.Query(context.Background(), req)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %s", err)
|
|
}
|
|
defer results.Release()
|
|
_, err = service.QueryRawJSON(context.Background(), req)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %s", err)
|
|
}
|
|
}
|