|
8 | 8 | from azure.eventhub._pyamqp import authentication, ReceiveClient, SendClient |
9 | 9 | from azure.eventhub._pyamqp.constants import TransportType |
10 | 10 | from azure.eventhub._pyamqp.message import Message |
| 11 | +from azure.eventhub._pyamqp._encode import encode_described |
11 | 12 |
|
12 | 13 |
|
13 | 14 | def send_message(live_eventhub): |
@@ -54,3 +55,43 @@ def test_event_hubs_client_amqp(live_eventhub): |
54 | 55 | ) as receive_client: |
55 | 56 | messages = receive_client.receive_message_batch(max_batch_size=1) |
56 | 57 | assert len(messages) > 0 |
| 58 | + |
| 59 | + |
| 60 | +def test_described(live_eventhub): |
| 61 | + uri = "sb://{}/{}".format(live_eventhub["hostname"], live_eventhub["event_hub"]) |
| 62 | + sas_auth = authentication.SASTokenAuth( |
| 63 | + uri=uri, audience=uri, username=live_eventhub["key_name"], password=live_eventhub["access_key"] |
| 64 | + ) |
| 65 | + |
| 66 | + target = "amqps://{}/{}/Partitions/{}".format( |
| 67 | + live_eventhub["hostname"], live_eventhub["event_hub"], live_eventhub["partition"] |
| 68 | + ) |
| 69 | + |
| 70 | + out = bytearray(); |
| 71 | + encode_described(out, (12345, "TEST")) |
| 72 | + |
| 73 | + message = Message(value=out) |
| 74 | + |
| 75 | + with SendClient( |
| 76 | + live_eventhub["hostname"], target, auth=sas_auth, debug=True, transport_type=TransportType.Amqp |
| 77 | + ) as send_client: |
| 78 | + send_client.send_message(message) |
| 79 | + |
| 80 | + source = "amqps://{}/{}/ConsumerGroups/{}/Partitions/{}".format( |
| 81 | + live_eventhub["hostname"], |
| 82 | + live_eventhub["event_hub"], |
| 83 | + live_eventhub["consumer_group"], |
| 84 | + live_eventhub["partition"], |
| 85 | + ) |
| 86 | + |
| 87 | + with ReceiveClient( |
| 88 | + live_eventhub["hostname"], |
| 89 | + source, |
| 90 | + auth=sas_auth, |
| 91 | + debug=False, |
| 92 | + timeout=500, |
| 93 | + prefetch=1, |
| 94 | + transport_type=TransportType.Amqp, |
| 95 | + ) as receive_client: |
| 96 | + messages = receive_client.receive_message_batch(max_batch_size=1) |
| 97 | + assert len(messages) > 0 |
0 commit comments