1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
# coding: utf-8
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
import datetime
import unittest
import azure.mgmt.consumption
import azure.mgmt.consumption.models
from devtools_testutils import AzureMgmtTestCase, ResourceGroupPreparer
class MgmtConsumptionTest(AzureMgmtTestCase):
def setUp(self):
super(MgmtConsumptionTest, self).setUp()
self.consumption_client = self.create_mgmt_client(
azure.mgmt.consumption.ConsumptionManagementClient
)
@ResourceGroupPreparer()
def test_budgets(self, resource_group):
SUBSCRIPTION_ID = getattr(self.settings, 'SUBSCRIPTION_ID', "123")
SCOPE = '/subscriptions/{}/resourceGroups/{}'.format(SUBSCRIPTION_ID, resource_group.name)
BUDGET_NAME = self.get_resource_name('budget')
# create
BODY = {
"category": "Cost",
"amount": '100',
"timeGrain": "Monthly",
"timePeriod": {
"startDate": "2020-10-01T00:00:00Z",
"endDate": "2021-10-31T00:00:00Z"
}
}
self.consumption_client.budgets.create_or_update(SCOPE, BUDGET_NAME, BODY)
# get
self.consumption_client.budgets.get(SCOPE, BUDGET_NAME)
# delete
self.consumption_client.budgets.delete(SCOPE, BUDGET_NAME)
# ------------------------------------------------------------------------------
if __name__ == '__main__':
unittest.main()
|
