127 lines
3.1 KiB
Go
127 lines
3.1 KiB
Go
package client_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"testing"
|
|
|
|
"github.com/influxdb/influxdb"
|
|
"github.com/influxdb/influxdb/client"
|
|
)
|
|
|
|
func TestNewClient(t *testing.T) {
|
|
config := client.Config{}
|
|
_, err := client.NewClient(config)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error. expected %v, actual %v", nil, err)
|
|
}
|
|
}
|
|
|
|
func TestClient_Ping(t *testing.T) {
|
|
ts := emptyTestServer()
|
|
defer ts.Close()
|
|
|
|
u, _ := url.Parse(ts.URL)
|
|
config := client.Config{URL: *u}
|
|
c, err := client.NewClient(config)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error. expected %v, actual %v", nil, err)
|
|
}
|
|
d, version, err := c.Ping()
|
|
if err != nil {
|
|
t.Fatalf("unexpected error. expected %v, actual %v", nil, err)
|
|
}
|
|
if d == 0 {
|
|
t.Fatalf("expected a duration greater than zero. actual %v", d)
|
|
}
|
|
if version != "x.x" {
|
|
t.Fatalf("unexpected version. expected %s, actual %v", "x.x", version)
|
|
}
|
|
}
|
|
|
|
func TestClient_Query(t *testing.T) {
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
var data influxdb.Results
|
|
w.WriteHeader(http.StatusOK)
|
|
_ = json.NewEncoder(w).Encode(data)
|
|
}))
|
|
defer ts.Close()
|
|
|
|
u, _ := url.Parse(ts.URL)
|
|
config := client.Config{URL: *u}
|
|
c, err := client.NewClient(config)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error. expected %v, actual %v", nil, err)
|
|
}
|
|
|
|
query := client.Query{}
|
|
_, err = c.Query(query)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error. expected %v, actual %v", nil, err)
|
|
}
|
|
}
|
|
|
|
func TestClient_BasicAuth(t *testing.T) {
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
u, p, ok := r.BasicAuth()
|
|
|
|
if !ok {
|
|
t.Errorf("basic auth failed")
|
|
}
|
|
if u != "username" {
|
|
t.Errorf("unexpected username, expected %q, actual %q", "username", u)
|
|
}
|
|
if p != "password" {
|
|
t.Errorf("unexpected password, expected %q, actual %q", "password", p)
|
|
}
|
|
w.WriteHeader(http.StatusNoContent)
|
|
}))
|
|
defer ts.Close()
|
|
|
|
u, _ := url.Parse(ts.URL)
|
|
u.User = url.UserPassword("username", "password")
|
|
config := client.Config{URL: *u, Username: "username", Password: "password"}
|
|
c, err := client.NewClient(config)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error. expected %v, actual %v", nil, err)
|
|
}
|
|
|
|
_, _, err = c.Ping()
|
|
if err != nil {
|
|
t.Fatalf("unexpected error. expected %v, actual %v", nil, err)
|
|
}
|
|
}
|
|
|
|
func TestClient_Write(t *testing.T) {
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
var data influxdb.Results
|
|
w.WriteHeader(http.StatusOK)
|
|
_ = json.NewEncoder(w).Encode(data)
|
|
}))
|
|
defer ts.Close()
|
|
|
|
u, _ := url.Parse(ts.URL)
|
|
config := client.Config{URL: *u}
|
|
c, err := client.NewClient(config)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error. expected %v, actual %v", nil, err)
|
|
}
|
|
|
|
write := client.Write{}
|
|
_, err = c.Write(write)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error. expected %v, actual %v", nil, err)
|
|
}
|
|
}
|
|
|
|
// helper functions
|
|
|
|
func emptyTestServer() *httptest.Server {
|
|
return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("X-Influxdb-Version", "x.x")
|
|
return
|
|
}))
|
|
}
|