public void BudgetCreateOrUpdateTest()
{
using (MockContext context = MockContext.Start(this.GetType().FullName))
{
var consumptionMgmtClient = ConsumptionTestUtilities.GetConsumptionManagementClient(
context, new RecordedDelegatingHandler {
StatusCodeToReturn = HttpStatusCode.OK
});
consumptionMgmtClient.SubscriptionId = subscriptionId;
var timePeriod = new BudgetTimePeriod
{
StartDate = new DateTime(2018, 2, 1),
EndDate = new DateTime(2018, 11, 1),
};
var budget = new Budget("Cost", 60, "Monthly", timePeriod);
var budgetResponse = consumptionMgmtClient.Budgets.CreateOrUpdate(budgetName, budget);
ValidateProperties(budgetResponse);
Assert.Equal(budget.Amount, budgetResponse.Amount);
Assert.Equal(budget.Category, budgetResponse.Category);
Assert.Equal(budget.TimeGrain, budgetResponse.TimeGrain);
Assert.Equal(budget.TimePeriod.StartDate, budgetResponse.TimePeriod.StartDate);
Assert.Equal(budget.TimePeriod.EndDate, budgetResponse.TimePeriod.EndDate);
}
}
private BudgetTimePeriod CreateBudgetTimePeriod()
{
var timePeriod = new BudgetTimePeriod();
if (this.StartDate != null)
{
timePeriod.StartDate = this.StartDate.Value;
}
if (this.EndDate != null)
{
timePeriod.EndDate = this.EndDate.Value;
}
return(timePeriod);
}
