github.com/geofffranks/garden-linux@v0.0.0-20160715111146-26c893169cfa/sysinfo/fake_sysinfo/FakeProvider.go (about)

     1  // This file was generated by counterfeiter
     2  package fake_sysinfo
     3  
     4  import (
     5  	"sync"
     6  
     7  	"code.cloudfoundry.org/garden-linux/sysinfo"
     8  )
     9  
    10  type FakeProvider struct {
    11  	TotalMemoryStub        func() (uint64, error)
    12  	totalMemoryMutex       sync.RWMutex
    13  	totalMemoryArgsForCall []struct{}
    14  	totalMemoryReturns     struct {
    15  		result1 uint64
    16  		result2 error
    17  	}
    18  	TotalDiskStub        func() (uint64, error)
    19  	totalDiskMutex       sync.RWMutex
    20  	totalDiskArgsForCall []struct{}
    21  	totalDiskReturns     struct {
    22  		result1 uint64
    23  		result2 error
    24  	}
    25  }
    26  
    27  func (fake *FakeProvider) TotalMemory() (uint64, error) {
    28  	fake.totalMemoryMutex.Lock()
    29  	fake.totalMemoryArgsForCall = append(fake.totalMemoryArgsForCall, struct{}{})
    30  	fake.totalMemoryMutex.Unlock()
    31  	if fake.TotalMemoryStub != nil {
    32  		return fake.TotalMemoryStub()
    33  	} else {
    34  		return fake.totalMemoryReturns.result1, fake.totalMemoryReturns.result2
    35  	}
    36  }
    37  
    38  func (fake *FakeProvider) TotalMemoryCallCount() int {
    39  	fake.totalMemoryMutex.RLock()
    40  	defer fake.totalMemoryMutex.RUnlock()
    41  	return len(fake.totalMemoryArgsForCall)
    42  }
    43  
    44  func (fake *FakeProvider) TotalMemoryReturns(result1 uint64, result2 error) {
    45  	fake.TotalMemoryStub = nil
    46  	fake.totalMemoryReturns = struct {
    47  		result1 uint64
    48  		result2 error
    49  	}{result1, result2}
    50  }
    51  
    52  func (fake *FakeProvider) TotalDisk() (uint64, error) {
    53  	fake.totalDiskMutex.Lock()
    54  	fake.totalDiskArgsForCall = append(fake.totalDiskArgsForCall, struct{}{})
    55  	fake.totalDiskMutex.Unlock()
    56  	if fake.TotalDiskStub != nil {
    57  		return fake.TotalDiskStub()
    58  	} else {
    59  		return fake.totalDiskReturns.result1, fake.totalDiskReturns.result2
    60  	}
    61  }
    62  
    63  func (fake *FakeProvider) TotalDiskCallCount() int {
    64  	fake.totalDiskMutex.RLock()
    65  	defer fake.totalDiskMutex.RUnlock()
    66  	return len(fake.totalDiskArgsForCall)
    67  }
    68  
    69  func (fake *FakeProvider) TotalDiskReturns(result1 uint64, result2 error) {
    70  	fake.TotalDiskStub = nil
    71  	fake.totalDiskReturns = struct {
    72  		result1 uint64
    73  		result2 error
    74  	}{result1, result2}
    75  }
    76  
    77  var _ sysinfo.Provider = new(FakeProvider)