228 lines
8.2 KiB
Go
228 lines
8.2 KiB
Go
package dovecot
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"io"
|
|
"net"
|
|
"net/textproto"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/docker/go-connections/nat"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/testcontainers/testcontainers-go/wait"
|
|
|
|
"github.com/influxdata/telegraf/testutil"
|
|
)
|
|
|
|
func TestDovecotIntegration(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("Skipping integration test in short mode")
|
|
}
|
|
|
|
fields := map[string]interface{}{
|
|
"reset_timestamp": time.Unix(1453969886, 0),
|
|
"last_update": time.Unix(1454603963, 39864),
|
|
"num_logins": int64(7503897),
|
|
"num_cmds": int64(52595715),
|
|
"num_connected_sessions": int64(1204),
|
|
"user_cpu": 1.00831175372e+08,
|
|
"sys_cpu": 8.3849071112e+07,
|
|
"clock_time": 4.3260019315281835e+15,
|
|
"min_faults": int64(763950011),
|
|
"maj_faults": int64(1112443),
|
|
"vol_cs": int64(4120386897),
|
|
"invol_cs": int64(3685239306),
|
|
"disk_input": int64(41679480946688),
|
|
"disk_output": int64(1819070669176832),
|
|
"read_count": int64(2368906465),
|
|
"read_bytes": int64(2957928122981169),
|
|
"write_count": int64(3545389615),
|
|
"write_bytes": int64(1666822498251286),
|
|
"mail_lookup_path": int64(24396105),
|
|
"mail_lookup_attr": int64(302845),
|
|
"mail_read_count": int64(20155768),
|
|
"mail_read_bytes": int64(669946617705),
|
|
"mail_cache_hits": int64(1557255080),
|
|
}
|
|
|
|
var acc testutil.Accumulator
|
|
|
|
// Test type=global server=unix
|
|
addr := "/tmp/socket"
|
|
waitCh := make(chan int)
|
|
go func() {
|
|
defer close(waitCh)
|
|
|
|
la, err := net.ResolveUnixAddr("unix", addr)
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
|
|
l, err := net.ListenUnix("unix", la)
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
defer l.Close()
|
|
defer os.Remove(addr)
|
|
|
|
waitCh <- 0
|
|
conn, err := l.Accept()
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
defer conn.Close()
|
|
|
|
readertp := textproto.NewReader(bufio.NewReader(conn))
|
|
if _, err = readertp.ReadLine(); err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
|
|
buf := bytes.NewBufferString(sampleGlobal)
|
|
if _, err = io.Copy(conn, buf); err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
}()
|
|
|
|
// Wait for server to start
|
|
<-waitCh
|
|
|
|
d := &Dovecot{Servers: []string{addr}, Type: "global"}
|
|
err := d.Gather(&acc)
|
|
require.NoError(t, err)
|
|
|
|
tags := map[string]string{"server": addr, "type": "global"}
|
|
acc.AssertContainsTaggedFields(t, "dovecot", fields, tags)
|
|
|
|
// Test type=global
|
|
tags = map[string]string{"server": "dovecot.test", "type": "global"}
|
|
buf := bytes.NewBufferString(sampleGlobal)
|
|
|
|
gatherStats(buf, &acc, "dovecot.test", "global")
|
|
acc.AssertContainsTaggedFields(t, "dovecot", fields, tags)
|
|
|
|
// Test type=domain
|
|
tags = map[string]string{"server": "dovecot.test", "type": "domain", "domain": "domain.test"}
|
|
buf = bytes.NewBufferString(sampleDomain)
|
|
|
|
gatherStats(buf, &acc, "dovecot.test", "domain")
|
|
acc.AssertContainsTaggedFields(t, "dovecot", fields, tags)
|
|
|
|
// Test type=ip
|
|
tags = map[string]string{"server": "dovecot.test", "type": "ip", "ip": "192.168.0.100"}
|
|
buf = bytes.NewBufferString(sampleIP)
|
|
|
|
gatherStats(buf, &acc, "dovecot.test", "ip")
|
|
acc.AssertContainsTaggedFields(t, "dovecot", fields, tags)
|
|
|
|
// Test type=user
|
|
fields = map[string]interface{}{
|
|
"reset_timestamp": time.Unix(1453969886, 0),
|
|
"last_update": time.Unix(1454603963, 39864),
|
|
"num_logins": int64(7503897),
|
|
"num_cmds": int64(52595715),
|
|
"user_cpu": 1.00831175372e+08,
|
|
"sys_cpu": 8.3849071112e+07,
|
|
"clock_time": 4.3260019315281835e+15,
|
|
"min_faults": int64(763950011),
|
|
"maj_faults": int64(1112443),
|
|
"vol_cs": int64(4120386897),
|
|
"invol_cs": int64(3685239306),
|
|
"disk_input": int64(41679480946688),
|
|
"disk_output": int64(1819070669176832),
|
|
"read_count": int64(2368906465),
|
|
"read_bytes": int64(2957928122981169),
|
|
"write_count": int64(3545389615),
|
|
"write_bytes": int64(1666822498251286),
|
|
"mail_lookup_path": int64(24396105),
|
|
"mail_lookup_attr": int64(302845),
|
|
"mail_read_count": int64(20155768),
|
|
"mail_read_bytes": int64(669946617705),
|
|
"mail_cache_hits": int64(1557255080),
|
|
}
|
|
|
|
tags = map[string]string{"server": "dovecot.test", "type": "user", "user": "user.1@domain.test"}
|
|
buf = bytes.NewBufferString(sampleUser)
|
|
|
|
gatherStats(buf, &acc, "dovecot.test", "user")
|
|
acc.AssertContainsTaggedFields(t, "dovecot", fields, tags)
|
|
}
|
|
|
|
const sampleGlobal = `reset_timestamp last_update num_logins num_cmds num_connected_sessions user_cpu sys_cpu clock_time ` +
|
|
`min_faults maj_faults vol_cs invol_cs disk_input disk_output read_count read_bytes write_count write_bytes ` +
|
|
`mail_lookup_path mail_lookup_attr mail_read_count mail_read_bytes mail_cache_hits
|
|
1453969886 1454603963.039864 7503897 52595715 1204 100831175.372000 83849071.112000 4326001931528183.495762 ` +
|
|
`763950011 1112443 4120386897 3685239306 41679480946688 1819070669176832 2368906465 2957928122981169 ` +
|
|
`3545389615 1666822498251286 24396105 302845 20155768 669946617705 1557255080`
|
|
|
|
const sampleDomain = `domain reset_timestamp last_update num_logins num_cmds num_connected_sessions user_cpu ` +
|
|
`sys_cpu clock_time min_faults maj_faults vol_cs invol_cs disk_input disk_output read_count read_bytes ` +
|
|
`write_count write_bytes mail_lookup_path mail_lookup_attr mail_read_count mail_read_bytes mail_cache_hits
|
|
domain.test 1453969886 1454603963.039864 7503897 52595715 1204 100831175.372000 83849071.112000 ` +
|
|
`4326001931528183.495762 763950011 1112443 4120386897 3685239306 41679480946688 1819070669176832 ` +
|
|
`2368906465 2957928122981169 3545389615 1666822498251286 24396105 302845 20155768 669946617705 1557255080`
|
|
|
|
const sampleIP = `ip reset_timestamp last_update num_logins num_cmds num_connected_sessions user_cpu ` +
|
|
`sys_cpu clock_time min_faults maj_faults vol_cs invol_cs disk_input disk_output read_count ` +
|
|
`read_bytes write_count write_bytes mail_lookup_path mail_lookup_attr mail_read_count mail_read_bytes mail_cache_hits
|
|
192.168.0.100 1453969886 1454603963.039864 7503897 52595715 1204 100831175.372000 83849071.112000 ` +
|
|
`4326001931528183.495762 763950011 1112443 4120386897 3685239306 41679480946688 1819070669176832 ` +
|
|
`2368906465 2957928122981169 3545389615 1666822498251286 24396105 302845 20155768 669946617705 1557255080`
|
|
|
|
const sampleUser = `user reset_timestamp last_update num_logins num_cmds user_cpu sys_cpu clock_time ` +
|
|
`min_faults maj_faults vol_cs invol_cs disk_input disk_output read_count read_bytes write_count ` +
|
|
`write_bytes mail_lookup_path mail_lookup_attr mail_read_count mail_read_bytes mail_cache_hits
|
|
user.1@domain.test 1453969886 1454603963.039864 7503897 52595715 100831175.372000 83849071.112000 ` +
|
|
`4326001931528183.495762 763950011 1112443 4120386897 3685239306 41679480946688 1819070669176832 ` +
|
|
`2368906465 2957928122981169 3545389615 1666822498251286 24396105 302845 20155768 669946617705 1557255080`
|
|
|
|
func TestDovecotContainerIntegration(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("Skipping dovecot integration tests.")
|
|
}
|
|
|
|
testdata, err := filepath.Abs("testdata/dovecot.conf")
|
|
require.NoError(t, err, "determining absolute path of test-data failed")
|
|
|
|
servicePort := "24242"
|
|
container := testutil.Container{
|
|
Image: "dovecot/dovecot:2.3-latest",
|
|
ExposedPorts: []string{servicePort},
|
|
Files: map[string]string{
|
|
"/etc/dovecot/dovecot.conf": testdata,
|
|
},
|
|
WaitingFor: wait.ForAll(
|
|
wait.ForLog("starting up for imap"),
|
|
wait.ForListeningPort(nat.Port(servicePort)),
|
|
),
|
|
}
|
|
require.NoError(t, container.Start(), "failed to start container")
|
|
defer container.Terminate()
|
|
|
|
d := &Dovecot{
|
|
Servers: []string{container.Address + ":" + container.Ports[servicePort]},
|
|
Type: "global",
|
|
}
|
|
|
|
var acc testutil.Accumulator
|
|
require.NoError(t, d.Gather(&acc))
|
|
require.Eventually(t,
|
|
func() bool { return acc.HasMeasurement("dovecot") },
|
|
5*time.Second,
|
|
10*time.Millisecond,
|
|
)
|
|
|
|
require.True(t, acc.HasTag("dovecot", "type"))
|
|
require.True(t, acc.HasField("dovecot", "sys_cpu"))
|
|
require.True(t, acc.HasField("dovecot", "write_count"))
|
|
require.True(t, acc.HasField("dovecot", "mail_read_count"))
|
|
require.True(t, acc.HasField("dovecot", "auth_failures"))
|
|
}
|