112 lines
2.8 KiB
Go
112 lines
2.8 KiB
Go
// Copyright 2015 The Prometheus Authors
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package influxdb
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"math"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"testing"
|
|
"time"
|
|
|
|
influx "github.com/influxdb/influxdb/client"
|
|
|
|
"github.com/prometheus/common/model"
|
|
)
|
|
|
|
func TestClient(t *testing.T) {
|
|
samples := model.Samples{
|
|
{
|
|
Metric: model.Metric{
|
|
model.MetricNameLabel: "testmetric",
|
|
"test_label": "test_label_value1",
|
|
},
|
|
Timestamp: model.Time(123456789123),
|
|
Value: 1.23,
|
|
},
|
|
{
|
|
Metric: model.Metric{
|
|
model.MetricNameLabel: "testmetric",
|
|
"test_label": "test_label_value2",
|
|
},
|
|
Timestamp: model.Time(123456789123),
|
|
Value: 5.1234,
|
|
},
|
|
{
|
|
Metric: model.Metric{
|
|
model.MetricNameLabel: "nan_value",
|
|
},
|
|
Timestamp: model.Time(123456789123),
|
|
Value: model.SampleValue(math.NaN()),
|
|
},
|
|
{
|
|
Metric: model.Metric{
|
|
model.MetricNameLabel: "pos_inf_value",
|
|
},
|
|
Timestamp: model.Time(123456789123),
|
|
Value: model.SampleValue(math.Inf(1)),
|
|
},
|
|
{
|
|
Metric: model.Metric{
|
|
model.MetricNameLabel: "neg_inf_value",
|
|
},
|
|
Timestamp: model.Time(123456789123),
|
|
Value: model.SampleValue(math.Inf(-1)),
|
|
},
|
|
}
|
|
|
|
expectedBody := `testmetric,test_label=test_label_value1 value=1.23 123456789123000000
|
|
testmetric,test_label=test_label_value2 value=5.1234 123456789123000000
|
|
`
|
|
|
|
server := httptest.NewServer(http.HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != "POST" {
|
|
t.Fatalf("Unexpected method; expected POST, got %s", r.Method)
|
|
}
|
|
if r.URL.Path != "/write" {
|
|
t.Fatalf("Unexpected path; expected %s, got %s", "/write", r.URL.Path)
|
|
}
|
|
b, err := ioutil.ReadAll(r.Body)
|
|
if err != nil {
|
|
t.Fatalf("Error reading body: %s", err)
|
|
}
|
|
|
|
if string(b) != expectedBody {
|
|
t.Fatalf("Unexpected request body; expected:\n\n%s\n\ngot:\n\n%s", expectedBody, string(b))
|
|
}
|
|
},
|
|
))
|
|
defer server.Close()
|
|
|
|
serverURL, err := url.Parse(server.URL)
|
|
if err != nil {
|
|
t.Fatalf("Unable to parse server URL %s: %s", server.URL, err)
|
|
}
|
|
|
|
conf := influx.Config{
|
|
URL: *serverURL,
|
|
Username: "testuser",
|
|
Password: "testpass",
|
|
Timeout: time.Minute,
|
|
}
|
|
c := NewClient(conf, "test_db", "default")
|
|
|
|
if err := c.Store(samples); err != nil {
|
|
t.Fatalf("Error sending samples: %s", err)
|
|
}
|
|
}
|