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
|
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
import os
import pytest
from azure.servicebus import (
ServiceBusConnectionStringProperties,
parse_connection_string,
)
from devtools_testutils import AzureMgmtRecordedTestCase
class ServiceBusConnectionStringParserTests(AzureMgmtRecordedTestCase):
def test_sb_conn_str_parse_cs(self, **kwargs):
conn_str = "Endpoint=sb://resourcename.servicebus.windows.net/;SharedAccessKeyName=test;SharedAccessKey=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
parse_result = parse_connection_string(conn_str)
assert parse_result.endpoint == "sb://resourcename.servicebus.windows.net/"
assert parse_result.fully_qualified_namespace == "resourcename.servicebus.windows.net"
assert parse_result.shared_access_key_name == "test"
assert parse_result.shared_access_key == "THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
def test_sb_conn_str_parse_sas_and_shared_key(self, **kwargs):
conn_str = "Endpoint=sb://resourcename.servicebus.windows.net/;SharedAccessKeyName=test;SharedAccessKey=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX=;SharedAccessSignature=THISISASASXXXXXXX="
with pytest.raises(ValueError) as e:
parse_result = parse_connection_string(conn_str)
assert str(e.value) == "Only one of the SharedAccessKey or SharedAccessSignature must be present."
def test_sb_parse_malformed_conn_str_no_endpoint(self, **kwargs):
conn_str = "SharedAccessKeyName=test;SharedAccessKey=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
with pytest.raises(ValueError) as e:
parse_result = parse_connection_string(conn_str)
assert str(e.value) == "Connection string is either blank or malformed."
def test_sb_parse_malformed_conn_str_no_endpoint_value(self, **kwargs):
conn_str = "Endpoint=;SharedAccessKeyName=test;SharedAccessKey=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
with pytest.raises(ValueError) as e:
parse_result = parse_connection_string(conn_str)
assert str(e.value) == "Connection string is either blank or malformed."
def test_sb_parse_malformed_conn_str_no_netloc(self, **kwargs):
conn_str = (
"Endpoint=MALFORMED;SharedAccessKeyName=test;SharedAccessKey=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
)
with pytest.raises(ValueError) as e:
parse_result = parse_connection_string(conn_str)
assert str(e.value) == "Invalid Endpoint on the Connection String."
def test_sb_parse_conn_str_sas(self, **kwargs):
conn_str = "Endpoint=sb://resourcename.servicebus.windows.net/;SharedAccessSignature=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
parse_result = parse_connection_string(conn_str)
assert parse_result.endpoint == "sb://resourcename.servicebus.windows.net/"
assert parse_result.fully_qualified_namespace == "resourcename.servicebus.windows.net"
assert parse_result.shared_access_signature == "THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
assert parse_result.shared_access_key_name == None
def test_sb_parse_conn_str_whitespace_trailing_semicolon(self, **kwargs):
conn_str = " Endpoint=sb://resourcename.servicebus.windows.net/;SharedAccessSignature=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX=; "
parse_result = parse_connection_string(conn_str)
assert parse_result.endpoint == "sb://resourcename.servicebus.windows.net/"
assert parse_result.fully_qualified_namespace == "resourcename.servicebus.windows.net"
assert parse_result.shared_access_signature == "THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
assert parse_result.shared_access_key_name == None
def test_sb_parse_conn_str_sas_trailing_semicolon(self, **kwargs):
conn_str = "Endpoint=sb://resourcename.servicebus.windows.net/;SharedAccessSignature=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX=;"
parse_result = parse_connection_string(conn_str)
assert parse_result.endpoint == "sb://resourcename.servicebus.windows.net/"
assert parse_result.fully_qualified_namespace == "resourcename.servicebus.windows.net"
assert parse_result.shared_access_signature == "THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
assert parse_result.shared_access_key_name == None
def test_sb_parse_conn_str_no_keyname(self, **kwargs):
conn_str = "Endpoint=sb://resourcename.servicebus.windows.net/;SharedAccessKey=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
with pytest.raises(ValueError) as e:
parse_result = parse_connection_string(conn_str)
assert str(e.value) == "Connection string must have both SharedAccessKeyName and SharedAccessKey."
def test_sb_parse_conn_str_no_key(self, **kwargs):
conn_str = "Endpoint=sb://resourcename.servicebus.windows.net/;SharedAccessKeyName=Test"
with pytest.raises(ValueError) as e:
parse_result = parse_connection_string(conn_str)
assert str(e.value) == "Connection string must have both SharedAccessKeyName and SharedAccessKey."
def test_sb_parse_conn_str_no_key_or_sas(self, **kwargs):
conn_str = "Endpoint=sb://resourcename.servicebus.windows.net/"
with pytest.raises(ValueError) as e:
parse_result = parse_connection_string(conn_str)
assert str(e.value) == "At least one of the SharedAccessKey or SharedAccessSignature must be present."
def test_sb_parse_malformed_conn_str_lowercase_endpoint(self, **kwargs):
conn_str = "endpoint=sb://resourcename.servicebus.windows.net/;SharedAccessKeyName=test;SharedAccessKey=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
with pytest.raises(ValueError) as e:
parse_result = parse_connection_string(conn_str)
assert str(e.value) == "Connection string is either blank or malformed."
def test_sb_parse_malformed_conn_str_lowercase_sa_key_name(self, **kwargs):
conn_str = "Endpoint=sb://resourcename.servicebus.windows.net/;sharedaccesskeyname=test;SharedAccessKey=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
with pytest.raises(ValueError) as e:
parse_result = parse_connection_string(conn_str)
assert str(e.value) == "Connection string must have both SharedAccessKeyName and SharedAccessKey."
def test_sb_parse_malformed_conn_str_lowercase_sa_key_name(self, **kwargs):
conn_str = "Endpoint=sb://resourcename.servicebus.windows.net/;SharedAccessKeyName=test;sharedaccesskey=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
with pytest.raises(ValueError) as e:
parse_result = parse_connection_string(conn_str)
assert str(e.value) == "Connection string must have both SharedAccessKeyName and SharedAccessKey."
def test_sb_parse_emulator_string(self, **kwargs):
conn_str = "Endpoint=sb://localhost;SharedAccessKeyName=RootManageSharedAccessKey;SharedAccessKey=SAS_KEY_VALUE;UseDevelopmentEmulator=true;"
parse_result = parse_connection_string(conn_str)
assert parse_result.endpoint == "sb://localhost"
assert parse_result.fully_qualified_namespace == "localhost"
conn_str = "Endpoint=sb://servicebus-emulator;SharedAccessKeyName=RootManageSharedAccessKey;SharedAccessKey=SAS_KEY_VALUE;UseDevelopmentEmulator=true;"
parse_result = parse_connection_string(conn_str)
assert parse_result.endpoint == "sb://servicebus-emulator"
assert parse_result.fully_qualified_namespace == "servicebus-emulator"
conn_str = "Endpoint=sb://192.168.y.z;SharedAccessKeyName=RootManageSharedAccessKey;SharedAccessKey=SAS_KEY_VALUE;UseDevelopmentEmulator=true;"
parse_result = parse_connection_string(conn_str)
assert parse_result.endpoint == "sb://192.168.y.z"
assert parse_result.fully_qualified_namespace == "192.168.y.z"
|