fake_windows_disk_partitioner.go

// Code generated by counterfeiter. DO NOT EDIT.
package fakes

import (
	"sync"

	"github.com/cloudfoundry/bosh-agent/platform/windows/disk"
)

type FakeWindowsDiskPartitioner struct {
	AssignDriveLetterStub        func(string, string) (string, error)
	assignDriveLetterMutex       sync.RWMutex
	assignDriveLetterArgsForCall []struct {
		arg1 string
		arg2 string
	}
	assignDriveLetterReturns struct {
		result1 string
		result2 error
	}
	assignDriveLetterReturnsOnCall map[int]struct {
		result1 string
		result2 error
	}
	GetCountOnDiskStub        func(string) (string, error)
	getCountOnDiskMutex       sync.RWMutex
	getCountOnDiskArgsForCall []struct {
		arg1 string
	}
	getCountOnDiskReturns struct {
		result1 string
		result2 error
	}
	getCountOnDiskReturnsOnCall map[int]struct {
		result1 string
		result2 error
	}
	GetFreeSpaceOnDiskStub        func(string) (int, error)
	getFreeSpaceOnDiskMutex       sync.RWMutex
	getFreeSpaceOnDiskArgsForCall []struct {
		arg1 string
	}
	getFreeSpaceOnDiskReturns struct {
		result1 int
		result2 error
	}
	getFreeSpaceOnDiskReturnsOnCall map[int]struct {
		result1 int
		result2 error
	}
	InitializeDiskStub        func(string) error
	initializeDiskMutex       sync.RWMutex
	initializeDiskArgsForCall []struct {
		arg1 string
	}
	initializeDiskReturns struct {
		result1 error
	}
	initializeDiskReturnsOnCall map[int]struct {
		result1 error
	}
	PartitionDiskStub        func(string) (string, error)
	partitionDiskMutex       sync.RWMutex
	partitionDiskArgsForCall []struct {
		arg1 string
	}
	partitionDiskReturns struct {
		result1 string
		result2 error
	}
	partitionDiskReturnsOnCall map[int]struct {
		result1 string
		result2 error
	}
	invocations      map[string][][]interface{}
	invocationsMutex sync.RWMutex
}

func (fake *FakeWindowsDiskPartitioner) AssignDriveLetter(arg1 string, arg2 string) (string, error) {
	fake.assignDriveLetterMutex.Lock()
	ret, specificReturn := fake.assignDriveLetterReturnsOnCall[len(fake.assignDriveLetterArgsForCall)]
	fake.assignDriveLetterArgsForCall = append(fake.assignDriveLetterArgsForCall, struct {
		arg1 string
		arg2 string
	}{arg1, arg2})
	fake.recordInvocation("AssignDriveLetter", []interface{}{arg1, arg2})
	fake.assignDriveLetterMutex.Unlock()
	if fake.AssignDriveLetterStub != nil {
		return fake.AssignDriveLetterStub(arg1, arg2)
	}
	if specificReturn {
		return ret.result1, ret.result2
	}
	fakeReturns := fake.assignDriveLetterReturns
	return fakeReturns.result1, fakeReturns.result2
}

func (fake *FakeWindowsDiskPartitioner) AssignDriveLetterCallCount() int {
	fake.assignDriveLetterMutex.RLock()
	defer fake.assignDriveLetterMutex.RUnlock()
	return len(fake.assignDriveLetterArgsForCall)
}

func (fake *FakeWindowsDiskPartitioner) AssignDriveLetterCalls(stub func(string, string) (string, error)) {
	fake.assignDriveLetterMutex.Lock()
	defer fake.assignDriveLetterMutex.Unlock()
	fake.AssignDriveLetterStub = stub
}

func (fake *FakeWindowsDiskPartitioner) AssignDriveLetterArgsForCall(i int) (string, string) {
	fake.assignDriveLetterMutex.RLock()
	defer fake.assignDriveLetterMutex.RUnlock()
	argsForCall := fake.assignDriveLetterArgsForCall[i]
	return argsForCall.arg1, argsForCall.arg2
}

func (fake *FakeWindowsDiskPartitioner) AssignDriveLetterReturns(result1 string, result2 error) {
	fake.assignDriveLetterMutex.Lock()
	defer fake.assignDriveLetterMutex.Unlock()
	fake.AssignDriveLetterStub = nil
	fake.assignDriveLetterReturns = struct {
		result1 string
		result2 error
	}{result1, result2}
}

