packer-cn/builder/vsphere/driver/datastore_mock.go

package driver

import (
	"github.com/vmware/govmomi/vim25/mo"
	"github.com/vmware/govmomi/vim25/types"
)

type DatastoreMock struct {
	FileExistsCalled bool
	FileExistsReturn bool

	DirExistsCalled bool
	DirExistsReturn bool

	NameReturn string

	MakeDirectoryCalled bool

	ResolvePathCalled bool
	ResolvePathReturn string

	DeleteCalled bool
	DeletePath   string
	DeleteErr    error

	UploadFileCalled  bool
	UploadFileSrc     string
	UploadFileDst     string
	UploadFileHost    string
	UploadFileSetHost bool
	UploadFileErr     error
}

func (ds *DatastoreMock) Info(params ...string) (*mo.Datastore, error) {
	return nil, nil
}

func (ds *DatastoreMock) FileExists(path string) bool {
	ds.FileExistsCalled = true
	return ds.FileExistsReturn
}

func (ds *DatastoreMock) DirExists(path string) bool {
	ds.DirExistsCalled = true
	return ds.DirExistsReturn
}
func (ds *DatastoreMock) Name() string {
	if ds.NameReturn == "" {
		return "datastore-mock"
	}
	return ds.NameReturn
}

func (ds *DatastoreMock) Reference() types.ManagedObjectReference {
	return types.ManagedObjectReference{}
}

func (ds *DatastoreMock) ResolvePath(path string) string {
	ds.ResolvePathCalled = true
	return ds.ResolvePathReturn
}

func (ds *DatastoreMock) UploadFile(src, dst, host string, setHost bool) error {
	ds.UploadFileCalled = true
	ds.UploadFileSrc = src
	ds.UploadFileDst = dst
	ds.UploadFileHost = host
	ds.UploadFileSetHost = setHost
	return ds.UploadFileErr
}

func (ds *DatastoreMock) Delete(path string) error {
	ds.DeleteCalled = true
	ds.DeletePath = path
	return ds.DeleteErr
}

func (ds *DatastoreMock) MakeDirectory(path string) error {
	ds.MakeDirectoryCalled = true
	return nil
}