fake_agent_client.go

package fakes

import (
	"github.com/cloudfoundry/bosh-agent/agentclient"
	"github.com/cloudfoundry/bosh-agent/agentclient/applyspec"
	"github.com/cloudfoundry/bosh-agent/settings"
)

type FakeAgentClient struct {
	PingResponses   []pingResponse
	PingCalledCount int

	StopCalled bool
	stopErr    error

	ApplyApplySpec applyspec.ApplySpec
	ApplyErr       error

	StartCalled bool
	startErr    error

	MountDiskCID string
	mountDiskErr error

	UnmountDiskCID string
	unmountDiskErr error

	listDiskDisks  []string
	listDiskErr    error
	ListDiskCalled bool

	GetStateCalledTimes int
	getStateOutputs     []getStateOutput

	MigrateDiskCalledTimes int
	migrateDiskErr         error

	UpdateSettingsCalledTimes int
	updateSettingsErr         error

	RunScriptCalledTimes int
	runScriptErr         error

	DeleteARPEntriesCalledTimes int
	deleteARPEntriesErr         error
}

type pingResponse struct {
	response string
	err      error
}

type getStateOutput struct {
	state agentclient.AgentState
	err   error
}

type compilePackageOutput struct {
	blobRef agentclient.BlobRef
	err     error
}

func NewFakeAgentClient() *FakeAgentClient {
	return &FakeAgentClient{
		getStateOutputs: []getStateOutput{},
	}
}

func (c *FakeAgentClient) Ping() (string, error) {
	c.PingCalledCount++

	if len(c.PingResponses) > 0 {
		response := c.PingResponses[0]
		c.PingResponses = c.PingResponses[1:]
		return response.response, response.err
	}

	return "", nil
}

func (c *FakeAgentClient) Stop() error {
	c.StopCalled = true
	return c.stopErr
}

func (c *FakeAgentClient) Apply(applySpec applyspec.ApplySpec) error {
	c.ApplyApplySpec = applySpec

	return c.ApplyErr
}

func (c *FakeAgentClient) Start() error {
	c.StartCalled = true
	return c.startErr
}

func (c *FakeAgentClient) GetState() (agentclient.AgentState, error) {
	c.GetStateCalledTimes++

	getStateReturn := c.getStateOutputs[0]
	c.getStateOutputs = c.getStateOutputs[1:]

	return getStateReturn.state, getStateReturn.err
}

func (c *FakeAgentClient) ListDisk() ([]string, error) {
	c.ListDiskCalled = true
	return c.listDiskDisks, c.listDiskErr
}

func (c *FakeAgentClient) MountDisk(diskCID string) error {
	c.MountDiskCID = diskCID

	return c.mountDiskErr
}

func (c *FakeAgentClient) UnmountDisk(diskCID string) error {
	c.UnmountDiskCID = diskCID
	return c.unmountDiskErr
}

func (c *FakeAgentClient) MigrateDisk() error {
	c.MigrateDiskCalledTimes++
	return c.migrateDiskErr
}

func (c *FakeAgentClient) UpdateSettings(settings settings.Settings) error {
	c.UpdateSettingsCalledTimes++
	return c.updateSettingsErr
}

func (c *FakeAgentClient) RunScript(scriptName string, options map[string]interface{}) error {
	c.RunScriptCalledTimes++
	return c.runScriptErr
}

func (c *FakeAgentClient) DeleteARPEntries(ips []string) error {
	c.DeleteARPEntriesCalledTimes++
	return c.deleteARPEntriesErr
}

func (c *FakeAgentClient) CompilePackage(
	packageSource agentclient.BlobRef,
	compiledPackageDependencies []agentclient.BlobRef,
) (
	compiledPackageRef agentclient.BlobRef,
	err error,
) {
	return agentclient.BlobRef{}, nil
}

func (c *FakeAgentClient) SetPingBehavior(response string, err error) {
	c.PingResponses = append(c.PingResponses, pingResponse{
		response: response,
		err:      err,
	})
}

func (c *FakeAgentClient) SetStopBehavior(err error) {
	c.stopErr = err
}

func (c *FakeAgentClient) SetStartBehavior(err error) {
	c.startErr = err
}

func (c *FakeAgentClient) SetGetStateBehavior(stateResponse agentclient.AgentState, err error) {
	c.getStateOutputs = append(c.getStateOutputs, getStateOutput{
		state: stateResponse,
		err:   err,
	})
}

func (c *FakeAgentClient) SetMountDiskBehavior(err error) {
	c.mountDiskErr = err
}

func (c *FakeAgentClient) SetUnmountDiskBehavior(err error) {
	c.unmountDiskErr = err
}

func (c *FakeAgentClient) SetMigrateDiskBehavior(err error) {
	c.migrateDiskErr = err
}

func (c *FakeAgentClient) SetUpdateSettingsBehavior(err error) {
	c.updateSettingsErr = err
}

func (c *FakeAgentClient) SetListDiskBehavior(disks []string, err error) {
	c.listDiskDisks = disks
	c.listDiskErr = err
}