power/go/decider/mock_decider.go

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)