2015-04-01 16:34:32 +00:00
|
|
|
package system
|
|
|
|
|
2015-04-06 16:32:10 +00:00
|
|
|
import "github.com/stretchr/testify/mock"
|
|
|
|
|
2015-05-22 23:45:14 +00:00
|
|
|
import "github.com/influxdb/telegraf/plugins/system/ps/cpu"
|
|
|
|
import "github.com/influxdb/telegraf/plugins/system/ps/disk"
|
2015-04-06 23:03:09 +00:00
|
|
|
|
2015-05-22 23:45:14 +00:00
|
|
|
import "github.com/influxdb/telegraf/plugins/system/ps/load"
|
|
|
|
import "github.com/influxdb/telegraf/plugins/system/ps/mem"
|
|
|
|
import "github.com/influxdb/telegraf/plugins/system/ps/net"
|
2015-04-01 16:34:32 +00:00
|
|
|
|
|
|
|
type MockPS struct {
|
|
|
|
mock.Mock
|
|
|
|
}
|
|
|
|
|
|
|
|
func (m *MockPS) LoadAvg() (*load.LoadAvgStat, error) {
|
|
|
|
ret := m.Called()
|
|
|
|
|
|
|
|
r0 := ret.Get(0).(*load.LoadAvgStat)
|
|
|
|
r1 := ret.Error(1)
|
|
|
|
|
|
|
|
return r0, r1
|
|
|
|
}
|
2015-08-13 17:27:24 +00:00
|
|
|
func (m *MockPS) CPUTimes(perCPU, totalCPU bool) ([]cpu.CPUTimesStat, error) {
|
2015-04-06 16:32:10 +00:00
|
|
|
ret := m.Called()
|
|
|
|
|
|
|
|
r0 := ret.Get(0).([]cpu.CPUTimesStat)
|
|
|
|
r1 := ret.Error(1)
|
|
|
|
|
|
|
|
return r0, r1
|
|
|
|
}
|
2015-04-06 17:34:55 +00:00
|
|
|
func (m *MockPS) DiskUsage() ([]*disk.DiskUsageStat, error) {
|
|
|
|
ret := m.Called()
|
|
|
|
|
|
|
|
r0 := ret.Get(0).([]*disk.DiskUsageStat)
|
|
|
|
r1 := ret.Error(1)
|
|
|
|
|
|
|
|
return r0, r1
|
|
|
|
}
|
2015-04-06 17:44:32 +00:00
|
|
|
func (m *MockPS) NetIO() ([]net.NetIOCountersStat, error) {
|
|
|
|
ret := m.Called()
|
|
|
|
|
|
|
|
r0 := ret.Get(0).([]net.NetIOCountersStat)
|
|
|
|
r1 := ret.Error(1)
|
|
|
|
|
|
|
|
return r0, r1
|
|
|
|
}
|
2015-04-06 17:59:05 +00:00
|
|
|
func (m *MockPS) DiskIO() (map[string]disk.DiskIOCountersStat, error) {
|
|
|
|
ret := m.Called()
|
|
|
|
|
|
|
|
r0 := ret.Get(0).(map[string]disk.DiskIOCountersStat)
|
|
|
|
r1 := ret.Error(1)
|
|
|
|
|
|
|
|
return r0, r1
|
|
|
|
}
|
2015-04-06 21:53:43 +00:00
|
|
|
func (m *MockPS) VMStat() (*mem.VirtualMemoryStat, error) {
|
|
|
|
ret := m.Called()
|
|
|
|
|
|
|
|
r0 := ret.Get(0).(*mem.VirtualMemoryStat)
|
|
|
|
r1 := ret.Error(1)
|
|
|
|
|
|
|
|
return r0, r1
|
|
|
|
}
|
|
|
|
func (m *MockPS) SwapStat() (*mem.SwapMemoryStat, error) {
|
|
|
|
ret := m.Called()
|
|
|
|
|
|
|
|
r0 := ret.Get(0).(*mem.SwapMemoryStat)
|
|
|
|
r1 := ret.Error(1)
|
|
|
|
|
|
|
|
return r0, r1
|
|
|
|
}
|
2015-04-06 23:03:09 +00:00
|
|
|
func (m *MockPS) DockerStat() ([]*DockerContainerStat, error) {
|
|
|
|
ret := m.Called()
|
|
|
|
|
|
|
|
r0 := ret.Get(0).([]*DockerContainerStat)
|
|
|
|
r1 := ret.Error(1)
|
|
|
|
|
|
|
|
return r0, r1
|
|
|
|
}
|