fake_fetcher.go

// This file was generated by counterfeiter
package fakes

import (
	"net/url"
	"sync"

	"code.cloudfoundry.org/garden-shed/layercake"
	"github.com/docker/docker/image"
)

type FakeFetcher struct {
	FetchStub        func(u *url.URL, diskQuota int64) (*image.Image, error)
	fetchMutex       sync.RWMutex
	fetchArgsForCall []struct {
		u         *url.URL
		diskQuota int64
	}
	fetchReturns struct {
		result1 *image.Image
		result2 error
	}
}

func (fake *FakeFetcher) Fetch(u *url.URL, diskQuota int64) (*image.Image, error) {
	fake.fetchMutex.Lock()
	fake.fetchArgsForCall = append(fake.fetchArgsForCall, struct {
		u         *url.URL
		diskQuota int64
	}{u, diskQuota})
	fake.fetchMutex.Unlock()
	if fake.FetchStub != nil {
		return fake.FetchStub(u, diskQuota)
	} else {
		return fake.fetchReturns.result1, fake.fetchReturns.result2
	}
}

func (fake *FakeFetcher) FetchCallCount() int {
	fake.fetchMutex.RLock()
	defer fake.fetchMutex.RUnlock()
	return len(fake.fetchArgsForCall)
}

func (fake *FakeFetcher) FetchArgsForCall(i int) (*url.URL, int64) {
	fake.fetchMutex.RLock()
	defer fake.fetchMutex.RUnlock()
	return fake.fetchArgsForCall[i].u, fake.fetchArgsForCall[i].diskQuota
}

func (fake *FakeFetcher) FetchReturns(result1 *image.Image, result2 error) {
	fake.FetchStub = nil
	fake.fetchReturns = struct {
		result1 *image.Image
		result2 error
	}{result1, result2}
}

var _ layercake.Fetcher = new(FakeFetcher)