def test_from_json_sb():
obj = MockServiceBusReceivedMessage(
body=MockBody(),
message_id='3f6c5441-5be5-4f33-80c3-3ffeb6a090ce',
content_type='application/cloudevents+json; charset=utf-8',
time_to_live=datetime.timedelta(days=14),
delivery_count=13,
enqueued_sequence_number=0,
enqueued_time_utc=datetime.datetime(2021, 7, 22, 22, 27, 41, 236000),
expires_at_utc=datetime.datetime(2021, 8, 5, 22, 27, 41, 236000),
sequence_number=11219,
lock_token='233146e3-d5a6-45eb-826f-691d82fb8b13')
event = EventGridEvent.from_json(obj)
assert event.id == "f208feff-099b-4bda-a341-4afd0fa02fef"
assert event.data == "ServiceBus"
def test_from_json_storage():
eg_storage_dict = """{
"id":"a0517898-9fa4-4e70-b4a3-afda1dd68672",
"subject":"/subscriptions/{subscription-id}/resourceGroups/{resource-group}/providers/Microsoft.Storage/storageAccounts/{storage-account}",
"data":{
"api":"PutBlockList",
"client_request_id":"6d79dbfb-0e37-4fc4-981f-442c9ca65760",
"request_id":"831e1650-001e-001b-66ab-eeb76e000000",
"e_tag":"0x8D4BCC2E4835CD0",
"content_type":"application/octet-stream",
"content_length":524288,
"blob_type":"BlockBlob",
"url":"https://oc2d2817345i60006.blob.core.windows.net/oc2d2817345i200097container/oc2d2817345i20002296blob",
"sequencer":"00000000000004420000000000028963",
"storage_diagnostics":{"batchId":"b68529f3-68cd-4744-baa4-3c0498ec19f0"}
},
"event_type":"Microsoft.Storage.BlobCreated",
"event_time":"2021-02-18T20:18:10.581147898Z",
"data_version":"1.0"
}"""
obj = MockQueueMessage(content=eg_storage_dict)
event = EventGridEvent.from_json(obj)
assert event.data == {
"api": "PutBlockList",
"client_request_id": "6d79dbfb-0e37-4fc4-981f-442c9ca65760",
"request_id": "831e1650-001e-001b-66ab-eeb76e000000",
"e_tag": "0x8D4BCC2E4835CD0",
"content_type": "application/octet-stream",
"content_length": 524288,
"blob_type": "BlockBlob",
"url":
"https://oc2d2817345i60006.blob.core.windows.net/oc2d2817345i200097container/oc2d2817345i20002296blob",
"sequencer": "00000000000004420000000000028963",
"storage_diagnostics": {
"batchId": "b68529f3-68cd-4744-baa4-3c0498ec19f0"
}
}
def test_from_json():
json_str = '{"id": "de0fd76c-4ef4-4dfb-ab3a-8f24a307e033", "subject": "https://egtest.dev/cloudcustomevent", "data": {"team": "event grid squad"}, "event_type": "Azure.Sdk.Sample", "event_time": "2020-08-07T02:06:08.11969Z", "data_version": "1.0"}'
event = EventGridEvent.from_json(json_str)
assert event.data == {"team": "event grid squad"}
assert event.event_time == datetime.datetime(2020, 8, 7, 2, 6, 8, 119690,
UTC())
def test_from_json_eh():
obj = MockEventhubData(body=MockEhBody())
event = EventGridEvent.from_json(obj)
assert event.id == "f208feff-099b-4bda-a341-4afd0fa02fef"
assert event.data == "Eventhub"
FILE: consume_cloud_events_from_service_bus_queue.py
DESCRIPTION:
These samples demonstrate receiving events from Service Bus.
USAGE:
python consume_cloud_events_from_service_bus_queue.py
Set the environment variables with your own values before running the sample:
1) SB_CONN_STR: The connection string to the Service Bus account
3) SERVICE_BUS_QUEUE_NAME: The name of the servicebus account
"""
# Note: This sample would not work on pypy since azure-servicebus
# depends on uamqp which is not pypy compatible.
from azure.eventgrid import EventGridEvent
from azure.servicebus import ServiceBusClient
import os
import json
# all types of EventGridEvents below produce same DeserializedEvent
connection_str = os.environ['SERVICE_BUS_CONNECTION_STR']
queue_name = os.environ['SERVICE_BUS_QUEUE_NAME']
with ServiceBusClient.from_connection_string(connection_str) as sb_client:
payload = sb_client.get_queue_receiver(queue_name).receive_messages()
## deserialize payload into a list of typed Events
events = [EventGridEvent.from_json(msg) for msg in payload]
for event in events:
print(type(event)) ## EventGridEvent