func (cmd CreateQuota) Run(context *cli.Context) {
name := context.Args()[0]
cmd.ui.Say("Creating quota %s as %s...",
terminal.EntityNameColor(name),
terminal.EntityNameColor(cmd.config.Username()))
quota := models.QuotaFields{
Name: name,
}
memoryLimit := context.String("m")
if memoryLimit != "" {
parsedMemory, err := formatters.ToMegabytes(memoryLimit)
if err != nil {
cmd.ui.Failed("Invalid memory limit: %s\n%s", memoryLimit, err)
}
quota.MemoryLimit = parsedMemory
}
if context.IsSet("r") {
quota.RoutesLimit = context.Int("r")
}
if context.IsSet("s") {
quota.ServicesLimit = context.Int("s")
}
if context.IsSet("allow-paid-service-plans") {
quota.NonBasicServicesAllowed = true
}
err := cmd.quotaRepo.Create(quota)
httpErr, ok := err.(errors.HttpError)
if ok && httpErr.ErrorCode() == errors.QUOTA_EXISTS {
cmd.ui.Ok()
cmd.ui.Warn("Quota Definition %s already exists", quota.Name)
return
}
if err != nil {
cmd.ui.Failed(err.Error())
}
cmd.ui.Ok()
}
"metadata": { "guid": "my-quota-guid" },
"entity": { "name": "my-remote-quota", "memory_limit": 1024 }
}
]}`},
})
ts, handler, repo := createQuotaRepo(req)
defer ts.Close()
quota, apiResponse := repo.FindByName("my-quota")
Expect(handler.AllRequestsCalled()).To(BeTrue())
Expect(apiResponse.IsNotSuccessful()).To(BeFalse())
expectedQuota := models.QuotaFields{}
expectedQuota.Guid = "my-quota-guid"
expectedQuota.Name = "my-remote-quota"
expectedQuota.MemoryLimit = 1024
Expect(quota).To(Equal(expectedQuota))
})
})
Describe("Update", func() {
It("sets the quota for an organization", func() {
req := testapi.NewCloudControllerTestRequest(testnet.TestRequest{
Method: "PUT",
Path: "/v2/organizations/my-org-guid",
Matcher: testnet.RequestBodyMatcher(`{"quota_definition_guid":"my-quota-guid"}`),
Response: testnet.TestResponse{Status: http.StatusCreated},
})
ts, handler, repo := createQuotaRepo(req)
var _ = Describe("Testing with ginkgo", func() {
It("TestListQuotasRequirements", func() {
quotaRepo := &testapi.FakeQuotaRepository{}
reqFactory := &testreq.FakeReqFactory{LoginSuccess: true}
callListQuotas(reqFactory, quotaRepo)
Expect(testcmd.CommandDidPassRequirements).To(BeTrue())
reqFactory = &testreq.FakeReqFactory{LoginSuccess: false}
callListQuotas(reqFactory, quotaRepo)
Expect(testcmd.CommandDidPassRequirements).To(BeFalse())
})
It("TestListQuotas", func() {
quota := models.QuotaFields{}
quota.Name = "quota-name"
quota.MemoryLimit = 1024
quotaRepo := &testapi.FakeQuotaRepository{FindAllQuotas: []models.QuotaFields{quota}}
reqFactory := &testreq.FakeReqFactory{LoginSuccess: true}
ui := callListQuotas(reqFactory, quotaRepo)
testassert.SliceContains(ui.Outputs, testassert.Lines{
{"Getting quotas as", "my-user"},
{"OK"},
{"name", "memory limit"},
{"quota-name", "1g"},
})
})
})