func fetchCapacity(logger lager.Logger, gardenClient GardenClient.Client, config Configuration) executor.ExecutorResources {
capacity, err := configuration.ConfigureCapacity(gardenClient, config.MemoryMB, config.DiskMB)
if err != nil {
logger.Error("failed-to-configure-capacity", err)
os.Exit(1)
}
logger.Info("initial-capacity", lager.Data{
"capacity": capacity,
})
return capacity
}
var _ = Describe("configuration", func() {
var gardenClient *fakes.FakeGardenClient
BeforeEach(func() {
gardenClient = fakes.NewGardenClient()
})
Describe("ConfigureCapacity", func() {
var capacity executor.ExecutorResources
var err error
var memLimit string
var diskLimit string
JustBeforeEach(func() {
capacity, err = configuration.ConfigureCapacity(gardenClient, memLimit, diskLimit)
})
Context("when getting the capacity fails", func() {
BeforeEach(func() {
gardenClient.Connection.CapacityReturns(garden.Capacity{}, errors.New("uh oh"))
})
It("returns an error", func() {
Expect(err).To(Equal(errors.New("uh oh")))
})
})
Context("when getting the capacity succeeds", func() {
BeforeEach(func() {
memLimit = "99"