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
|
# -------------------------------------------------------------------------
# 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.eventhub import (
EventHubConnectionStringProperties,
parse_connection_string,
)
from azure.eventhub._client_base import _parse_conn_str
class TestEventHubConnectionStringParser:
def test_eh_conn_str_parse_cs(self, **kwargs):
conn_str = "Endpoint=sb://eh-namespace.servicebus.windows.net/;SharedAccessKeyName=test-policy;SharedAccessKey=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
parse_result = parse_connection_string(conn_str)
assert parse_result.endpoint == "sb://eh-namespace.servicebus.windows.net/"
assert parse_result.fully_qualified_namespace == "eh-namespace.servicebus.windows.net"
assert parse_result.shared_access_key_name == "test-policy"
assert parse_result.shared_access_key == "THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
def test_eh_conn_str_parse_with_entity_path(self, **kwargs):
conn_str = "Endpoint=sb://eh-namespace.servicebus.windows.net/;SharedAccessKeyName=test-policy;SharedAccessKey=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX=;EntityPath=eventhub-name"
parse_result = parse_connection_string(conn_str)
assert parse_result.endpoint == "sb://eh-namespace.servicebus.windows.net/"
assert parse_result.fully_qualified_namespace == "eh-namespace.servicebus.windows.net"
assert parse_result.shared_access_key_name == "test-policy"
assert parse_result.shared_access_key == "THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
assert parse_result.eventhub_name == "eventhub-name"
def test_eh_conn_str_parse_sas_and_shared_key(self, **kwargs):
conn_str = "Endpoint=sb://eh-namespace.servicebus.windows.net/;SharedAccessKeyName=test-policy;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_eh_parse_malformed_conn_str_no_endpoint(self, **kwargs):
conn_str = "SharedAccessKeyName=test-policy;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_eh_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_eh_parse_malformed_conn_str_no_netloc(self, **kwargs):
conn_str = "Endpoint=MALFORMED;SharedAccessKeyName=test-policy;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_eh_parse_conn_str_sas(self, **kwargs):
conn_str = "Endpoint=sb://eh-namespace.servicebus.windows.net/;SharedAccessSignature=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
parse_result = parse_connection_string(conn_str)
assert parse_result.endpoint == "sb://eh-namespace.servicebus.windows.net/"
assert parse_result.fully_qualified_namespace == "eh-namespace.servicebus.windows.net"
assert parse_result.shared_access_signature == "THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
assert parse_result.shared_access_key_name == None
def test_eh_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_eh_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_eh_parse_conn_str_no_keyname(self, **kwargs):
conn_str = "Endpoint=sb://eh-namespace.servicebus.windows.net/;SharedAccessKey=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX="
with pytest.raises(ValueError) as e:
parse_result = parse_connection_string(conn_str)
assert "Invalid connection string" in str(e.value)
def test_eh_parse_conn_str_no_key(self, **kwargs):
conn_str = "Endpoint=sb://eh-namespace.servicebus.windows.net/;SharedAccessKeyName=test-policy"
with pytest.raises(ValueError) as e:
parse_result = parse_connection_string(conn_str)
assert "Invalid connection string" in str(e.value)
def test_eh_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_eh_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_eh_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 "Invalid connection string" in str(e.value)
def test_eh_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 "Invalid connection string" in str(e.value)
def test_eh_emulator_slug_parse_localhost(self, **kwargs):
conn_str = "Endpoint=sb://localhost:6065;SharedAccessKeyName=test-policy;SharedAccessKey=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX=;UseDevelopmentEmulator=true"
fully_qualified_namespace, policy, key, entity, signature, expiry, emulator = _parse_conn_str(conn_str)
assert fully_qualified_namespace == "localhost:6065"
assert emulator == True
def test_eh_emulator_slug_parse_ipv6(self, **kwargs):
conn_str = "Endpoint=sb://::1;SharedAccessKeyName=test-policy;SharedAccessKey=THISISATESTKEYXXXXXXXXXXXXXXXXXXXXXXXXXXXX=;UseDevelopmentEmulator=true"
fully_qualified_namespace, policy, key, entity, signature, expiry, emulator = _parse_conn_str(conn_str)
assert fully_qualified_namespace == "::1"
assert emulator == True
|
