289 lines
8.2 KiB
Go
289 lines
8.2 KiB
Go
package powerdns
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"encoding/binary"
|
|
"fmt"
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/influxdata/telegraf/testutil"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type statServer struct{}
|
|
|
|
var metrics = "corrupt-packets=0,deferred-cache-inserts=0,deferred-cache-lookup=0," +
|
|
"dnsupdate-answers=0,dnsupdate-changes=0,dnsupdate-queries=0," +
|
|
"dnsupdate-refused=0,packetcache-hit=0,packetcache-miss=1,packetcache-size=0," +
|
|
"query-cache-hit=0,query-cache-miss=6,rd-queries=1,recursing-answers=0," +
|
|
"recursing-questions=0,recursion-unanswered=0,security-status=3," +
|
|
"servfail-packets=0,signatures=0,tcp-answers=0,tcp-queries=0," +
|
|
"timedout-packets=0,udp-answers=1,udp-answers-bytes=50,udp-do-queries=0," +
|
|
"udp-queries=0,udp4-answers=1,udp4-queries=1,udp6-answers=0,udp6-queries=0," +
|
|
"key-cache-size=0,latency=26,meta-cache-size=0,qsize-q=0," +
|
|
"signature-cache-size=0,sys-msec=2889,uptime=86317,user-msec=2167,"
|
|
|
|
// first metric has no "="
|
|
var corruptMetrics = "corrupt-packets--0,deferred-cache-inserts=0,deferred-cache-lookup=0," +
|
|
"dnsupdate-answers=0,dnsupdate-changes=0,dnsupdate-queries=0," +
|
|
"dnsupdate-refused=0,packetcache-hit=0,packetcache-miss=1,packetcache-size=0," +
|
|
"query-cache-hit=0,query-cache-miss=6,rd-queries=1,recursing-answers=0," +
|
|
"recursing-questions=0,recursion-unanswered=0,security-status=3," +
|
|
"servfail-packets=0,signatures=0,tcp-answers=0,tcp-queries=0," +
|
|
"timedout-packets=0,udp-answers=1,udp-answers-bytes=50,udp-do-queries=0," +
|
|
"udp-queries=0,udp4-answers=1,udp4-queries=1,udp6-answers=0,udp6-queries=0," +
|
|
"key-cache-size=0,latency=26,meta-cache-size=0,qsize-q=0," +
|
|
"signature-cache-size=0,sys-msec=2889,uptime=86317,user-msec=2167,"
|
|
|
|
// integer overflow
|
|
var intOverflowMetrics = "corrupt-packets=18446744073709550195,deferred-cache-inserts=0,deferred-cache-lookup=0," +
|
|
"dnsupdate-answers=0,dnsupdate-changes=0,dnsupdate-queries=0," +
|
|
"dnsupdate-refused=0,packetcache-hit=0,packetcache-miss=1,packetcache-size=0," +
|
|
"query-cache-hit=0,query-cache-miss=6,rd-queries=1,recursing-answers=0," +
|
|
"recursing-questions=0,recursion-unanswered=0,security-status=3," +
|
|
"servfail-packets=0,signatures=0,tcp-answers=0,tcp-queries=0," +
|
|
"timedout-packets=0,udp-answers=1,udp-answers-bytes=50,udp-do-queries=0," +
|
|
"udp-queries=0,udp4-answers=1,udp4-queries=1,udp6-answers=0,udp6-queries=0," +
|
|
"key-cache-size=0,latency=26,meta-cache-size=0,qsize-q=0," +
|
|
"signature-cache-size=0,sys-msec=2889,uptime=86317,user-msec=2167,"
|
|
|
|
func (s statServer) serverSocket(l net.Listener) {
|
|
|
|
for {
|
|
conn, err := l.Accept()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
go func(c net.Conn) {
|
|
buf := make([]byte, 1024)
|
|
n, _ := c.Read(buf)
|
|
|
|
data := buf[:n]
|
|
if string(data) == "show * \n" {
|
|
c.Write([]byte(metrics))
|
|
c.Close()
|
|
}
|
|
}(conn)
|
|
}
|
|
}
|
|
|
|
func TestMemcachedGeneratesMetrics(t *testing.T) {
|
|
// We create a fake server to return test data
|
|
var randomNumber int64
|
|
binary.Read(rand.Reader, binary.LittleEndian, &randomNumber)
|
|
socket, err := net.Listen("unix", fmt.Sprintf("/tmp/pdns%d.controlsocket", randomNumber))
|
|
if err != nil {
|
|
t.Fatal("Cannot initialize server on port ")
|
|
}
|
|
|
|
defer socket.Close()
|
|
|
|
s := statServer{}
|
|
go s.serverSocket(socket)
|
|
|
|
p := &Powerdns{
|
|
UnixSockets: []string{fmt.Sprintf("/tmp/pdns%d.controlsocket", randomNumber)},
|
|
}
|
|
|
|
var acc testutil.Accumulator
|
|
|
|
err = acc.GatherError(p.Gather)
|
|
require.NoError(t, err)
|
|
|
|
intMetrics := []string{"corrupt-packets", "deferred-cache-inserts",
|
|
"deferred-cache-lookup", "dnsupdate-answers", "dnsupdate-changes",
|
|
"dnsupdate-queries", "dnsupdate-refused", "packetcache-hit",
|
|
"packetcache-miss", "packetcache-size", "query-cache-hit", "query-cache-miss",
|
|
"rd-queries", "recursing-answers", "recursing-questions",
|
|
"recursion-unanswered", "security-status", "servfail-packets", "signatures",
|
|
"tcp-answers", "tcp-queries", "timedout-packets", "udp-answers",
|
|
"udp-answers-bytes", "udp-do-queries", "udp-queries", "udp4-answers",
|
|
"udp4-queries", "udp6-answers", "udp6-queries", "key-cache-size", "latency",
|
|
"meta-cache-size", "qsize-q", "signature-cache-size", "sys-msec", "uptime", "user-msec"}
|
|
|
|
for _, metric := range intMetrics {
|
|
assert.True(t, acc.HasInt64Field("powerdns", metric), metric)
|
|
}
|
|
}
|
|
|
|
func TestPowerdnsParseMetrics(t *testing.T) {
|
|
values := parseResponse(metrics)
|
|
|
|
tests := []struct {
|
|
key string
|
|
value int64
|
|
}{
|
|
{"corrupt-packets", 0},
|
|
{"deferred-cache-inserts", 0},
|
|
{"deferred-cache-lookup", 0},
|
|
{"dnsupdate-answers", 0},
|
|
{"dnsupdate-changes", 0},
|
|
{"dnsupdate-queries", 0},
|
|
{"dnsupdate-refused", 0},
|
|
{"packetcache-hit", 0},
|
|
{"packetcache-miss", 1},
|
|
{"packetcache-size", 0},
|
|
{"query-cache-hit", 0},
|
|
{"query-cache-miss", 6},
|
|
{"rd-queries", 1},
|
|
{"recursing-answers", 0},
|
|
{"recursing-questions", 0},
|
|
{"recursion-unanswered", 0},
|
|
{"security-status", 3},
|
|
{"servfail-packets", 0},
|
|
{"signatures", 0},
|
|
{"tcp-answers", 0},
|
|
{"tcp-queries", 0},
|
|
{"timedout-packets", 0},
|
|
{"udp-answers", 1},
|
|
{"udp-answers-bytes", 50},
|
|
{"udp-do-queries", 0},
|
|
{"udp-queries", 0},
|
|
{"udp4-answers", 1},
|
|
{"udp4-queries", 1},
|
|
{"udp6-answers", 0},
|
|
{"udp6-queries", 0},
|
|
{"key-cache-size", 0},
|
|
{"latency", 26},
|
|
{"meta-cache-size", 0},
|
|
{"qsize-q", 0},
|
|
{"signature-cache-size", 0},
|
|
{"sys-msec", 2889},
|
|
{"uptime", 86317},
|
|
{"user-msec", 2167},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
value, ok := values[test.key]
|
|
if !ok {
|
|
t.Errorf("Did not find key for metric %s in values", test.key)
|
|
continue
|
|
}
|
|
if value != test.value {
|
|
t.Errorf("Metric: %s, Expected: %d, actual: %d",
|
|
test.key, test.value, value)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPowerdnsParseCorruptMetrics(t *testing.T) {
|
|
values := parseResponse(corruptMetrics)
|
|
|
|
tests := []struct {
|
|
key string
|
|
value int64
|
|
}{
|
|
{"deferred-cache-inserts", 0},
|
|
{"deferred-cache-lookup", 0},
|
|
{"dnsupdate-answers", 0},
|
|
{"dnsupdate-changes", 0},
|
|
{"dnsupdate-queries", 0},
|
|
{"dnsupdate-refused", 0},
|
|
{"packetcache-hit", 0},
|
|
{"packetcache-miss", 1},
|
|
{"packetcache-size", 0},
|
|
{"query-cache-hit", 0},
|
|
{"query-cache-miss", 6},
|
|
{"rd-queries", 1},
|
|
{"recursing-answers", 0},
|
|
{"recursing-questions", 0},
|
|
{"recursion-unanswered", 0},
|
|
{"security-status", 3},
|
|
{"servfail-packets", 0},
|
|
{"signatures", 0},
|
|
{"tcp-answers", 0},
|
|
{"tcp-queries", 0},
|
|
{"timedout-packets", 0},
|
|
{"udp-answers", 1},
|
|
{"udp-answers-bytes", 50},
|
|
{"udp-do-queries", 0},
|
|
{"udp-queries", 0},
|
|
{"udp4-answers", 1},
|
|
{"udp4-queries", 1},
|
|
{"udp6-answers", 0},
|
|
{"udp6-queries", 0},
|
|
{"key-cache-size", 0},
|
|
{"latency", 26},
|
|
{"meta-cache-size", 0},
|
|
{"qsize-q", 0},
|
|
{"signature-cache-size", 0},
|
|
{"sys-msec", 2889},
|
|
{"uptime", 86317},
|
|
{"user-msec", 2167},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
value, ok := values[test.key]
|
|
if !ok {
|
|
t.Errorf("Did not find key for metric %s in values", test.key)
|
|
continue
|
|
}
|
|
if value != test.value {
|
|
t.Errorf("Metric: %s, Expected: %d, actual: %d",
|
|
test.key, test.value, value)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPowerdnsParseIntOverflowMetrics(t *testing.T) {
|
|
values := parseResponse(intOverflowMetrics)
|
|
|
|
tests := []struct {
|
|
key string
|
|
value int64
|
|
}{
|
|
{"deferred-cache-inserts", 0},
|
|
{"deferred-cache-lookup", 0},
|
|
{"dnsupdate-answers", 0},
|
|
{"dnsupdate-changes", 0},
|
|
{"dnsupdate-queries", 0},
|
|
{"dnsupdate-refused", 0},
|
|
{"packetcache-hit", 0},
|
|
{"packetcache-miss", 1},
|
|
{"packetcache-size", 0},
|
|
{"query-cache-hit", 0},
|
|
{"query-cache-miss", 6},
|
|
{"rd-queries", 1},
|
|
{"recursing-answers", 0},
|
|
{"recursing-questions", 0},
|
|
{"recursion-unanswered", 0},
|
|
{"security-status", 3},
|
|
{"servfail-packets", 0},
|
|
{"signatures", 0},
|
|
{"tcp-answers", 0},
|
|
{"tcp-queries", 0},
|
|
{"timedout-packets", 0},
|
|
{"udp-answers", 1},
|
|
{"udp-answers-bytes", 50},
|
|
{"udp-do-queries", 0},
|
|
{"udp-queries", 0},
|
|
{"udp4-answers", 1},
|
|
{"udp4-queries", 1},
|
|
{"udp6-answers", 0},
|
|
{"udp6-queries", 0},
|
|
{"key-cache-size", 0},
|
|
{"latency", 26},
|
|
{"meta-cache-size", 0},
|
|
{"qsize-q", 0},
|
|
{"signature-cache-size", 0},
|
|
{"sys-msec", 2889},
|
|
{"uptime", 86317},
|
|
{"user-msec", 2167},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
value, ok := values[test.key]
|
|
if !ok {
|
|
t.Errorf("Did not find key for metric %s in values", test.key)
|
|
continue
|
|
}
|
|
if value != test.value {
|
|
t.Errorf("Metric: %s, Expected: %d, actual: %d",
|
|
test.key, test.value, value)
|
|
}
|
|
}
|
|
}
|