telegraf/plugins/inputs/system/mock_PS.go

99 lines
1.8 KiB
Go
Raw Permalink Normal View History

package system
import (
"github.com/stretchr/testify/mock"
"github.com/shirou/gopsutil/cpu"
"github.com/shirou/gopsutil/disk"
2015-04-06 23:03:09 +00:00
"github.com/shirou/gopsutil/load"
"github.com/shirou/gopsutil/mem"
"github.com/shirou/gopsutil/net"
)
type MockPS struct {
mock.Mock
}
func (m *MockPS) LoadAvg() (*load.AvgStat, error) {
ret := m.Called()
r0 := ret.Get(0).(*load.AvgStat)
r1 := ret.Error(1)
return r0, r1
}
func (m *MockPS) CPUTimes(perCPU, totalCPU bool) ([]cpu.TimesStat, error) {
ret := m.Called()
r0 := ret.Get(0).([]cpu.TimesStat)
r1 := ret.Error(1)
return r0, r1
}
func (m *MockPS) DiskUsage(mountPointFilter []string, fstypeExclude []string) ([]*disk.UsageStat, []*disk.PartitionStat, error) {
ret := m.Called(mountPointFilter, fstypeExclude)
2015-04-06 17:34:55 +00:00
r0 := ret.Get(0).([]*disk.UsageStat)
r1 := ret.Get(1).([]*disk.PartitionStat)
r2 := ret.Error(2)
2015-04-06 17:34:55 +00:00
return r0, r1, r2
2015-04-06 17:34:55 +00:00
}
func (m *MockPS) NetIO() ([]net.IOCountersStat, error) {
2015-04-06 17:44:32 +00:00
ret := m.Called()
r0 := ret.Get(0).([]net.IOCountersStat)
2015-04-06 17:44:32 +00:00
r1 := ret.Error(1)
return r0, r1
}
func (m *MockPS) NetProto() ([]net.ProtoCountersStat, error) {
ret := m.Called()
r0 := ret.Get(0).([]net.ProtoCountersStat)
r1 := ret.Error(1)
return r0, r1
}
func (m *MockPS) DiskIO() (map[string]disk.IOCountersStat, error) {
2015-04-06 17:59:05 +00:00
ret := m.Called()
r0 := ret.Get(0).(map[string]disk.IOCountersStat)
2015-04-06 17:59:05 +00:00
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
}
2015-04-06 21:53:43 +00:00
func (m *MockPS) SwapStat() (*mem.SwapMemoryStat, error) {
ret := m.Called()
r0 := ret.Get(0).(*mem.SwapMemoryStat)
r1 := ret.Error(1)
return r0, r1
}
func (m *MockPS) NetConnections() ([]net.ConnectionStat, error) {
2015-10-08 13:55:34 +00:00
ret := m.Called()
r0 := ret.Get(0).([]net.ConnectionStat)
2015-10-08 13:55:34 +00:00
r1 := ret.Error(1)
return r0, r1
}