Golang models.QuotaFields类代码示例

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()
}

		BeforeEach(func() {
			reqFactory.LoginSuccess = true
		})

		It("passes requirements when provided two args", func() {
			callSetQuota([]string{"my-org", "my-quota"}, reqFactory, quotaRepo)
			Expect(testcmd.CommandDidPassRequirements).To(BeTrue())
			Expect(reqFactory.OrganizationName).To(Equal("my-org"))
		})

		It("TestSetQuota", func() {
			org := models.Organization{}
			org.Name = "my-org"
			org.Guid = "my-org-guid"

			quota := models.QuotaFields{}
			quota.Name = "my-found-quota"
			quota.Guid = "my-quota-guid"

			quotaRepo.FindByNameQuota = quota
			reqFactory.Organization = org

			ui := callSetQuota([]string{"my-org", "my-quota"}, reqFactory, quotaRepo)

			testassert.SliceContains(ui.Outputs, testassert.Lines{
				{"Setting quota", "my-found-quota", "my-org", "my-user"},
				{"OK"},
			})

			Expect(quotaRepo.FindByNameName).To(Equal("my-quota"))
			Expect(quotaRepo.UpdateOrgGuid).To(Equal("my-org-guid"))

					Status: http.StatusOK,
					Body: `{"resources": [
							{
							  "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},

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"},
		})
	})
})