package system import "github.com/stretchr/testify/mock" import "github.com/influxdb/tivan/plugins/system/ps/cpu" import "github.com/influxdb/tivan/plugins/system/ps/load" 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 } func (m *MockPS) CPUTimes() ([]cpu.CPUTimesStat, error) { ret := m.Called() r0 := ret.Get(0).([]cpu.CPUTimesStat) r1 := ret.Error(1) return r0, r1 }