Adding upstream version 1.34.4.
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
parent
e393c3af3f
commit
4978089aab
GPG key ID:
FBB4F0E80A80222F
4963 changed files with 677545 additions and 0 deletions
274
plugins/inputs/couchbase/couchbase_test.go
Normal file
274
plugins/inputs/couchbase/couchbase_test.go
Normal file
|
|
@ -0,0 +1,274 @@
|
|||
package couchbase
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"github.com/influxdata/telegraf/plugins/common/tls"
|
||||
"github.com/influxdata/telegraf/testutil"
|
||||
)
|
||||
|
||||
func TestGatherServer(t *testing.T) {
|
||||
bucket := "blastro-df"
|
||||
fakeServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.URL.Path == "/pools" {
|
||||
if _, err := w.Write(readJSON(t, "testdata/pools_response.json")); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
} else if r.URL.Path == "/pools/default" {
|
||||
if _, err := w.Write(readJSON(t, "testdata/pools_default_response.json")); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
} else if r.URL.Path == "/pools/default/buckets" {
|
||||
if _, err := w.Write(readJSON(t, "testdata/bucket_response.json")); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
} else if r.URL.Path == "/pools/default/buckets/"+bucket+"/stats" {
|
||||
if _, err := w.Write(readJSON(t, "testdata/bucket_stats_response.json")); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
} else {
|
||||
w.WriteHeader(http.StatusNotFound)
|
||||
}
|
||||
}))
|
||||
|
||||
cb := Couchbase{
|
||||
ClusterBucketStats: true,
|
||||
BucketStatsIncluded: []string{"quota_percent_used", "ops_per_sec", "disk_fetches", "item_count", "disk_used", "data_used", "mem_used"},
|
||||
}
|
||||
require.NoError(t, cb.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, cb.gatherServer(&acc, fakeServer.URL))
|
||||
|
||||
acc.AssertContainsTaggedFields(t, "couchbase_node",
|
||||
map[string]interface{}{"memory_free": 23181365248.0, "memory_total": 64424656896.0},
|
||||
map[string]string{"cluster": fakeServer.URL, "hostname": "172.16.10.187:8091"})
|
||||
acc.AssertContainsTaggedFields(t, "couchbase_node",
|
||||
map[string]interface{}{"memory_free": 23665811456.0, "memory_total": 64424656896.0},
|
||||
map[string]string{"cluster": fakeServer.URL, "hostname": "172.16.10.65:8091"})
|
||||
acc.AssertContainsTaggedFields(t, "couchbase_bucket",
|
||||
map[string]interface{}{
|
||||
"quota_percent_used": 68.85424936294555,
|
||||
"ops_per_sec": 5686.789686789687,
|
||||
"disk_fetches": 0.0,
|
||||
"item_count": 943239752.0,
|
||||
"disk_used": 409178772321.0,
|
||||
"data_used": 212179309111.0,
|
||||
"mem_used": 202156957464.0,
|
||||
},
|
||||
map[string]string{"cluster": fakeServer.URL, "bucket": "blastro-df"})
|
||||
}
|
||||
|
||||
func TestSanitizeURI(t *testing.T) {
|
||||
var sanitizeTest = []struct {
|
||||
input string
|
||||
expected string
|
||||
}{
|
||||
{"http://user:password@localhost:121", "http://localhost:121"},
|
||||
{"user:password@localhost:12/endpoint", "localhost:12/endpoint"},
|
||||
{"https://mail@address.com:password@localhost", "https://localhost"},
|
||||
{"localhost", "localhost"},
|
||||
{"user:password@localhost:2321", "localhost:2321"},
|
||||
{"http://user:password@couchbase-0.example.com:8091/endpoint", "http://couchbase-0.example.com:8091/endpoint"},
|
||||
{" ", " "},
|
||||
}
|
||||
|
||||
for _, test := range sanitizeTest {
|
||||
result := regexpURI.ReplaceAllString(test.input, "${1}")
|
||||
|
||||
if result != test.expected {
|
||||
t.Errorf("TestSanitizeAddress: input %s, expected %s, actual %s", test.input, test.expected, result)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestGatherDetailedBucketMetrics(t *testing.T) {
|
||||
bucket := "Ducks"
|
||||
node := "172.94.77.2:8091"
|
||||
|
||||
bucketStatsResponse := readJSON(t, "testdata/bucket_stats_response.json")
|
||||
bucketStatsResponseWithMissing := readJSON(t, "testdata/bucket_stats_response_with_missing.json")
|
||||
nodeBucketStatsResponse := readJSON(t, "testdata/node_bucket_stats_response.json")
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
node string
|
||||
response []byte
|
||||
}{
|
||||
{
|
||||
name: "cluster-level with all fields",
|
||||
response: bucketStatsResponse,
|
||||
},
|
||||
{
|
||||
name: "cluster-level with missing fields",
|
||||
response: bucketStatsResponseWithMissing,
|
||||
},
|
||||
{
|
||||
name: "node-level with all fields",
|
||||
response: nodeBucketStatsResponse,
|
||||
node: node,
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
t.Run(test.name, func(t *testing.T) {
|
||||
fakeServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.URL.Path == "/pools/default/buckets/"+bucket+"/stats" || r.URL.Path == "/pools/default/buckets/"+bucket+"/nodes/"+node+"/stats" {
|
||||
if _, err := w.Write(test.response); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
} else {
|
||||
w.WriteHeader(http.StatusNotFound)
|
||||
}
|
||||
}))
|
||||
|
||||
var err error
|
||||
var cb Couchbase
|
||||
cb.BucketStatsIncluded = []string{"couch_total_disk_size"}
|
||||
cb.ClientConfig = tls.ClientConfig{
|
||||
InsecureSkipVerify: true,
|
||||
}
|
||||
err = cb.Init()
|
||||
require.NoError(t, err)
|
||||
var acc testutil.Accumulator
|
||||
bucketStats := &bucketStats{}
|
||||
if err := json.Unmarshal(test.response, bucketStats); err != nil {
|
||||
t.Fatal("parse bucketResponse", err)
|
||||
}
|
||||
|
||||
fields := make(map[string]interface{})
|
||||
err = cb.gatherDetailedBucketStats(fakeServer.URL, bucket, test.node, fields)
|
||||
require.NoError(t, err)
|
||||
|
||||
acc.AddFields("couchbase_bucket", fields, nil)
|
||||
|
||||
// Ensure we gathered only one metric (the one that we configured).
|
||||
require.Len(t, acc.Metrics, 1)
|
||||
require.Len(t, acc.Metrics[0].Fields, 1)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestGatherNodeOnly(t *testing.T) {
|
||||
faker := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.URL.Path == "/pools" {
|
||||
if _, err := w.Write(readJSON(t, "testdata/pools_response.json")); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
} else if r.URL.Path == "/pools/default" {
|
||||
if _, err := w.Write(readJSON(t, "testdata/pools_default_response.json")); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
} else if r.URL.Path == "/pools/default/buckets" {
|
||||
if _, err := w.Write(readJSON(t, "testdata/bucket_response.json")); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
} else {
|
||||
w.WriteHeader(http.StatusNotFound)
|
||||
}
|
||||
}))
|
||||
|
||||
cb := Couchbase{
|
||||
Servers: []string{faker.URL},
|
||||
}
|
||||
require.NoError(t, cb.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, cb.gatherServer(&acc, faker.URL))
|
||||
|
||||
require.Empty(t, acc.Errors)
|
||||
require.Len(t, acc.Metrics, 7)
|
||||
acc.AssertDoesNotContainMeasurement(t, "couchbase_bucket")
|
||||
}
|
||||
|
||||
func TestGatherFailover(t *testing.T) {
|
||||
faker := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
switch r.URL.Path {
|
||||
case "/pools":
|
||||
if _, err := w.Write(readJSON(t, "testdata/pools_response.json")); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
case "/pools/default":
|
||||
if _, err := w.Write(readJSON(t, "testdata/pools_default_response.json")); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
case "/pools/default/buckets":
|
||||
if _, err := w.Write(readJSON(t, "testdata/bucket_response.json")); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
case "/settings/autoFailover":
|
||||
if _, err := w.Write(readJSON(t, "testdata/settings_autofailover.json")); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
default:
|
||||
w.WriteHeader(http.StatusNotFound)
|
||||
}
|
||||
}))
|
||||
|
||||
cb := Couchbase{
|
||||
Servers: []string{faker.URL},
|
||||
ClusterBucketStats: false,
|
||||
NodeBucketStats: false,
|
||||
AdditionalStats: []string{"autofailover"},
|
||||
}
|
||||
require.NoError(t, cb.Init())
|
||||
|
||||
var acc testutil.Accumulator
|
||||
require.NoError(t, cb.gatherServer(&acc, faker.URL))
|
||||
require.Empty(t, acc.Errors)
|
||||
require.Len(t, acc.Metrics, 8)
|
||||
|
||||
var metric *testutil.Metric
|
||||
for _, m := range acc.Metrics {
|
||||
if m.Measurement == "couchbase_autofailover" {
|
||||
metric = m
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
require.NotNil(t, metric)
|
||||
require.Equal(t, 1, metric.Fields["count"])
|
||||
v, ok := metric.Fields["enabled"].(bool)
|
||||
require.Truef(t, ok, "bool type expected, got '%T' with '%v' value instead", metric.Fields["enabled"], metric.Fields["enabled"])
|
||||
require.True(t, v)
|
||||
|
||||
require.Equal(t, 2, metric.Fields["max_count"])
|
||||
require.Equal(t, 72, metric.Fields["timeout"])
|
||||
}
|
||||
|
||||
func readJSON(t *testing.T, jsonFilePath string) []byte {
|
||||
data, err := os.ReadFile(jsonFilePath)
|
||||
require.NoErrorf(t, err, "could not read from data file %s", jsonFilePath)
|
||||
|
||||
return data
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue