fakeConnection.WhenLimitingBandwidth = func(handle string, limits warden.BandwidthLimits) (warden.BandwidthLimits, error) {
return warden.BandwidthLimits{}, disaster
}
})
It("returns the error", func() {
err := container.LimitBandwidth(warden.BandwidthLimits{})
Ω(err).Should(Equal(disaster))
})
})
})
Describe("LimitCPU", func() {
It("sends a limit cpu request", func() {
err := container.LimitCPU(warden.CPULimits{
LimitInShares: 1,
})
Ω(err).ShouldNot(HaveOccurred())
Ω(fakeConnection.LimitedCPU("some-handle")).Should(ContainElement(
warden.CPULimits{
LimitInShares: 1,
},
))
})
Context("when the request fails", func() {
disaster := errors.New("oh no!")
BeforeEach(func() {
fakeConnection.WhenLimitingCPU = func(handle string, limits warden.CPULimits) (warden.CPULimits, error) {