File: test_cli_mgmt_servicebus_topic_test.py

package info (click to toggle)
python-azure 20250603%2Bgit-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 851,724 kB
  • sloc: python: 7,362,925; ansic: 804; javascript: 287; makefile: 195; sh: 145; xml: 109
file content (277 lines) | stat: -rw-r--r-- 12,764 bytes parent folder | download
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
# 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.
# --------------------------------------------------------------------------

# Current Operation Coverage:
#   Topics: 10/10
#   Subscriptions: 4/4
#   Rules: 4/4

import unittest
import pytest
import azure.mgmt.servicebus
from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy

AZURE_LOCATION = "eastus"


@pytest.mark.skip("denied due to internal policy")
@pytest.mark.live_test_only
class TestMgmtServiceBus(AzureMgmtRecordedTestCase):

    def setup_method(self, method):
        self.mgmt_client = self.create_mgmt_client(azure.mgmt.servicebus.ServiceBusManagementClient)

    @RandomNameResourceGroupPreparer(location=AZURE_LOCATION)
    @recorded_by_proxy
    def test_subscrpition_and_rule(self, resource_group):

        RESOURCE_GROUP = resource_group.name
        NAMESPACE_NAME = "myNamespacexxyyzzxyye"
        TOPIC_NAME = "myTopic"
        SUBSCRIPTION_NAME = "mySubscription"
        RULE_NAME = "myRule"

        # --------------------------------------------------------------------------
        # /Namespaces/put/NameSpaceCreate[put]
        # --------------------------------------------------------------------------
        BODY = {
            "sku": {"name": "Standard", "tier": "Standard"},
            "location": AZURE_LOCATION,
            "tags": {"tag1": "value1", "tag2": "value2"},
        }
        result = self.mgmt_client.namespaces.begin_create_or_update(
            resource_group_name=RESOURCE_GROUP, namespace_name=NAMESPACE_NAME, parameters=BODY
        )
        namespace = result.result()

        # --------------------------------------------------------------------------
        # /Topics/put/TopicCreate[put]
        # --------------------------------------------------------------------------
        BODY = {"enable_express": True}
        result = self.mgmt_client.topics.create_or_update(
            resource_group_name=RESOURCE_GROUP, namespace_name=NAMESPACE_NAME, topic_name=TOPIC_NAME, parameters=BODY
        )

        # --------------------------------------------------------------------------
        # /Subscriptions/put/SubscriptionCreate[put]
        # --------------------------------------------------------------------------
        BODY = {"enable_batched_operations": True}
        result = self.mgmt_client.subscriptions.create_or_update(
            resource_group_name=RESOURCE_GROUP,
            namespace_name=NAMESPACE_NAME,
            topic_name=TOPIC_NAME,
            subscription_name=SUBSCRIPTION_NAME,
            parameters=BODY,
        )

        # --------------------------------------------------------------------------
        # /Rules/put/RulesCreateOrUpdate[put]
        # --------------------------------------------------------------------------
        BODY = {}
        result = self.mgmt_client.rules.create_or_update(
            resource_group_name=RESOURCE_GROUP,
            namespace_name=NAMESPACE_NAME,
            topic_name=TOPIC_NAME,
            subscription_name=SUBSCRIPTION_NAME,
            rule_name=RULE_NAME,
            parameters=BODY,
        )

        # --------------------------------------------------------------------------
        # /Subscriptions/get/SubscriptionGet[get]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.subscriptions.get(
            resource_group_name=RESOURCE_GROUP,
            namespace_name=NAMESPACE_NAME,
            topic_name=TOPIC_NAME,
            subscription_name=SUBSCRIPTION_NAME,
        )

        # --------------------------------------------------------------------------
        # /Rules/get/RulesGet[get]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.rules.get(
            resource_group_name=RESOURCE_GROUP,
            namespace_name=NAMESPACE_NAME,
            topic_name=TOPIC_NAME,
            subscription_name=SUBSCRIPTION_NAME,
            rule_name=RULE_NAME,
        )

        # --------------------------------------------------------------------------
        # /Subscriptions/get/SubscriptionListByTopic[get]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.subscriptions.list_by_topic(
            resource_group_name=RESOURCE_GROUP, namespace_name=NAMESPACE_NAME, topic_name=TOPIC_NAME
        )

        # --------------------------------------------------------------------------
        # /Rules/get/RulesListBySubscriptions[get]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.rules.list_by_subscriptions(
            resource_group_name=RESOURCE_GROUP,
            namespace_name=NAMESPACE_NAME,
            topic_name=TOPIC_NAME,
            subscription_name=SUBSCRIPTION_NAME,
        )

        # --------------------------------------------------------------------------
        # /Rules/delete/RulesDelete[delete]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.rules.delete(
            resource_group_name=RESOURCE_GROUP,
            namespace_name=NAMESPACE_NAME,
            topic_name=TOPIC_NAME,
            subscription_name=SUBSCRIPTION_NAME,
            rule_name=RULE_NAME,
        )

        # --------------------------------------------------------------------------
        # /Subscriptions/delete/SubscriptionDelete[delete]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.subscriptions.delete(
            resource_group_name=RESOURCE_GROUP,
            namespace_name=NAMESPACE_NAME,
            topic_name=TOPIC_NAME,
            subscription_name=SUBSCRIPTION_NAME,
        )

        # --------------------------------------------------------------------------
        # /Topics/delete/TopicDelete[delete]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.topics.delete(
            resource_group_name=RESOURCE_GROUP, namespace_name=NAMESPACE_NAME, topic_name=TOPIC_NAME
        )

        # --------------------------------------------------------------------------
        # /Namespaces/delete/NameSpaceDelete[delete]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.namespaces.begin_delete(
            resource_group_name=RESOURCE_GROUP, namespace_name=NAMESPACE_NAME
        )
        result = result.result()

    @RandomNameResourceGroupPreparer(location=AZURE_LOCATION)
    @recorded_by_proxy
    def test_topic(self, resource_group):

        RESOURCE_GROUP = resource_group.name
        NAMESPACE_NAME = "myNamespacexxyyzzxyyf"
        AUTHORIZATION_RULE_NAME = "myAuthorizationRule"
        TOPIC_NAME = "myTopic"

        # --------------------------------------------------------------------------
        # /Namespaces/put/NameSpaceCreate[put]
        # --------------------------------------------------------------------------
        BODY = {
            "sku": {"name": "Standard", "tier": "Standard"},
            "location": AZURE_LOCATION,
            "tags": {"tag1": "value1", "tag2": "value2"},
        }
        result = self.mgmt_client.namespaces.begin_create_or_update(
            resource_group_name=RESOURCE_GROUP, namespace_name=NAMESPACE_NAME, parameters=BODY
        )
        namespace = result.result()

        # --------------------------------------------------------------------------
        # /Topics/put/TopicCreate[put]
        # --------------------------------------------------------------------------
        BODY = {"enable_express": True}
        result = self.mgmt_client.topics.create_or_update(
            resource_group_name=RESOURCE_GROUP, namespace_name=NAMESPACE_NAME, topic_name=TOPIC_NAME, parameters=BODY
        )

        # --------------------------------------------------------------------------
        # /Topics/put/TopicAuthorizationRuleCreate[put]
        # --------------------------------------------------------------------------
        BODY = {"rights": ["Listen", "Send"]}
        result = self.mgmt_client.topics.create_or_update_authorization_rule(
            resource_group_name=RESOURCE_GROUP,
            namespace_name=NAMESPACE_NAME,
            topic_name=TOPIC_NAME,
            authorization_rule_name=AUTHORIZATION_RULE_NAME,
            parameters=BODY,
        )

        # --------------------------------------------------------------------------
        # /Topics/get/TopicAuthorizationRuleGet[get]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.topics.get_authorization_rule(
            resource_group_name=RESOURCE_GROUP,
            namespace_name=NAMESPACE_NAME,
            topic_name=TOPIC_NAME,
            authorization_rule_name=AUTHORIZATION_RULE_NAME,
        )

        # --------------------------------------------------------------------------
        # /Topics/get/TopicAuthorizationRuleListAll[get]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.topics.list_authorization_rules(
            resource_group_name=RESOURCE_GROUP, namespace_name=NAMESPACE_NAME, topic_name=TOPIC_NAME
        )

        # --------------------------------------------------------------------------
        # /Topics/get/TopicGet[get]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.topics.list_by_namespace(
            resource_group_name=RESOURCE_GROUP, namespace_name=NAMESPACE_NAME
        )

        # --------------------------------------------------------------------------
        # /Topics/get/TopicGet[get]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.topics.list_by_namespace(
            resource_group_name=RESOURCE_GROUP, namespace_name=NAMESPACE_NAME
        )

        # --------------------------------------------------------------------------
        # /Topics/post/TopicAuthorizationRuleRegenerateKey[post]
        # --------------------------------------------------------------------------
        BODY = {"key_type": "PrimaryKey"}
        result = self.mgmt_client.topics.regenerate_keys(
            resource_group_name=RESOURCE_GROUP,
            namespace_name=NAMESPACE_NAME,
            topic_name=TOPIC_NAME,
            authorization_rule_name=AUTHORIZATION_RULE_NAME,
            parameters=BODY,
        )

        # --------------------------------------------------------------------------
        # /Topics/post/TopicAuthorizationRuleListKey[post]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.topics.list_keys(
            resource_group_name=RESOURCE_GROUP,
            namespace_name=NAMESPACE_NAME,
            topic_name=TOPIC_NAME,
            authorization_rule_name=AUTHORIZATION_RULE_NAME,
        )

        # --------------------------------------------------------------------------
        # /Topics/delete/TopicAuthorizationRuleDelete[delete]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.topics.delete_authorization_rule(
            resource_group_name=RESOURCE_GROUP,
            namespace_name=NAMESPACE_NAME,
            topic_name=TOPIC_NAME,
            authorization_rule_name=AUTHORIZATION_RULE_NAME,
        )

        # --------------------------------------------------------------------------
        # /Topics/delete/TopicDelete[delete]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.topics.delete(
            resource_group_name=RESOURCE_GROUP, namespace_name=NAMESPACE_NAME, topic_name=TOPIC_NAME
        )

        # --------------------------------------------------------------------------
        # /Namespaces/delete/NameSpaceDelete[delete]
        # --------------------------------------------------------------------------
        result = self.mgmt_client.namespaces.begin_delete(
            resource_group_name=RESOURCE_GROUP, namespace_name=NAMESPACE_NAME
        )
        result = result.result()