def test_mgmt_rule_list_and_delete(self,
servicebus_namespace_connection_string):
mgmt_service = ServiceBusManagementClient.from_connection_string(
servicebus_namespace_connection_string)
clear_topics(mgmt_service)
topic_name = "topic_testaddf"
subscription_name = "sub_testkkk"
rule_name_1 = 'test_rule_1'
rule_name_2 = 'test_rule_2'
rule_name_3 = 'test_rule_3'
sql_filter_1 = SqlRuleFilter("Priority = 'low'")
sql_filter_2 = SqlRuleFilter("Priority = 'middle'")
sql_filter_3 = SqlRuleFilter("Priority = 'high'")
rule_1 = RuleDescription(name=rule_name_1, filter=sql_filter_1)
rule_2 = RuleDescription(name=rule_name_2, filter=sql_filter_2)
rule_3 = RuleDescription(name=rule_name_3, filter=sql_filter_3)
try:
mgmt_service.create_topic(topic_name)
mgmt_service.create_subscription(topic_name, subscription_name)
rules = list(mgmt_service.list_rules(topic_name,
subscription_name))
assert len(rules) == 1 # by default there is a True filter
mgmt_service.create_rule(topic_name, subscription_name, rule_1)
mgmt_service.create_rule(topic_name, subscription_name, rule_2)
mgmt_service.create_rule(topic_name, subscription_name, rule_3)
rules = list(mgmt_service.list_rules(topic_name,
subscription_name))
assert len(rules) == 3 + 1
mgmt_service.delete_rule(topic_name, subscription_name,
rule_name_2)
rules = list(mgmt_service.list_rules(topic_name,
subscription_name))
assert len(rules) == 2 + 1
assert rules[0].name == "$Default"
assert rules[1].name == rule_name_1
assert type(rules[1].filter) == SqlRuleFilter
assert rules[1].filter.sql_expression == "Priority = 'low'"
assert rules[2].name == rule_name_3
assert type(rules[2].filter) == SqlRuleFilter
assert rules[2].filter.sql_expression == "Priority = 'high'"
mgmt_service.delete_rule(topic_name, subscription_name,
rule_name_1)
mgmt_service.delete_rule(topic_name, subscription_name,
rule_name_3)
rules = list(mgmt_service.list_rules(topic_name,
subscription_name))
assert len(rules) == 1
finally:
mgmt_service.delete_subscription(topic_name, subscription_name)
mgmt_service.delete_topic(topic_name)
async def test_async_mgmt_rule_update_success(self, servicebus_namespace_connection_string, **kwargs):
mgmt_service = ServiceBusManagementClient.from_connection_string(servicebus_namespace_connection_string)
await clear_topics(mgmt_service)
topic_name = "fjrui"
subscription_name = "eqkovc"
rule_name = 'rule'
sql_filter = SqlRuleFilter("Priority = 'low'")
rule = RuleDescription(name=rule_name, filter=sql_filter)
try:
topic_description = await mgmt_service.create_topic(topic_name)
subscription_description = await mgmt_service.create_subscription(topic_description, subscription_name)
await mgmt_service.create_rule(topic_name, subscription_name, rule)
rule_desc = await mgmt_service.get_rule(topic_name, subscription_name, rule_name)
assert type(rule_desc.filter) == SqlRuleFilter
assert rule_desc.filter.sql_expression == "Priority = 'low'"
correlation_fitler = CorrelationRuleFilter(correlation_id='testcid')
sql_rule_action = SqlRuleAction(sql_expression="SET Priority = 'low'")
rule_desc.filter = correlation_fitler
rule_desc.action = sql_rule_action
await mgmt_service.update_rule(topic_description, subscription_description, rule_desc)
rule_desc = await mgmt_service.get_rule(topic_name, subscription_name, rule_name)
assert type(rule_desc.filter) == CorrelationRuleFilter
assert rule_desc.filter.correlation_id == 'testcid'
assert rule_desc.action.sql_expression == "SET Priority = 'low'"
finally:
await mgmt_service.delete_rule(topic_name, subscription_name, rule_name)
await mgmt_service.delete_subscription(topic_name, subscription_name)
await mgmt_service.delete_topic(topic_name)
async def test_async_mgmt_rule_create(self, servicebus_namespace_connection_string, **kwargs):
mgmt_service = ServiceBusManagementClient.from_connection_string(servicebus_namespace_connection_string)
await clear_topics(mgmt_service)
topic_name = "topic_testaddf"
subscription_name = "sub_testkkk"
rule_name_1 = 'test_rule_1'
rule_name_2 = 'test_rule_2'
rule_name_3 = 'test_rule_3'
correlation_fitler = CorrelationRuleFilter(correlation_id='testcid')
sql_rule_action = SqlRuleAction(sql_expression="SET Priority = 'low'")
rule_1 = RuleDescription(name=rule_name_1, filter=correlation_fitler, action=sql_rule_action)
sql_filter = SqlRuleFilter("Priority = 'low'")
rule_2 = RuleDescription(name=rule_name_2, filter=sql_filter)
bool_filter = TrueRuleFilter()
rule_3 = RuleDescription(name=rule_name_3, filter=bool_filter)
try:
await mgmt_service.create_topic(topic_name)
await mgmt_service.create_subscription(topic_name, subscription_name)
await mgmt_service.create_rule(topic_name, subscription_name, rule_1)
rule_desc = await mgmt_service.get_rule(topic_name, subscription_name, rule_name_1)
assert type(rule_desc.filter) == CorrelationRuleFilter
assert rule_desc.filter.correlation_id == 'testcid'
assert rule_desc.action.sql_expression == "SET Priority = 'low'"
await mgmt_service.create_rule(topic_name, subscription_name, rule_2)
rule_desc = await mgmt_service.get_rule(topic_name, subscription_name, rule_name_2)
assert type(rule_desc.filter) == SqlRuleFilter
assert rule_desc.filter.sql_expression == "Priority = 'low'"
await mgmt_service.create_rule(topic_name, subscription_name, rule_3)
rule_desc = await mgmt_service.get_rule(topic_name, subscription_name, rule_name_3)
assert type(rule_desc.filter) == TrueRuleFilter
finally:
await mgmt_service.delete_rule(topic_name, subscription_name, rule_name_1)
await mgmt_service.delete_rule(topic_name, subscription_name, rule_name_2)
await mgmt_service.delete_rule(topic_name, subscription_name, rule_name_3)
await mgmt_service.delete_subscription(topic_name, subscription_name)
await mgmt_service.delete_topic(topic_name)
def test_mgmt_rule_update_invalid(self,
servicebus_namespace_connection_string,
**kwargs):
mgmt_service = ServiceBusManagementClient.from_connection_string(
servicebus_namespace_connection_string)
clear_topics(mgmt_service)
topic_name = "fjrui"
subscription_name = "eqkovc"
rule_name = 'rule'
sql_filter = SqlRuleFilter("Priority = 'low'")
rule = RuleDescription(name=rule_name, filter=sql_filter)
try:
topic_description = mgmt_service.create_topic(topic_name)
subscription_description = mgmt_service.create_subscription(
topic_name, subscription_name)
mgmt_service.create_rule(topic_name, subscription_name, rule)
rule_desc = mgmt_service.get_rule(topic_name, subscription_name,
rule_name)
# handle a null update properly.
with pytest.raises(AttributeError):
mgmt_service.update_rule(topic_name, subscription_name, None)
# handle an invalid type update properly.
with pytest.raises(AttributeError):
mgmt_service.update_rule(topic_name, subscription_name,
Exception("test"))
# change the name to a topic that doesn't exist; should fail.
rule_desc.name = "iewdm"
with pytest.raises(HttpResponseError):
mgmt_service.update_rule(topic_name, subscription_description,
rule_desc)
rule_desc.name = rule_name
# change the name to a topic with an invalid name exist; should fail.
rule_desc.name = ''
with pytest.raises(msrest.exceptions.ValidationError):
mgmt_service.update_rule(topic_name, subscription_description,
rule_desc)
rule_desc.name = rule_name
finally:
mgmt_service.delete_rule(topic_name, subscription_name, rule_name)
mgmt_service.delete_subscription(topic_name, subscription_name)
mgmt_service.delete_topic(topic_name)
async def test_async_mgmt_rule_create_duplicate(self, servicebus_namespace_connection_string, **kwargs):
mgmt_service = ServiceBusManagementClient.from_connection_string(servicebus_namespace_connection_string)
await clear_topics(mgmt_service)
topic_name = "dqkodq"
subscription_name = 'kkaqo'
rule_name = 'rule'
sql_filter = SqlRuleFilter("Priority = 'low'")
rule = RuleDescription(name=rule_name, filter=sql_filter)
try:
await mgmt_service.create_topic(topic_name)
await mgmt_service.create_subscription(topic_name, subscription_name)
await mgmt_service.create_rule(topic_name, subscription_name, rule)
with pytest.raises(ResourceExistsError):
await mgmt_service.create_rule(topic_name, subscription_name, rule)
finally:
await mgmt_service.delete_rule(topic_name, subscription_name, rule_name)
await mgmt_service.delete_subscription(topic_name, subscription_name)
await mgmt_service.delete_topic(topic_name)
async def test_async_mgmt_rule_create(self, servicebus_namespace_connection_string, **kwargs):
mgmt_service = ServiceBusManagementClient.from_connection_string(servicebus_namespace_connection_string)
await clear_topics(mgmt_service)
topic_name = "topic_testaddf"
subscription_name = "sub_testkkk"
rule_name_1 = 'test_rule_1'
rule_name_2 = 'test_rule_2'
rule_name_3 = 'test_rule_3'
correlation_fitler = CorrelationRuleFilter(correlation_id='testcid', properties={
"key_string": "str1",
"key_int": 2,
"key_long": INT32_MAX_VALUE + 3,
"key_bool": False,
"key_datetime": datetime(2020, 7, 5, 11, 12, 13),
"key_duration": timedelta(days=1, hours=2, minutes=3)
})
sql_rule_action = SqlRuleAction(sql_expression="SET Priority = @param", parameters={
"@param": datetime(2020, 7, 5, 11, 12, 13),
})
rule_1 = RuleDescription(name=rule_name_1, filter=correlation_fitler, action=sql_rule_action)
sql_filter = SqlRuleFilter("Priority = @param1", parameters={
"@param1": "str1",
})
rule_2 = RuleDescription(name=rule_name_2, filter=sql_filter)
bool_filter = TrueRuleFilter()
rule_3 = RuleDescription(name=rule_name_3, filter=bool_filter)
try:
await mgmt_service.create_topic(topic_name)
await mgmt_service.create_subscription(topic_name, subscription_name)
await mgmt_service.create_rule(topic_name, subscription_name, rule_1)
rule_desc = await mgmt_service.get_rule(topic_name, subscription_name, rule_name_1)
rule_properties = rule_desc.filter.properties
assert type(rule_desc.filter) == CorrelationRuleFilter
assert rule_desc.filter.correlation_id == 'testcid'
assert rule_desc.action.sql_expression == "SET Priority = @param"
assert rule_desc.action.parameters["@param"] == datetime(2020, 7, 5, 11, 12, 13)
assert rule_properties["key_string"] == "str1"
assert rule_properties["key_int"] == 2
assert rule_properties["key_long"] == INT32_MAX_VALUE + 3
assert rule_properties["key_bool"] is False
assert rule_properties["key_datetime"] == datetime(2020, 7, 5, 11, 12, 13)
assert rule_properties["key_duration"] == timedelta(days=1, hours=2, minutes=3)
await mgmt_service.create_rule(topic_name, subscription_name, rule_2)
rule_desc = await mgmt_service.get_rule(topic_name, subscription_name, rule_name_2)
assert type(rule_desc.filter) == SqlRuleFilter
assert rule_desc.filter.sql_expression == "Priority = @param1"
assert rule_desc.filter.parameters["@param1"] == "str1"
await mgmt_service.create_rule(topic_name, subscription_name, rule_3)
rule_desc = await mgmt_service.get_rule(topic_name, subscription_name, rule_name_3)
assert type(rule_desc.filter) == TrueRuleFilter
finally:
await mgmt_service.delete_rule(topic_name, subscription_name, rule_name_1)
await mgmt_service.delete_rule(topic_name, subscription_name, rule_name_2)
await mgmt_service.delete_rule(topic_name, subscription_name, rule_name_3)
await mgmt_service.delete_subscription(topic_name, subscription_name)
await mgmt_service.delete_topic(topic_name)