| package decider |
| |
| import ( |
| "github.com/stretchr/testify/mock" |
| swarming "go.chromium.org/luci/common/api/swarming/swarming/v1" |
| ) |
| |
| type MockDecider struct { |
| mock.Mock |
| } |
| |
| // NewMockDecider returns a pointer to a newly created struct. We return the pointer because we want to |
| // make sure the methods on mock.Mock stay accessible, e.g. m.On() |
| func NewMockDecider() *MockDecider { |
| return &MockDecider{} |
| } |
| |
| func (m *MockDecider) ShouldPowercycleBot(bot *swarming.SwarmingRpcsBotInfo) bool { |
| args := m.Called(bot) |
| r0 := false |
| if rf, ok := args.Get(0).(func(*swarming.SwarmingRpcsBotInfo) bool); ok { |
| r0 = rf(bot) |
| } else { |
| r0 = args.Bool(0) |
| } |
| return r0 |
| } |
| |
| func (m *MockDecider) ShouldPowercycleDevice(bot *swarming.SwarmingRpcsBotInfo) bool { |
| args := m.Called(bot) |
| r0 := false |
| if rf, ok := args.Get(0).(func(*swarming.SwarmingRpcsBotInfo) bool); ok { |
| r0 = rf(bot) |
| } else { |
| r0 = args.Bool(0) |
| } |
| return r0 |
| } |
| |
| // Ensure MockDecider fulfills Decider |
| var _ Decider = (*MockDecider)(nil) |