fake_device_path_resolver.go

package fakes

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

type FakeDevicePathResolver struct {
	GetRealDevicePathDiskSettings boshsettings.DiskSettings
	RealDevicePath                string
	GetRealDevicePathStub         func(boshsettings.DiskSettings) (string, bool, error)
	GetRealDevicePathTimedOut     bool
	GetRealDevicePathErr          error
}

func NewFakeDevicePathResolver() *FakeDevicePathResolver {
	return &FakeDevicePathResolver{}
}

func (r *FakeDevicePathResolver) GetRealDevicePath(diskSettings boshsettings.DiskSettings) (string, bool, error) {
	r.GetRealDevicePathDiskSettings = diskSettings

	if r.GetRealDevicePathErr != nil {
		return "", r.GetRealDevicePathTimedOut, r.GetRealDevicePathErr
	}

	if r.GetRealDevicePathStub != nil {
		return r.GetRealDevicePathStub(diskSettings)
	}

	return r.RealDevicePath, false, nil
}