def _extract_message_id(response_body: bytes) -> Optional[str]:
return safe_key_from_xml(
response_body,
"SendMessageResponse/SendMessageResult/MessageId" # Single.
) or safe_key_from_xml(
response_body,
"SendMessageBatchResponse/SendMessageBatchResult/SendMessageBatchResultEntry/0/MessageId", # Batch.
)
def parse_response(self, url: str, status_code: int, headers,
body: bytes) -> dict:
return recursive_json_join(
{
"info": {
"messageId":
safe_key_from_xml(
body, "PublishResponse/PublishResult/MessageId")
}
},
super().parse_response(url, status_code, headers, body),
)
def test_key_from_xml(input_params, expected_output):
assert safe_key_from_xml(*input_params) == expected_output