func (fake *FakeWindowsDiskPartitioner) AssignDriveLetterReturnsOnCall(i int, result1 string, result2 error) {
	fake.assignDriveLetterMutex.Lock()
	defer fake.assignDriveLetterMutex.Unlock()
	fake.AssignDriveLetterStub = nil
	if fake.assignDriveLetterReturnsOnCall == nil {
		fake.assignDriveLetterReturnsOnCall = make(map[int]struct {
			result1 string
			result2 error
		})
	}
	fake.assignDriveLetterReturnsOnCall[i] = struct {
		result1 string
		result2 error
	}{result1, result2}
}

func (fake *FakeWindowsDiskPartitioner) GetCountOnDisk(arg1 string) (string, error) {
	fake.getCountOnDiskMutex.Lock()
	ret, specificReturn := fake.getCountOnDiskReturnsOnCall[len(fake.getCountOnDiskArgsForCall)]
	fake.getCountOnDiskArgsForCall = append(fake.getCountOnDiskArgsForCall, struct {
		arg1 string
	}{arg1})
	fake.recordInvocation("GetCountOnDisk", []interface{}{arg1})
	fake.getCountOnDiskMutex.Unlock()
	if fake.GetCountOnDiskStub != nil {
		return fake.GetCountOnDiskStub(arg1)
	}
	if specificReturn {
		return ret.result1, ret.result2
	}
	fakeReturns := fake.getCountOnDiskReturns
	return fakeReturns.result1, fakeReturns.result2
}

func (fake *FakeWindowsDiskPartitioner) GetCountOnDiskCallCount() int {
	fake.getCountOnDiskMutex.RLock()
	defer fake.getCountOnDiskMutex.RUnlock()
	return len(fake.getCountOnDiskArgsForCall)
}

func (fake *FakeWindowsDiskPartitioner) GetCountOnDiskCalls(stub func(string) (string, error)) {
	fake.getCountOnDiskMutex.Lock()
	defer fake.getCountOnDiskMutex.Unlock()
	fake.GetCountOnDiskStub = stub
}

func (fake *FakeWindowsDiskPartitioner) GetCountOnDiskArgsForCall(i int) string {
	fake.getCountOnDiskMutex.RLock()
	defer fake.getCountOnDiskMutex.RUnlock()
	argsForCall := fake.getCountOnDiskArgsForCall[i]
	return argsForCall.arg1
}

func (fake *FakeWindowsDiskPartitioner) GetCountOnDiskReturns(result1 string, result2 error) {
	fake.getCountOnDiskMutex.Lock()
	defer fake.getCountOnDiskMutex.Unlock()
	fake.GetCountOnDiskStub = nil
	fake.getCountOnDiskReturns = struct {
		result1 string
		result2 error
	}{result1, result2}
}

func (fake *FakeWindowsDiskPartitioner) GetCountOnDiskReturnsOnCall(i int, result1 string, result2 error) {
	fake.getCountOnDiskMutex.Lock()
	defer fake.getCountOnDiskMutex.Unlock()
	fake.GetCountOnDiskStub = nil
	if fake.getCountOnDiskReturnsOnCall == nil {
		fake.getCountOnDiskReturnsOnCall = make(map[int]struct {
			result1 string
			result2 error
		})
	}
	fake.getCountOnDiskReturnsOnCall[i] = struct {
		result1 string
		result2 error
	}{result1, result2}
}

func (fake *FakeWindowsDiskPartitioner) GetFreeSpaceOnDisk(arg1 string) (int, error) {
	fake.getFreeSpaceOnDiskMutex.Lock()
	ret, specificReturn := fake.getFreeSpaceOnDiskReturnsOnCall[len(fake.getFreeSpaceOnDiskArgsForCall)]
	fake.getFreeSpaceOnDiskArgsForCall = append(fake.getFreeSpaceOnDiskArgsForCall, struct {
		arg1 string
	}{arg1})
	fake.recordInvocation("GetFreeSpaceOnDisk", []interface{}{arg1})
	fake.getFreeSpaceOnDiskMutex.Unlock()
	if fake.GetFreeSpaceOnDiskStub != nil {
		return fake.GetFreeSpaceOnDiskStub(arg1)
	}
	if specificReturn {
		return ret.result1, ret.result2
	}
	fakeReturns := fake.getFreeSpaceOnDiskReturns
	return fakeReturns.result1, fakeReturns.result2
}

