def cli_namespace_create(client, resource_group_name, namespace_name, location=None, tags=None, sku='Standard', capacity=None, is_auto_inflate_enabled=None, maximum_throughput_units=None, is_kafka_enabled=None):
from azure.mgmt.eventhub.models import EHNamespace, Sku
return client.create_or_update(
resource_group_name=resource_group_name,
namespace_name=namespace_name,
parameters=EHNamespace(
location=location,
tags=tags,
sku=Sku(name=sku, tier=sku, capacity=capacity),
is_auto_inflate_enabled=is_auto_inflate_enabled,
maximum_throughput_units=maximum_throughput_units,
kafka_enabled=is_kafka_enabled))
def test_eh_eventhub_curd(self, resource_group, location):
resource_group_name = resource_group.name
#Create a Namespace
namespace_name = "pythontestcaseeventhubnamespaceEventhub"
namespaceparameter = EHNamespace(location=location,
tags={
'tag1': 'value1',
'tag2': 'value2'
},
sku=Sku(name=SkuName.standard))
poller = self.eventhub_client.namespaces.create_or_update(
resource_group_name, namespace_name, namespaceparameter)
creatednamespace = poller.result()
self.assertEqual(creatednamespace.name, namespace_name)
#
# # Get created Namespace
#
getnamespaceresponse = self.eventhub_client.namespaces.get(
resource_group_name, namespace_name)
self.assertEqual(getnamespaceresponse.name, namespace_name)
# Create a Eventhub
eventhub_name = "testingpythontestcaseeventhub"
eventhubparameter = Eventhub(
message_retention_in_days=4,
partition_count=4,
capture_description=CaptureDescription(
enabled=True,
encoding=EncodingCaptureDescription.avro,
interval_in_seconds=120,
size_limit_in_bytes=10485763,
destination=Destination(
name="EventHubArchive.AzureBlockBlob",
storage_account_resource_id="/subscriptions/" +
self.settings.SUBSCRIPTION_ID +
"/resourceGroups/Default-Storage-SouthCentralUS/providers/Microsoft.ClassicStorage/storageAccounts/arjunteststorage",
blob_container="container",
archive_name_format=
"{Namespace}/{EventHub}/{PartitionId}/{Year}/{Month}/{Day}/{Hour}/{Minute}/{Second}"
)))
createdeventhubresponse = self.eventhub_client.event_hubs.create_or_update(
resource_group_name, namespace_name, eventhub_name,
eventhubparameter)
self.assertEqual(createdeventhubresponse.name, eventhub_name)
self.assertEqual(
createdeventhubresponse.capture_description.interval_in_seconds,
120)
#Get the created eventhub
geteventhubresponse = self.eventhub_client.event_hubs.get(
resource_group_name, namespace_name, eventhub_name)
self.assertEqual(geteventhubresponse.name, eventhub_name)
self.assertEqual(
geteventhubresponse.capture_description.interval_in_seconds, 120)
#Get the List of eventhub by namespaces
getlistbynamespaceeventhubresponse = list(
self.eventhub_client.event_hubs.list_by_namespace(
resource_group_name, namespace_name))
self.assertGreater(len(getlistbynamespaceeventhubresponse), 0)
# update the Created eventhub
eventhubupdateparameter = Eventhub(
message_retention_in_days=6,
partition_count=4,
capture_description=CaptureDescription(
enabled=True,
encoding=EncodingCaptureDescription.avro,
interval_in_seconds=130,
size_limit_in_bytes=10485900,
destination=Destination(
name="EventHubArchive.AzureBlockBlob",
storage_account_resource_id="/subscriptions/" +
self.settings.SUBSCRIPTION_ID +
"/resourceGroups/Default-Storage-SouthCentralUS/providers/Microsoft.ClassicStorage/storageAccounts/arjunteststorage",
blob_container="container",
archive_name_format=
"{Namespace}/{EventHub}/{PartitionId}/{Year}/{Month}/{Day}/{Hour}/{Minute}/{Second}"
)))
updateeventhubresponse = self.eventhub_client.event_hubs.create_or_update(
resource_group_name, namespace_name, eventhub_name,
eventhubupdateparameter)
self.assertEqual(updateeventhubresponse.name, eventhub_name)
self.assertEqual(
updateeventhubresponse.capture_description.interval_in_seconds,
130)
self.assertEqual(updateeventhubresponse.message_retention_in_days, 6)
self.assertEqual(
updateeventhubresponse.capture_description.size_limit_in_bytes,
10485900)
# # get the default authorizationrule
# defaultauthorule_name = "RootManageSharedAccessKey"
# defaultamespaceauthorule = self.eventhub_client.namespaces.get_authorization_rule(resource_group_name, namespace_name, defaultauthorule_name)
# self.assertEqual(defaultamespaceauthorule.name, defaultauthorule_name, "Default Authorization rule not returned - RootManageSharedAccessKey")
# self.assertEqual(len(defaultamespaceauthorule.rights), 3, "rights for deafult not as required - send, listen and manage")
# Create a new authorizationrule
authoRule_name = "testingauthrulepy"
createnamespaceauthorule = self.eventhub_client.event_hubs.create_or_update_authorization_rule(
resource_group_name, namespace_name, eventhub_name, authoRule_name,
[AccessRights('Send'),
AccessRights('Listen')])
self.assertEqual(
createnamespaceauthorule.name, authoRule_name,
"Authorization rule name not as created - create_or_update_authorization_rule "
)
self.assertEqual(len(createnamespaceauthorule.rights), 2)
# Get the created authorizationrule
getnamespaceauthorule = self.eventhub_client.event_hubs.get_authorization_rule(
resource_group_name, namespace_name, eventhub_name, authoRule_name)
self.assertEqual(
getnamespaceauthorule.name, authoRule_name,
"Authorization rule name not as passed as parameter - get_authorization_rule "
)
self.assertEqual(
len(getnamespaceauthorule.rights), 2,
"Access rights mis match as created - get_authorization_rule ")
# update the rights of the authorizatiorule
getnamespaceauthorule.rights.append('Manage')
updatenamespaceauthorule = self.eventhub_client.event_hubs.create_or_update_authorization_rule(
resource_group_name, namespace_name, eventhub_name, authoRule_name,
getnamespaceauthorule.rights)
self.assertEqual(
updatenamespaceauthorule.name, authoRule_name,
"Authorization rule name not as passed as parameter for update call - create_or_update_authorization_rule "
)
self.assertEqual(
len(updatenamespaceauthorule.rights), 3,
"Access rights mis match as updated - create_or_update_authorization_rule "
)
#list all the authorization ruels for the given namespace
createnamespaceauthorule = list(
self.eventhub_client.event_hubs.list_authorization_rules(
resource_group_name, namespace_name, eventhub_name))
self.assertEqual(
len(createnamespaceauthorule), 1,
"number of authorization rule mismatch with the created + default = 2 - list_authorization_rules"
)
#List keys for the authorization rule
listkeysauthorizationrule = self.eventhub_client.event_hubs.list_keys(
resource_group_name, namespace_name, eventhub_name, authoRule_name)
self.assertIsNotNone(listkeysauthorizationrule)
# regenerate Keys for authorizationrule - Primary
regenratePrimarykeyauthorizationrule = self.eventhub_client.event_hubs.regenerate_keys(
resource_group_name, namespace_name, eventhub_name, authoRule_name,
'PrimaryKey')
self.assertNotEqual(listkeysauthorizationrule.primary_key,
regenratePrimarykeyauthorizationrule.primary_key)
# regenerate Keys for authorizationrule - Primary
regenrateSecondarykeyauthorizationrule = self.eventhub_client.event_hubs.regenerate_keys(
resource_group_name, namespace_name, eventhub_name, authoRule_name,
'SecondaryKey')
self.assertNotEqual(
listkeysauthorizationrule.secondary_key,
regenrateSecondarykeyauthorizationrule.secondary_key)
# delete the authorizationrule
self.eventhub_client.event_hubs.delete_authorization_rule(
resource_group_name, namespace_name, eventhub_name, authoRule_name)
# Delete the created eventhub
geteventhubresponse = self.eventhub_client.event_hubs.delete(
resource_group_name, namespace_name, eventhub_name)
# Delete the create namespace
try:
deletenamespace = self.eventhub_client.namespaces.delete(
resource_group_name, namespace_name).result()
except CloudError as ErrorResponse:
self.assertTrue("not found" in ErrorResponse.message)
def test_eh_consumergroup_curd(self, resource_group, location):
resource_group_name = resource_group.name
# Create a Namespace
namespace_name = "pythontestcaseeventhubnamespaceConsumerGroup"
namespaceparameter = EHNamespace(location=location,
tags={
'tag1': 'value1',
'tag2': 'value2'
},
sku=Sku(name=SkuName.standard))
poller = self.eventhub_client.namespaces.create_or_update(
resource_group_name, namespace_name, namespaceparameter)
creatednamespace = poller.result()
self.assertEqual(creatednamespace.name, namespace_name)
#
# # Get created Namespace
#
getnamespaceresponse = self.eventhub_client.namespaces.get(
resource_group_name, namespace_name)
self.assertEqual(getnamespaceresponse.name, namespace_name)
# # Create a Eventhub
eventhub_name = "testingpythontestcaseeventhubConsumerGroup"
eventhubparameter = Eventhub(
message_retention_in_days=4,
partition_count=4,
capture_description=CaptureDescription(
enabled=True,
encoding=EncodingCaptureDescription.avro,
interval_in_seconds=120,
size_limit_in_bytes=10485763,
destination=Destination(
name="EventHubArchive.AzureBlockBlob",
storage_account_resource_id="/subscriptions/" +
self.settings.SUBSCRIPTION_ID +
"/resourceGroups/Default-Storage-SouthCentralUS/providers/Microsoft.ClassicStorage/storageAccounts/arjunteststorage",
blob_container="container",
archive_name_format=
"{Namespace}/{EventHub}/{PartitionId}/{Year}/{Month}/{Day}/{Hour}/{Minute}/{Second}"
)))
createdeventhubresponse = self.eventhub_client.event_hubs.create_or_update(
resource_group_name, namespace_name, eventhub_name,
eventhubparameter)
self.assertEqual(createdeventhubresponse.name, eventhub_name)
self.assertEqual(
createdeventhubresponse.capture_description.interval_in_seconds,
120)
# Get the created eventhub
geteventhubresponse = self.eventhub_client.event_hubs.get(
resource_group_name, namespace_name, eventhub_name)
self.assertEqual(geteventhubresponse.name, eventhub_name)
self.assertEqual(
geteventhubresponse.capture_description.interval_in_seconds, 120)
#Get the List of eventhub by namespaces
getlistbynamespaceeventhubresponse = list(
self.eventhub_client.event_hubs.list_by_namespace(
resource_group_name, namespace_name))
self.assertGreater(len(getlistbynamespaceeventhubresponse), 0)
# # update the Created eventhub
eventhubupdateparameter = Eventhub(
message_retention_in_days=6,
partition_count=4,
capture_description=CaptureDescription(
enabled=True,
encoding=EncodingCaptureDescription.avro,
interval_in_seconds=130,
size_limit_in_bytes=10485900,
destination=Destination(
name="EventHubArchive.AzureBlockBlob",
storage_account_resource_id="/subscriptions/" +
self.settings.SUBSCRIPTION_ID +
"/resourceGroups/Default-Storage-SouthCentralUS/providers/Microsoft.ClassicStorage/storageAccounts/arjunteststorage",
blob_container="container",
archive_name_format=
"{Namespace}/{EventHub}/{PartitionId}/{Year}/{Month}/{Day}/{Hour}/{Minute}/{Second}"
)))
updateeventhubresponse = self.eventhub_client.event_hubs.create_or_update(
resource_group_name, namespace_name, eventhub_name,
eventhubupdateparameter)
self.assertEqual(updateeventhubresponse.name, eventhub_name)
self.assertEqual(
updateeventhubresponse.capture_description.interval_in_seconds,
130)
self.assertEqual(updateeventhubresponse.message_retention_in_days, 6)
self.assertEqual(
updateeventhubresponse.capture_description.size_limit_in_bytes,
10485900)
# Create ConsumerGroup
consumergroup_name = "testingpythontestcaseconsumergroup"
createconsumergroupresponse = self.eventhub_client.consumer_groups.create_or_update(
resource_group_name, namespace_name, eventhub_name,
consumergroup_name, "Testing the User Metadata")
self.assertEqual(createconsumergroupresponse.name, consumergroup_name)
self.assertEqual(createconsumergroupresponse.user_metadata,
"Testing the User Metadata")
# Get the Created Consumer group
getconsumergroupresponse = self.eventhub_client.consumer_groups.get(
resource_group_name, namespace_name, eventhub_name,
consumergroup_name)
# update the Created Consumer group
updateconsumergroupresponse = self.eventhub_client.consumer_groups.create_or_update(
resource_group_name, namespace_name, eventhub_name,
consumergroup_name, "update user matadata for testing")
self.assertEqual(updateconsumergroupresponse.name, consumergroup_name)
self.assertEqual(updateconsumergroupresponse.user_metadata,
"update user matadata for testing")
# delete the consumergroup
deleteconsumergroupresponse = self.eventhub_client.consumer_groups.delete(
resource_group_name, namespace_name, eventhub_name,
consumergroup_name)
# Delete the created eventhub
geteventhubresponse = self.eventhub_client.event_hubs.delete(
resource_group_name, namespace_name, eventhub_name)
# Delete the create namespace
try:
deletenamespace = self.eventhub_client.namespaces.delete(
resource_group_name, namespace_name).result()
except CloudError as ErrorResponse:
self.assertTrue("not found" in ErrorResponse.message)
def test_eh_namespace_curd(self, resource_group, location):
resource_group_name = resource_group.name
# Create a Namespace
namespace_name = "pythontestcasenamespace"
namespaceparameter=EHNamespace(location=location, tags={'tag1' : 'value1', 'tag2' : 'value2'}, sku=Sku(name=SkuName.standard))
poller = self.eventhub_client.namespaces.create_or_update(resource_group_name, namespace_name, namespaceparameter)
creatednamespace = poller.result()
self.assertEqual(creatednamespace.name, namespace_name)
#
# # Get created Namespace
#
getnamespaceresponse = self.eventhub_client.namespaces.get(resource_group_name, namespace_name)
self.assertEqual(getnamespaceresponse.name, namespace_name)
# Get the List of Namespaces under the resourceGroup - list_by_resource_group
listbyresourcegroupresponse = list(self.eventhub_client.namespaces.list_by_resource_group(resource_group_name, False, False))
self.assertGreater(len(listbyresourcegroupresponse), 0, "No Namespace returned, List is empty")
self.assertEqual(listbyresourcegroupresponse[0].name, namespace_name, "Created namespace not found - ListByResourgroup")
# Get the List of namespace under the subscription - list
listbysubscriptionresponse = list(self.eventhub_client.namespaces.list())
self.assertGreater(len(listbysubscriptionresponse), 0, "No Namespace returned, List is empty")
# get the default authorizationrule
defaultauthorule_name = "RootManageSharedAccessKey"
defaultamespaceauthorule = self.eventhub_client.namespaces.get_authorization_rule(resource_group_name, namespace_name, defaultauthorule_name)
self.assertEqual(defaultamespaceauthorule.name, defaultauthorule_name, "Default Authorization rule not returned - RootManageSharedAccessKey")
self.assertEqual(len(defaultamespaceauthorule.rights), 3, "rights for deafult not as required - send, listen and manage ")
# Create a new authorizationrule
authoRule_name = "testingauthrulepy"
createnamespaceauthorule = self.eventhub_client.namespaces.create_or_update_authorization_rule(resource_group_name,namespace_name,authoRule_name,[AccessRights('Send'),AccessRights('Listen')])
self.assertEqual(createnamespaceauthorule.name,authoRule_name, "Authorization rule name not as created - create_or_update_authorization_rule ")
self.assertEqual(len(createnamespaceauthorule.rights),2)
# Get the created authorizationrule
getnamespaceauthorule = self.eventhub_client.namespaces.get_authorization_rule(resource_group_name, namespace_name, authoRule_name)
self.assertEqual(getnamespaceauthorule.name, authoRule_name, "Authorization rule name not as passed as parameter - get_authorization_rule ")
self.assertEqual(len(getnamespaceauthorule.rights), 2, "Access rights mis match as created - get_authorization_rule ")
# update the rights of the authorizatiorule
getnamespaceauthorule.rights.append('Manage')
updatenamespaceauthorule = self.eventhub_client.namespaces.create_or_update_authorization_rule(resource_group_name, namespace_name, authoRule_name, getnamespaceauthorule.rights)
self.assertEqual(updatenamespaceauthorule.name, authoRule_name, "Authorization rule name not as passed as parameter for update call - create_or_update_authorization_rule ")
self.assertEqual(len(updatenamespaceauthorule.rights), 3, "Access rights mis match as updated - create_or_update_authorization_rule ")
#list all the authorization ruels for the given namespace
createnamespaceauthorule = list(self.eventhub_client.namespaces.list_authorization_rules(resource_group_name, namespace_name))
self.assertEqual(len(createnamespaceauthorule),2, "number of authorization rule mismatch with the created + default = 2 - list_authorization_rules")
#List keys for the authorization rule
listkeysauthorizationrule = self.eventhub_client.namespaces.list_keys(resource_group_name, namespace_name, authoRule_name)
self.assertIsNotNone(listkeysauthorizationrule)
# regenerate Keys for authorizationrule - Primary
regenratePrimarykeyauthorizationrule = self.eventhub_client.namespaces.regenerate_keys(resource_group_name, namespace_name,authoRule_name,'PrimaryKey')
self.assertNotEqual(listkeysauthorizationrule.primary_key,regenratePrimarykeyauthorizationrule.primary_key)
# regenerate Keys for authorizationrule - Primary
regenrateSecondarykeyauthorizationrule = self.eventhub_client.namespaces.regenerate_keys(resource_group_name,namespace_name,authoRule_name, 'SecondaryKey')
self.assertNotEqual(listkeysauthorizationrule.secondary_key, regenrateSecondarykeyauthorizationrule.secondary_key)
# delete the authorizationrule
self.eventhub_client.namespaces.delete_authorization_rule(resource_group_name, namespace_name, authoRule_name)
# list all the authorization ruels for the given namespace
createnamespaceauthorule = list(self.eventhub_client.namespaces.list_authorization_rules(resource_group_name, namespace_name))
self.assertEqual(len(createnamespaceauthorule), 1)
self.assertEqual(createnamespaceauthorule[0].name, defaultauthorule_name)
# Delete the create namespace
try:
deletenamespace = self.eventhub_client.namespaces.delete(resource_group_name, namespace_name).result()
except CloudError as ErrorResponse:
self.assertTrue("not found" in ErrorResponse.message)