}) Ω(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) { return warden.CPULimits{}, disaster } }) It("returns the error", func() { err := container.LimitCPU(warden.CPULimits{}) Ω(err).Should(Equal(disaster)) }) }) }) Describe("LimitDisk", func() { It("sends a limit bandwidth request", func() { err := container.LimitDisk(warden.DiskLimits{ ByteLimit: 1, })