Function TestEnterpriseService_UpdateBudget

github/enterprise_budgets_test.go:220–273

(t *testing.T)

}

func TestEnterpriseService_UpdateBudget(t *testing.T) {
	t.Parallel()
	client, mux, _ := setup(t)

	req := EnterpriseUpdateBudget{
		BudgetAmount:        Ptr(10),
		PreventFurtherUsage: Ptr(false),
	}

	mux.HandleFunc("/enterprises/e/settings/billing/budgets/b-123", func(w http.ResponseWriter, r *http.Request) {
		testMethod(t, r, "PATCH")
		testJSONBody(t, r, req)
		fmt.Fprint(w, `{
			"message": "Budget successfully updated.",
			"budget": {
				"id": "b-123",
				"budget_amount": 10,
				"prevent_further_usage": false
			}
		}`)
	})

	ctx := t.Context()
	resp, _, err := client.Enterprise.UpdateBudget(ctx, "e", "b-123", req)
	if err != nil {
		t.Errorf("Enterprise.UpdateBudget returned error: %v", err)
	}

	want := &EnterpriseCreateOrUpdateBudgetResponse{
		Message: "Budget successfully updated.",
		Budget: &EnterpriseBudget{
			ID:                  Ptr("b-123"),
			BudgetAmount:        Ptr(10),
			PreventFurtherUsage: Ptr(false),
		},
	}
	if !cmp.Equal(resp, want) {
		t.Errorf("Enterprise.UpdateBudget returned %+v, want %+v", resp, want)
	}

	const methodName = "UpdateBudget"
	testNewRequestAndDoFailure(t, methodName, client, func() (*Response, error) {
		got, resp, err := client.Enterprise.UpdateBudget(ctx, "e", "b-123", req)
		if got != nil {
			t.Errorf("testNewRequestAndDoFailure %v = %#v, want nil", methodName, got)
		}
		return resp, err
	})

	testBadOptions(t, methodName, func() (err error) {
		_, _, err = client.Enterprise.UpdateBudget(ctx, "\n", "\n", req)
		return err
	})
}

func TestEnterpriseService_UpdateBudget_invalidEnterprise(t *testing.T) {
	t.Parallel()