It("returns the error", func() { err := container.LimitMemory(garden.MemoryLimits{}) Ω(err).Should(Equal(disaster)) }) }) }) Describe("CurrentBandwidthLimits", func() { It("sends an empty limit request and returns its response", func() { limitsToReturn := garden.BandwidthLimits{ RateInBytesPerSecond: 1, BurstRateInBytesPerSecond: 2, } fakeConnection.CurrentBandwidthLimitsReturns(limitsToReturn, nil) limits, err := container.CurrentBandwidthLimits() Ω(err).ShouldNot(HaveOccurred()) Ω(limits).Should(Equal(limitsToReturn)) }) Context("when the request fails", func() { disaster := errors.New("oh no!") BeforeEach(func() { fakeConnection.CurrentBandwidthLimitsReturns(garden.BandwidthLimits{}, disaster) }) It("returns the error", func() {
Describe("CurrentBandwidthLimits", func() { handle := "suitcase" limits := garden.BandwidthLimits{ RateInBytesPerSecond: 234, } var gotLimits garden.BandwidthLimits itRetries(func() error { var err error gotLimits, err = conn.CurrentBandwidthLimits(handle) return err }, func(err error) { innerConnection.CurrentBandwidthLimitsReturns(limits, err) }, func() int { return innerConnection.CurrentBandwidthLimitsCallCount() }, func() { It("calls through to garden", func() { Ω(innerConnection.CurrentBandwidthLimitsCallCount()).Should(Equal(1)) calledHandle := innerConnection.CurrentBandwidthLimitsArgsForCall(0) Ω(calledHandle).Should(Equal(handle)) }) It("returns the limits", func() { Ω(gotLimits).Should(Equal(limits)) }) }) })