107 lines
2.2 KiB
Go
107 lines
2.2 KiB
Go
package system
|
|
|
|
import (
|
|
"syscall"
|
|
"testing"
|
|
|
|
"github.com/influxdb/telegraf/testutil"
|
|
"github.com/shirou/gopsutil/net"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestNetStats(t *testing.T) {
|
|
var mps MockPS
|
|
var err error
|
|
defer mps.AssertExpectations(t)
|
|
var acc testutil.Accumulator
|
|
|
|
netio := net.NetIOCountersStat{
|
|
Name: "eth0",
|
|
BytesSent: 1123,
|
|
BytesRecv: 8734422,
|
|
PacketsSent: 781,
|
|
PacketsRecv: 23456,
|
|
Errin: 832,
|
|
Errout: 8,
|
|
Dropin: 7,
|
|
Dropout: 1,
|
|
}
|
|
|
|
mps.On("NetIO").Return([]net.NetIOCountersStat{netio}, nil)
|
|
|
|
netprotos := []net.NetProtoCountersStat{
|
|
net.NetProtoCountersStat{
|
|
Protocol: "Udp",
|
|
Stats: map[string]int64{
|
|
"InDatagrams": 4655,
|
|
"NoPorts": 892592,
|
|
},
|
|
},
|
|
}
|
|
mps.On("NetProto").Return(netprotos, nil)
|
|
|
|
netstats := []net.NetConnectionStat{
|
|
net.NetConnectionStat{
|
|
Type: syscall.SOCK_DGRAM,
|
|
},
|
|
net.NetConnectionStat{
|
|
Status: "ESTABLISHED",
|
|
},
|
|
net.NetConnectionStat{
|
|
Status: "ESTABLISHED",
|
|
},
|
|
net.NetConnectionStat{
|
|
Status: "CLOSE",
|
|
},
|
|
}
|
|
|
|
mps.On("NetConnections").Return(netstats, nil)
|
|
|
|
err = (&NetIOStats{ps: &mps, skipChecks: true}).Gather(&acc)
|
|
require.NoError(t, err)
|
|
|
|
ntags := map[string]string{
|
|
"interface": "eth0",
|
|
}
|
|
|
|
fields1 := map[string]interface{}{
|
|
"bytes_sent": uint64(1123),
|
|
"bytes_recv": uint64(8734422),
|
|
"packets_sent": uint64(781),
|
|
"packets_recv": uint64(23456),
|
|
"err_in": uint64(832),
|
|
"err_out": uint64(8),
|
|
"drop_in": uint64(7),
|
|
"drop_out": uint64(1),
|
|
}
|
|
acc.AssertContainsFields(t, "net", fields1, ntags)
|
|
|
|
fields2 := map[string]interface{}{
|
|
"udp_noports": int64(892592),
|
|
"udp_indatagrams": int64(4655),
|
|
}
|
|
acc.AssertContainsFields(t, "net", fields2, nil)
|
|
|
|
acc.Points = nil
|
|
|
|
err = (&NetStats{&mps}).Gather(&acc)
|
|
require.NoError(t, err)
|
|
|
|
fields3 := map[string]interface{}{
|
|
"tcp_established": 2,
|
|
"tcp_syn_sent": 0,
|
|
"tcp_syn_recv": 0,
|
|
"tcp_fin_wait1": 0,
|
|
"tcp_fin_wait2": 0,
|
|
"tcp_time_wait": 0,
|
|
"tcp_close": 1,
|
|
"tcp_close_wait": 0,
|
|
"tcp_last_ack": 0,
|
|
"tcp_listen": 0,
|
|
"tcp_closing": 0,
|
|
"tcp_none": 0,
|
|
"udp_socket": 1,
|
|
}
|
|
acc.AssertContainsFields(t, "netstat", fields3, nil)
|
|
}
|