func (fake *FakeWindowsDiskPartitioner) GetFreeSpaceOnDiskCallCount() int {
	fake.getFreeSpaceOnDiskMutex.RLock()
	defer fake.getFreeSpaceOnDiskMutex.RUnlock()
	return len(fake.getFreeSpaceOnDiskArgsForCall)
}

func (fake *FakeWindowsDiskPartitioner) GetFreeSpaceOnDiskCalls(stub func(string) (int, error)) {
	fake.getFreeSpaceOnDiskMutex.Lock()
	defer fake.getFreeSpaceOnDiskMutex.Unlock()
	fake.GetFreeSpaceOnDiskStub = stub
}

func (fake *FakeWindowsDiskPartitioner) GetFreeSpaceOnDiskArgsForCall(i int) string {
	fake.getFreeSpaceOnDiskMutex.RLock()
	defer fake.getFreeSpaceOnDiskMutex.RUnlock()
	argsForCall := fake.getFreeSpaceOnDiskArgsForCall[i]
	return argsForCall.arg1
}

func (fake *FakeWindowsDiskPartitioner) GetFreeSpaceOnDiskReturns(result1 int, result2 error) {
	fake.getFreeSpaceOnDiskMutex.Lock()
	defer fake.getFreeSpaceOnDiskMutex.Unlock()
	fake.GetFreeSpaceOnDiskStub = nil
	fake.getFreeSpaceOnDiskReturns = struct {
		result1 int
		result2 error
	}{result1, result2}
}

func (fake *FakeWindowsDiskPartitioner) GetFreeSpaceOnDiskReturnsOnCall(i int, result1 int, result2 error) {
	fake.getFreeSpaceOnDiskMutex.Lock()
	defer fake.getFreeSpaceOnDiskMutex.Unlock()
	fake.GetFreeSpaceOnDiskStub = nil
	if fake.getFreeSpaceOnDiskReturnsOnCall == nil {
		fake.getFreeSpaceOnDiskReturnsOnCall = make(map[int]struct {
			result1 int
			result2 error
		})
	}
	fake.getFreeSpaceOnDiskReturnsOnCall[i] = struct {
		result1 int
		result2 error
	}{result1, result2}
}

func (fake *FakeWindowsDiskPartitioner) InitializeDisk(arg1 string) error {
	fake.initializeDiskMutex.Lock()
	ret, specificReturn := fake.initializeDiskReturnsOnCall[len(fake.initializeDiskArgsForCall)]
	fake.initializeDiskArgsForCall = append(fake.initializeDiskArgsForCall, struct {
		arg1 string
	}{arg1})
	fake.recordInvocation("InitializeDisk", []interface{}{arg1})
	fake.initializeDiskMutex.Unlock()
	if fake.InitializeDiskStub != nil {
		return fake.InitializeDiskStub(arg1)
	}
	if specificReturn {
		return ret.result1
	}
	fakeReturns := fake.initializeDiskReturns
	return fakeReturns.result1
}

func (fake *FakeWindowsDiskPartitioner) InitializeDiskCallCount() int {
	fake.initializeDiskMutex.RLock()
	defer fake.initializeDiskMutex.RUnlock()
	return len(fake.initializeDiskArgsForCall)
}

func (fake *FakeWindowsDiskPartitioner) InitializeDiskCalls(stub func(string) error) {
	fake.initializeDiskMutex.Lock()
	defer fake.initializeDiskMutex.Unlock()
	fake.InitializeDiskStub = stub
}

func (fake *FakeWindowsDiskPartitioner) InitializeDiskArgsForCall(i int) string {
	fake.initializeDiskMutex.RLock()
	defer fake.initializeDiskMutex.RUnlock()
	argsForCall := fake.initializeDiskArgsForCall[i]
	return argsForCall.arg1
}

func (fake *FakeWindowsDiskPartitioner) InitializeDiskReturns(result1 error) {
	fake.initializeDiskMutex.Lock()
	defer fake.initializeDiskMutex.Unlock()
	fake.InitializeDiskStub = nil
	fake.initializeDiskReturns = struct {
		result1 error
	}{result1}
}

