aws_update_lbs.go

package fakes

import (
	"github.com/cloudfoundry/bosh-bootloader/commands"
	"github.com/cloudfoundry/bosh-bootloader/storage"
)

type AWSUpdateLBs struct {
	Name        string
	ExecuteCall struct {
		CallCount int
		Receives  struct {
			Config commands.AWSCreateLBsConfig
			State  storage.State
		}
		Returns struct {
			Error error
		}
	}
}

func (u *AWSUpdateLBs) Execute(config commands.AWSCreateLBsConfig, state storage.State) error {
	u.ExecuteCall.CallCount++
	u.ExecuteCall.Receives.Config = config
	u.ExecuteCall.Receives.State = state
	return u.ExecuteCall.Returns.Error
}