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
|
# 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 unittest
import time
import azure.mgmt.servicebus.models
from azure.mgmt.servicebus.models import SBNamespace, SBSku, SkuName, SBTopic, SBSubscription
from azure.common.credentials import ServicePrincipalCredentials
from devtools_testutils import AzureMgmtTestCase, RandomNameResourceGroupPreparer
class MgmtServiceBusTest(AzureMgmtTestCase):
def setUp(self):
super(MgmtServiceBusTest, self).setUp()
self.servicebus_client = self.create_mgmt_client(azure.mgmt.servicebus.ServiceBusManagementClient)
@RandomNameResourceGroupPreparer()
def test_sb_subscription_curd(self, resource_group, location):
# List all topic types
resource_group_name = resource_group.name # "ardsouza-resourcemovetest-group2"
# Create a Namespace
namespace_name = self.get_replayable_random_resource_name("testingpythontestcasesubscription")
namespaceparameter = SBNamespace(
location=location, tags={"tag1": "value1", "tag2": "value2"}, sku=SBSku(name=SkuName.standard)
)
creatednamespace = self.servicebus_client.namespaces.create_or_update(
resource_group_name, namespace_name, namespaceparameter
).result()
self.assertEqual(creatednamespace.name, namespace_name)
while (
self.servicebus_client.namespaces.get(resource_group_name, namespace_name).provisioning_state != "Succeeded"
):
if self.is_live:
time.sleep(15)
continue
# Create a Topic
topic_name = self.get_replayable_random_resource_name("testingpythonsdktopic")
createtopicresponse = self.servicebus_client.topics.create_or_update(
resource_group_name, namespace_name, topic_name, SBTopic()
)
self.assertEqual(createtopicresponse.name, topic_name)
# Get the created Topic
gettopicresponse = self.servicebus_client.topics.get(resource_group_name, namespace_name, topic_name)
self.assertEqual(gettopicresponse.name, topic_name)
# Create subscription
subscription_name = self.get_replayable_random_resource_name("testingpythonsdksubscription")
createsubscriptionresponse = self.servicebus_client.subscriptions.create_or_update(
resource_group_name, namespace_name, topic_name, subscription_name, SBSubscription()
)
self.assertEqual(createsubscriptionresponse.name, subscription_name)
# Get created subscription
getsubscriptionresponse = self.servicebus_client.subscriptions.get(
resource_group_name, namespace_name, topic_name, subscription_name
)
self.assertEqual(getsubscriptionresponse.name, subscription_name)
# list the subscription by Topic
listsubscriptionresponse = list(
self.servicebus_client.subscriptions.list_by_topic(resource_group_name, namespace_name, topic_name)
)
self.assertEqual(len(listsubscriptionresponse), 1)
# Update created subscription
updatesubscriptionresponse = self.servicebus_client.subscriptions.create_or_update(
resource_group_name,
namespace_name,
topic_name,
subscription_name,
SBSubscription(enable_batched_operations=True, dead_lettering_on_message_expiration=True),
)
self.assertEqual(updatesubscriptionresponse.enable_batched_operations, True)
self.assertEqual(updatesubscriptionresponse.dead_lettering_on_message_expiration, True)
# Delete the created subscription
self.servicebus_client.subscriptions.delete(resource_group_name, namespace_name, topic_name, subscription_name)
# delete the Topic
self.servicebus_client.topics.delete(resource_group_name, namespace_name, topic_name)
# Delete the create namespace
deletenamespace = self.servicebus_client.namespaces.delete(resource_group_name, namespace_name).result()
# ------------------------------------------------------------------------------
if __name__ == "__main__":
unittest.main()
|