func (fake *FakeWindowsDiskPartitioner) InitializeDiskReturnsOnCall(i int, result1 error) {
	fake.initializeDiskMutex.Lock()
	defer fake.initializeDiskMutex.Unlock()
	fake.InitializeDiskStub = nil
	if fake.initializeDiskReturnsOnCall == nil {
		fake.initializeDiskReturnsOnCall = make(map[int]struct {
			result1 error
		})
	}
	fake.initializeDiskReturnsOnCall[i] = struct {
		result1 error
	}{result1}
}

func (fake *FakeWindowsDiskPartitioner) PartitionDisk(arg1 string) (string, error) {
	fake.partitionDiskMutex.Lock()
	ret, specificReturn := fake.partitionDiskReturnsOnCall[len(fake.partitionDiskArgsForCall)]
	fake.partitionDiskArgsForCall = append(fake.partitionDiskArgsForCall, struct {
		arg1 string
	}{arg1})
	fake.recordInvocation("PartitionDisk", []interface{}{arg1})
	fake.partitionDiskMutex.Unlock()
	if fake.PartitionDiskStub != nil {
		return fake.PartitionDiskStub(arg1)
	}
	if specificReturn {
		return ret.result1, ret.result2
	}
	fakeReturns := fake.partitionDiskReturns
	return fakeReturns.result1, fakeReturns.result2
}

func (fake *FakeWindowsDiskPartitioner) PartitionDiskCallCount() int {
	fake.partitionDiskMutex.RLock()
	defer fake.partitionDiskMutex.RUnlock()
	return len(fake.partitionDiskArgsForCall)
}

func (fake *FakeWindowsDiskPartitioner) PartitionDiskCalls(stub func(string) (string, error)) {
	fake.partitionDiskMutex.Lock()
	defer fake.partitionDiskMutex.Unlock()
	fake.PartitionDiskStub = stub
}

func (fake *FakeWindowsDiskPartitioner) PartitionDiskArgsForCall(i int) string {
	fake.partitionDiskMutex.RLock()
	defer fake.partitionDiskMutex.RUnlock()
	argsForCall := fake.partitionDiskArgsForCall[i]
	return argsForCall.arg1
}

func (fake *FakeWindowsDiskPartitioner) PartitionDiskReturns(result1 string, result2 error) {
	fake.partitionDiskMutex.Lock()
	defer fake.partitionDiskMutex.Unlock()
	fake.PartitionDiskStub = nil
	fake.partitionDiskReturns = struct {
		result1 string
		result2 error
	}{result1, result2}
}

func (fake *FakeWindowsDiskPartitioner) PartitionDiskReturnsOnCall(i int, result1 string, result2 error) {
	fake.partitionDiskMutex.Lock()
	defer fake.partitionDiskMutex.Unlock()
	fake.PartitionDiskStub = nil
	if fake.partitionDiskReturnsOnCall == nil {
		fake.partitionDiskReturnsOnCall = make(map[int]struct {
			result1 string
			result2 error
		})
	}
	fake.partitionDiskReturnsOnCall[i] = struct {
		result1 string
		result2 error
	}{result1, result2}
}

func (fake *FakeWindowsDiskPartitioner) Invocations() map[string][][]interface{} {
	fake.invocationsMutex.RLock()
	defer fake.invocationsMutex.RUnlock()
	fake.assignDriveLetterMutex.RLock()
	defer fake.assignDriveLetterMutex.RUnlock()
	fake.getCountOnDiskMutex.RLock()
	defer fake.getCountOnDiskMutex.RUnlock()
	fake.getFreeSpaceOnDiskMutex.RLock()
	defer fake.getFreeSpaceOnDiskMutex.RUnlock()
	fake.initializeDiskMutex.RLock()
	defer fake.initializeDiskMutex.RUnlock()
	fake.partitionDiskMutex.RLock()
	defer fake.partitionDiskMutex.RUnlock()
	copiedInvocations := map[string][][]interface{}{}
	for key, value := range fake.invocations {
		copiedInvocations[key] = value
	}
	return copiedInvocations
}

func (fake *FakeWindowsDiskPartitioner) recordInvocation(key string, args []interface{}) {
	fake.invocationsMutex.Lock()
	defer fake.invocationsMutex.Unlock()
	if fake.invocations == nil {
		fake.invocations = map[string][][]interface{}{}
	}
	if fake.invocations[key] == nil {
		fake.invocations[key] = [][]interface{}{}
	}
	fake.invocations[key] = append(fake.invocations[key], args)
}

var _ disk.WindowsDiskPartitioner = new(FakeWindowsDiskPartitioner)