Ω(fakeContainer.LimitMemoryArgsForCall(0)).Should(Equal(setLimits))
})
itResetsGraceTimeWhenHandling(func() {
err := container.LimitMemory(setLimits)
Ω(err).ShouldNot(HaveOccurred())
})
itFailsWhenTheContainerIsNotFound(func() error {
return container.LimitMemory(garden.MemoryLimits{123})
})
Context("when limiting the memory fails", func() {
BeforeEach(func() {
fakeContainer.LimitMemoryReturns(errors.New("oh no!"))
})
It("fail", func() {
err := container.LimitMemory(garden.MemoryLimits{123})
Ω(err).Should(HaveOccurred())
})
})
})
Describe("getting memory limits", func() {
It("obtains the current limits", func() {
effectiveLimits := garden.MemoryLimits{2048}
fakeContainer.CurrentMemoryLimitsReturns(effectiveLimits, nil)
limits, err := container.CurrentMemoryLimits()