github.com/geofffranks/garden-linux@v0.0.0-20160715111146-26c893169cfa/linux_container/fake_quota_manager/fake_quota_manager.go (about) 1 // This file was generated by counterfeiter 2 package fake_quota_manager 3 4 import ( 5 "sync" 6 7 "code.cloudfoundry.org/garden" 8 "code.cloudfoundry.org/garden-linux/linux_container" 9 "code.cloudfoundry.org/lager" 10 ) 11 12 type FakeQuotaManager struct { 13 SetLimitsStub func(logger lager.Logger, containerRootFSPath string, limits garden.DiskLimits) error 14 setLimitsMutex sync.RWMutex 15 setLimitsArgsForCall []struct { 16 logger lager.Logger 17 containerRootFSPath string 18 limits garden.DiskLimits 19 } 20 setLimitsReturns struct { 21 result1 error 22 } 23 GetLimitsStub func(logger lager.Logger, containerRootFSPath string) (garden.DiskLimits, error) 24 getLimitsMutex sync.RWMutex 25 getLimitsArgsForCall []struct { 26 logger lager.Logger 27 containerRootFSPath string 28 } 29 getLimitsReturns struct { 30 result1 garden.DiskLimits 31 result2 error 32 } 33 GetUsageStub func(logger lager.Logger, containerRootFSPath string) (garden.ContainerDiskStat, error) 34 getUsageMutex sync.RWMutex 35 getUsageArgsForCall []struct { 36 logger lager.Logger 37 containerRootFSPath string 38 } 39 getUsageReturns struct { 40 result1 garden.ContainerDiskStat 41 result2 error 42 } 43 SetupStub func() error 44 setupMutex sync.RWMutex 45 setupArgsForCall []struct{} 46 setupReturns struct { 47 result1 error 48 } 49 } 50 51 func (fake *FakeQuotaManager) SetLimits(logger lager.Logger, containerRootFSPath string, limits garden.DiskLimits) error { 52 fake.setLimitsMutex.Lock() 53 fake.setLimitsArgsForCall = append(fake.setLimitsArgsForCall, struct { 54 logger lager.Logger 55 containerRootFSPath string 56 limits garden.DiskLimits 57 }{logger, containerRootFSPath, limits}) 58 fake.setLimitsMutex.Unlock() 59 if fake.SetLimitsStub != nil { 60 return fake.SetLimitsStub(logger, containerRootFSPath, limits) 61 } else { 62 return fake.setLimitsReturns.result1 63 } 64 } 65 66 func (fake *FakeQuotaManager) SetLimitsCallCount() int { 67 fake.setLimitsMutex.RLock() 68 defer fake.setLimitsMutex.RUnlock() 69 return len(fake.setLimitsArgsForCall) 70 } 71 72 func (fake *FakeQuotaManager) SetLimitsArgsForCall(i int) (lager.Logger, string, garden.DiskLimits) { 73 fake.setLimitsMutex.RLock() 74 defer fake.setLimitsMutex.RUnlock() 75 return fake.setLimitsArgsForCall[i].logger, fake.setLimitsArgsForCall[i].containerRootFSPath, fake.setLimitsArgsForCall[i].limits 76 } 77 78 func (fake *FakeQuotaManager) SetLimitsReturns(result1 error) { 79 fake.SetLimitsStub = nil 80 fake.setLimitsReturns = struct { 81 result1 error 82 }{result1} 83 } 84 85 func (fake *FakeQuotaManager) GetLimits(logger lager.Logger, containerRootFSPath string) (garden.DiskLimits, error) { 86 fake.getLimitsMutex.Lock() 87 fake.getLimitsArgsForCall = append(fake.getLimitsArgsForCall, struct { 88 logger lager.Logger 89 containerRootFSPath string 90 }{logger, containerRootFSPath}) 91 fake.getLimitsMutex.Unlock() 92 if fake.GetLimitsStub != nil { 93 return fake.GetLimitsStub(logger, containerRootFSPath) 94 } else { 95 return fake.getLimitsReturns.result1, fake.getLimitsReturns.result2 96 } 97 } 98 99 func (fake *FakeQuotaManager) GetLimitsCallCount() int { 100 fake.getLimitsMutex.RLock() 101 defer fake.getLimitsMutex.RUnlock() 102 return len(fake.getLimitsArgsForCall) 103 } 104 105 func (fake *FakeQuotaManager) GetLimitsArgsForCall(i int) (lager.Logger, string) { 106 fake.getLimitsMutex.RLock() 107 defer fake.getLimitsMutex.RUnlock() 108 return fake.getLimitsArgsForCall[i].logger, fake.getLimitsArgsForCall[i].containerRootFSPath 109 } 110 111 func (fake *FakeQuotaManager) GetLimitsReturns(result1 garden.DiskLimits, result2 error) { 112 fake.GetLimitsStub = nil 113 fake.getLimitsReturns = struct { 114 result1 garden.DiskLimits 115 result2 error 116 }{result1, result2} 117 } 118 119 func (fake *FakeQuotaManager) GetUsage(logger lager.Logger, containerRootFSPath string) (garden.ContainerDiskStat, error) { 120 fake.getUsageMutex.Lock() 121 fake.getUsageArgsForCall = append(fake.getUsageArgsForCall, struct { 122 logger lager.Logger 123 containerRootFSPath string 124 }{logger, containerRootFSPath}) 125 fake.getUsageMutex.Unlock() 126 if fake.GetUsageStub != nil { 127 return fake.GetUsageStub(logger, containerRootFSPath) 128 } else { 129 return fake.getUsageReturns.result1, fake.getUsageReturns.result2 130 } 131 } 132 133 func (fake *FakeQuotaManager) GetUsageCallCount() int { 134 fake.getUsageMutex.RLock() 135 defer fake.getUsageMutex.RUnlock() 136 return len(fake.getUsageArgsForCall) 137 } 138 139 func (fake *FakeQuotaManager) GetUsageArgsForCall(i int) (lager.Logger, string) { 140 fake.getUsageMutex.RLock() 141 defer fake.getUsageMutex.RUnlock() 142 return fake.getUsageArgsForCall[i].logger, fake.getUsageArgsForCall[i].containerRootFSPath 143 } 144 145 func (fake *FakeQuotaManager) GetUsageReturns(result1 garden.ContainerDiskStat, result2 error) { 146 fake.GetUsageStub = nil 147 fake.getUsageReturns = struct { 148 result1 garden.ContainerDiskStat 149 result2 error 150 }{result1, result2} 151 } 152 153 func (fake *FakeQuotaManager) Setup() error { 154 fake.setupMutex.Lock() 155 fake.setupArgsForCall = append(fake.setupArgsForCall, struct{}{}) 156 fake.setupMutex.Unlock() 157 if fake.SetupStub != nil { 158 return fake.SetupStub() 159 } else { 160 return fake.setupReturns.result1 161 } 162 } 163 164 func (fake *FakeQuotaManager) SetupCallCount() int { 165 fake.setupMutex.RLock() 166 defer fake.setupMutex.RUnlock() 167 return len(fake.setupArgsForCall) 168 } 169 170 func (fake *FakeQuotaManager) SetupReturns(result1 error) { 171 fake.SetupStub = nil 172 fake.setupReturns = struct { 173 result1 error 174 }{result1} 175 } 176 177 var _ linux_container.QuotaManager = new(FakeQuotaManager)