package bind
import (
"net"
"net/http"
"net/http/httptest"
"testing"
"github.com/influxdata/telegraf/testutil"
"github.com/stretchr/testify/assert"
)
func TestBindJsonStats(t *testing.T) {
ts := httptest.NewServer(http.FileServer(http.Dir("testdata")))
url := ts.Listener.Addr().String()
host, port, _ := net.SplitHostPort(url)
defer ts.Close()
b := Bind{
Urls: []string{ts.URL + "/json/v1"},
GatherMemoryContexts: true,
GatherViews: true,
}
var acc testutil.Accumulator
err := acc.GatherError(b.Gather)
assert.Nil(t, err)
// Use subtests for counters, since they are similar structure
type fieldSet struct {
fieldKey string
fieldValue int64
}
testCases := []struct {
counterType string
values []fieldSet
}{
{
"opcode",
[]fieldSet{
{"NOTIFY", 0},
{"UPDATE", 0},
{"IQUERY", 0},
{"QUERY", 13},
{"STATUS", 0},
},
},
{
"qtype",
[]fieldSet{
{"A", 2},
{"AAAA", 2},
{"PTR", 7},
{"SRV", 2},
},
},
{
"nsstat",
[]fieldSet{
{"QrySuccess", 6},
{"QryRecursion", 12},
{"Requestv4", 13},
{"QryNXDOMAIN", 4},
{"QryAuthAns", 1},
{"QryNxrrset", 1},
{"QryNoauthAns", 10},
{"QryUDP", 13},
{"QryDuplicate", 1},
{"QrySERVFAIL", 1},
{"Response", 12},
},
},
{
"sockstat",
[]fieldSet{
{"TCP4Open", 118},
{"UDP6Close", 112},
{"UDP4Close", 333},
{"TCP4Close", 119},
{"TCP6Active", 2},
{"UDP4Active", 2},
{"UDP4RecvErr", 1},
{"UDP4Open", 335},
{"TCP4Active", 10},
{"RawActive", 1},
{"UDP6ConnFail", 112},
{"TCP4Conn", 114},
{"UDP6Active", 1},
{"UDP6Open", 113},
{"UDP4Conn", 333},
{"UDP6SendErr", 112},
{"RawOpen", 1},
{"TCP4Accept", 6},
{"TCP6Open", 2},
},
},
}
for _, tc := range testCases {
t.Run(tc.counterType, func(t *testing.T) {
tags := map[string]string{
"url": url,
"type": tc.counterType,
"source": host,
"port": port,
}
fields := map[string]interface{}{}
for _, val := range tc.values {
fields[val.fieldKey] = val.fieldValue
}
acc.AssertContainsTaggedFields(t, "bind_counter", fields, tags)
})
}
// Subtest for memory stats
t.Run("memory", func(t *testing.T) {
tags := map[string]string{
"url": url,
"source": host,
"port": port,
}
fields := map[string]interface{}{
"block_size": 13893632,
"context_size": 3685480,
"in_use": 3064368,
"lost": 0,
"total_use": 18206566,
}
acc.AssertContainsTaggedFields(t, "bind_memory", fields, tags)
})
// Subtest for per-context memory stats
t.Run("memory_context", func(t *testing.T) {
assert.True(t, acc.HasIntField("bind_memory_context", "total"))
assert.True(t, acc.HasIntField("bind_memory_context", "in_use"))
})
}
func TestBindXmlStatsV2(t *testing.T) {
ts := httptest.NewServer(http.FileServer(http.Dir("testdata")))
url := ts.Listener.Addr().String()
host, port, _ := net.SplitHostPort(url)
defer ts.Close()
b := Bind{
Urls: []string{ts.URL + "/xml/v2"},
GatherMemoryContexts: true,
GatherViews: true,
}
var acc testutil.Accumulator
err := acc.GatherError(b.Gather)
assert.Nil(t, err)
// Use subtests for counters, since they are similar structure
type fieldSet struct {
fieldKey string
fieldValue int64
}
testCases := []struct {
counterType string
values []fieldSet
}{
{
"opcode",
[]fieldSet{
{"UPDATE", 238},
{"QUERY", 102312374},
},
},
{
"qtype",
[]fieldSet{
{"ANY", 7},
{"DNSKEY", 452},
{"SSHFP", 2987},
{"SOA", 100415},
{"AAAA", 37786321},
{"MX", 441155},
{"IXFR", 157},
{"CNAME", 531},
{"NS", 1999},
{"TXT", 34628},
{"A", 58951432},
{"SRV", 741082},
{"PTR", 4211487},
{"NAPTR", 39137},
{"DS", 584},
},
},
{
"nsstat",
[]fieldSet{
{"XfrReqDone", 157},
{"ReqEdns0", 441758},
{"ReqTSIG", 0},
{"UpdateRespFwd", 0},
{"RespEDNS0", 441748},
{"QryDropped", 16},
{"RPZRewrites", 0},
{"XfrRej", 0},
{"RecQryRej", 0},
{"QryNxrrset", 24423133},
{"QryFORMERR", 0},
{"ReqTCP", 1548156},
{"UpdateDone", 0},
{"QrySERVFAIL", 14422},
{"QryRecursion", 2104239},
{"Requestv4", 102312611},
{"UpdateFwdFail", 0},
{"QryReferral", 3},
{"Response", 102301560},
{"RespTSIG", 0},
{"QrySuccess", 63811668},
{"QryFailure", 0},
{"RespSIG0", 0},
{"ReqSIG0", 0},
{"UpdateRej", 238},
{"QryAuthAns", 72180718},
{"UpdateFail", 0},
{"QryDuplicate", 10879},
{"RateDropped", 0},
{"QryNoauthAns", 30106182},
{"QryNXDOMAIN", 14052096},
{"ReqBadSIG", 0},
{"UpdateReqFwd", 0},
{"RateSlipped", 0},
{"TruncatedResp", 3787},
{"Requestv6", 1},
{"UpdateBadPrereq", 0},
{"AuthQryRej", 0},
{"ReqBadEDNSVer", 0},
},
},
{
"sockstat",
[]fieldSet{
{"FdwatchBindFail", 0},
{"UDP6Open", 238269},
{"UDP6SendErr", 238250},
{"TCP4ConnFail", 0},
{"TCP4Conn", 590},
{"TCP6AcceptFail", 0},
{"UDP4SendErr", 0},
{"FDwatchConn", 0},
{"TCP4RecvErr", 1},
{"TCP4OpenFail", 0},
{"UDP4OpenFail", 0},
{"UDP6OpenFail", 0},
{"TCP4Close", 1548268},
{"TCP6BindFail", 0},
{"TCP4AcceptFail", 0},
{"UnixConn", 0},
{"UDP4Open", 3765532},
{"TCP6Close", 0},
{"FDwatchRecvErr", 0},
{"UDP4Conn", 3764828},
{"UnixConnFail", 0},
{"TCP6Conn", 0},
{"TCP6OpenFail", 0},
{"TCP6SendErr", 0},
{"TCP6RecvErr", 0},
{"FDwatchSendErr", 0},
{"UDP4RecvErr", 1650},
{"UDP4ConnFail", 0},
{"UDP6Close", 238267},
{"FDWatchClose", 0},
{"TCP4Accept", 1547672},
{"UnixAccept", 0},
{"TCP4Open", 602},
{"UDP4BindFail", 219},
{"UDP6ConnFail", 238250},
{"UnixClose", 0},
{"TCP4BindFail", 0},
{"UnixOpenFail", 0},
{"UDP6BindFail", 16},
{"UnixOpen", 0},
{"UnixAcceptFail", 0},
{"UnixRecvErr", 0},
{"UDP6RecvErr", 0},
{"TCP6ConnFail", 0},
{"FDwatchConnFail", 0},
{"TCP4SendErr", 0},
{"UDP4Close", 3765528},
{"UnixSendErr", 0},
{"TCP6Open", 2},
{"UDP6Conn", 1},
{"TCP6Accept", 0},
{"UnixBindFail", 0},
},
},
}
for _, tc := range testCases {
t.Run(tc.counterType, func(t *testing.T) {
tags := map[string]string{
"url": url,
"type": tc.counterType,
"source": host,
"port": port,
}
fields := map[string]interface{}{}
for _, val := range tc.values {
fields[val.fieldKey] = val.fieldValue
}
acc.AssertContainsTaggedFields(t, "bind_counter", fields, tags)
})
}
// Subtest for memory stats
t.Run("memory", func(t *testing.T) {
tags := map[string]string{
"url": url,
"source": host,
"port": port,
}
fields := map[string]interface{}{
"block_size": 77070336,
"context_size": 6663840,
"in_use": 20772579,
"lost": 0,
"total_use": 81804609,
}
acc.AssertContainsTaggedFields(t, "bind_memory", fields, tags)
})
// Subtest for per-context memory stats
t.Run("memory_context", func(t *testing.T) {
assert.True(t, acc.HasIntField("bind_memory_context", "total"))
assert.True(t, acc.HasIntField("bind_memory_context", "in_use"))
})
}
func TestBindXmlStatsV3(t *testing.T) {
ts := httptest.NewServer(http.FileServer(http.Dir("testdata")))
url := ts.Listener.Addr().String()
host, port, _ := net.SplitHostPort(url)
defer ts.Close()
b := Bind{
Urls: []string{ts.URL + "/xml/v3"},
GatherMemoryContexts: true,
GatherViews: true,
}
var acc testutil.Accumulator
err := acc.GatherError(b.Gather)
assert.Nil(t, err)
// Use subtests for counters, since they are similar structure
type fieldSet struct {
fieldKey string
fieldValue int64
}
testCases := []struct {
counterType string
values []fieldSet
}{
{
"opcode",
[]fieldSet{
{"NOTIFY", 0},
{"UPDATE", 0},
{"IQUERY", 0},
{"QUERY", 74941},
{"STATUS", 0},
},
},
{
"qtype",
[]fieldSet{
{"ANY", 22},
{"SOA", 18},
{"AAAA", 5735},
{"MX", 618},
{"NS", 373},
{"TXT", 970},
{"A", 63672},
{"SRV", 139},
{"PTR", 3393},
{"RRSIG", 1},
},
},
{
"nsstat",
[]fieldSet{
{"DNS64", 0},
{"ExpireOpt", 0},
{"NSIDOpt", 0},
{"OtherOpt", 59},
{"XfrReqDone", 0},
{"ReqEdns0", 9250},
{"ReqTSIG", 0},
{"UpdateRespFwd", 0},
{"RespEDNS0", 9250},
{"QryDropped", 11},
{"RPZRewrites", 0},
{"XfrRej", 0},
{"RecQryRej", 35},
{"QryNxrrset", 2452},
{"QryFORMERR", 0},
{"ReqTCP", 260},
{"QryTCP", 258},
{"QryUDP", 74648},
{"UpdateDone", 0},
{"QrySERVFAIL", 122},
{"QryRecursion", 53750},
{"RecursClients", 0},
{"Requestv4", 74942},
{"UpdateFwdFail", 0},
{"QryReferral", 0},
{"Response", 63264},
{"RespTSIG", 0},
{"QrySuccess", 49044},
{"QryFailure", 35},
{"RespSIG0", 0},
{"ReqSIG0", 0},
{"UpdateRej", 0},
{"QryAuthAns", 2752},
{"UpdateFail", 0},
{"QryDuplicate", 11667},
{"RateDropped", 0},
{"QryNoauthAns", 60354},
{"QryNXDOMAIN", 11610},
{"ReqBadSIG", 0},
{"UpdateReqFwd", 0},
{"RateSlipped", 0},
{"TruncatedResp", 365},
{"Requestv6", 0},
{"UpdateBadPrereq", 0},
{"AuthQryRej", 0},
{"ReqBadEDNSVer", 0},
{"SitBadSize", 0},
{"SitBadTime", 0},
{"SitMatch", 0},
{"SitNew", 0},
{"SitNoMatch", 0},
{"SitOpt", 0},
{"TruncatedResp", 365},
},
},
{
"sockstat",
[]fieldSet{
{"FDwatchConnFail", 0},
{"UnixClose", 0},
{"TCP6OpenFail", 0},
{"TCP6Active", 0},
{"UDP4RecvErr", 14},
{"TCP6Conn", 0},
{"FDWatchClose", 0},
{"TCP4ConnFail", 0},
{"UnixConn", 0},
{"UnixSendErr", 0},
{"UDP6Close", 0},
{"UnixOpen", 0},
{"UDP4Conn", 92535},
{"TCP4Close", 336},
{"UnixAcceptFail", 0},
{"UnixAccept", 0},
{"TCP6AcceptFail", 0},
{"UDP6Open", 0},
{"UDP6BindFail", 0},
{"UDP6RecvErr", 0},
{"RawOpenFail", 0},
{"TCP4Accept", 293},
{"UDP6SendErr", 0},
{"UDP6Conn", 0},
{"TCP4SendErr", 0},
{"UDP4BindFail", 1},
{"UDP4Active", 4},
{"TCP4Active", 297},
{"UnixConnFail", 0},
{"UnixOpenFail", 0},
{"UDP6ConnFail", 0},
{"TCP6Accept", 0},
{"UnixRecvErr", 0},
{"RawActive", 1},
{"UDP6OpenFail", 0},
{"RawClose", 0},
{"UnixBindFail", 0},
{"UnixActive", 0},
{"FdwatchBindFail", 0},
{"UDP4SendErr", 0},
{"RawRecvErr", 0},
{"TCP6Close", 0},
{"FDwatchRecvErr", 0},
{"TCP4BindFail", 0},
{"TCP4AcceptFail", 0},
{"TCP4OpenFail", 0},
{"UDP4Open", 92542},
{"UDP4ConnFail", 0},
{"TCP4Conn", 44},
{"TCP6ConnFail", 0},
{"FDwatchConn", 0},
{"UDP6Active", 0},
{"RawOpen", 1},
{"TCP6BindFail", 0},
{"UDP4Close", 92538},
{"TCP6Open", 0},
{"TCP6SendErr", 0},
{"TCP4Open", 48},
{"FDwatchSendErr", 0},
{"TCP6RecvErr", 0},
{"UDP4OpenFail", 0},
{"TCP4RecvErr", 0},
},
},
}
for _, tc := range testCases {
t.Run(tc.counterType, func(t *testing.T) {
tags := map[string]string{
"url": url,
"type": tc.counterType,
"source": host,
"port": port,
}
fields := map[string]interface{}{}
for _, val := range tc.values {
fields[val.fieldKey] = val.fieldValue
}
acc.AssertContainsTaggedFields(t, "bind_counter", fields, tags)
})
}
// Subtest for memory stats
t.Run("memory", func(t *testing.T) {
tags := map[string]string{
"url": url,
"source": host,
"port": port,
}
fields := map[string]interface{}{
"block_size": 45875200,
"context_size": 10037400,
"in_use": 6000232,
"lost": 0,
"total_use": 777821909,
}
acc.AssertContainsTaggedFields(t, "bind_memory", fields, tags)
})
// Subtest for per-context memory stats
t.Run("memory_context", func(t *testing.T) {
assert.True(t, acc.HasIntField("bind_memory_context", "total"))
assert.True(t, acc.HasIntField("bind_memory_context", "in_use"))
})
}
func TestBindUnparseableURL(t *testing.T) {
b := Bind{
Urls: []string{"://example.com"},
}
var acc testutil.Accumulator
err := acc.GatherError(b.Gather)
assert.Contains(t, err.Error(), "Unable to parse address")
}