def test_get_bytes_storage_queue_wrong_content():
string = u'This is a random string which must fail'
obj = MockQueueMessage(content=string)
with pytest.raises(ValueError,
match="Failed to load JSON content from the object."):
_get_json_content(obj)
def test_get_bytes_random_obj():
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"}'
random_obj = {
"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",
}
assert _get_json_content(json_str) == random_obj
def test_get_bytes_servicebus_wrong_content():
obj = MockServiceBusReceivedMessage(
body=MockBody(data='random'),
message_id='3f6c5441-5be5-4f33-80c3-3ffeb6a090ce',
content_type='application/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')
with pytest.raises(ValueError,
match="Failed to load JSON content from the object."):
dict = _get_json_content(obj)
def test_get_bytes_servicebus():
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')
dict = _get_json_content(obj)
assert dict.get('data') == "ServiceBus"
assert dict.get('data_version') == '1.0'
def test_get_bytes_storage_queue():
cloud_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=cloud_storage_dict)
dict = _get_json_content(obj)
assert dict.get('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"
}
}
assert dict.get('data_version') == "1.0"
def test_get_bytes_eventhubs_wrong_content():
obj = MockEventhubData(body=MockEhBody(data='random string'))
with pytest.raises(ValueError,
match="Failed to load JSON content from the object."):
dict = _get_json_content(obj)
def test_get_bytes_eventhubs():
obj = MockEventhubData(body=MockEhBody())
dict = _get_json_content(obj)
assert dict.get('data') == 'Eventhub'
assert dict.get('data_version') == '1.0'