fake_device_path_resolver.go

package fakes

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

type FakeDevicePathResolver struct {
	GetRealDevicePathDiskSettings boshsettings.DiskSettings
	RealDevicePath                string
	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
	}

	return r.RealDevicePath, false, nil
}