def test_log_deleted_message(queue_url, caplog):
sqs = SqsReceiver(queue_url=queue_url)
message_id = '96427844-a8c4-41e5-beed-dab5780cc681'
receipt_handle = (
'AQEBY/+I4IQJzCkjsop/FaaKLb5VKXL9JHF4BWbJwOD0osZbenehbjp4xQ59r1DD0pEonhUCaILKkQfq7FrPhUomHBzOzK+e4qCGI62QgkLt6V'
'8vYJtHlhSNrroy+c9LyuZFKtVNJQ/8rjFwEDQ/jnFdA6MLGnG2Aaj+8LTdvXb4aG+r1PwkS4AEEa4IvMty5NJOwmd70QynfopkRO18EGe6pKkE'
'6d0j1XvZNkd7peisPLxwU9cljjyu+Hdds12atn71Ss7nn/jr2ElhsnzJrB3eQ4XCtjYP/9sfXT8+0BT5SPb2zCIe1bDzXCLxZypikkdEkQq33z'
'/TtjWk0/b8yyReM4o52m83tYP6fB0XNlQnWebpO/9leappxF5mFXl7kO2N4ga0dBJ0ts+I/j4rn6GP5w=='
)
message = SqsMessage(message_id=message_id,
receipt_handle=receipt_handle,
payload='hello world!',
payload_type=PayloadType.PLAINTEXT)
with caplog.at_level(logging.DEBUG, logger=logger.name):
with vcr.use_cassette('delete_message_that_exists'):
sqs.delete_message(message)
records = caplog.records
assert len(records) == 2
assert records[0].levelno == logging.INFO
assert records[0].message == '[{}][{}] Deleted message'.format(
sqs.queue_url, message_id)
assert records[1].levelno == logging.DEBUG
assert records[1].message == '[{}][{}] ReceiptHandle={}'.format(
sqs.queue_url, message_id, receipt_handle)
def test_delete_message_that_has_already_been_deleted(queue_url):
sqs = SqsReceiver(queue_url=queue_url)
message_id = '96427844-a8c4-41e5-beed-dab5780cc681'
receipt_handle = (
'AQEBY/+I4IQJzCkjsop/FaaKLb5VKXL9JHF4BWbJwOD0osZbenehbjp4xQ59r1DD0pEonhUCaILKkQfq7FrPhUomHBzOzK+e4qCGI62QgkLt6V'
'8vYJtHlhSNrroy+c9LyuZFKtVNJQ/8rjFwEDQ/jnFdA6MLGnG2Aaj+8LTdvXb4aG+r1PwkS4AEEa4IvMty5NJOwmd70QynfopkRO18EGe6pKkE'
'6d0j1XvZNkd7peisPLxwU9cljjyu+Hdds12atn71Ss7nn/jr2ElhsnzJrB3eQ4XCtjYP/9sfXT8+0BT5SPb2zCIe1bDzXCLxZypikkdEkQq33z'
'/TtjWk0/b8yyReM4o52m83tYP6fB0XNlQnWebpO/9leappxF5mFXl7kO2N4ga0dBJ0ts+I/j4rn6GP5w=='
)
message = SqsMessage(message_id=message_id,
receipt_handle=receipt_handle,
payload='hello world!',
payload_type=PayloadType.PLAINTEXT)
with vcr.use_cassette(
'delete_message_that_has_already_been_deleted') as cassette:
sqs.delete_message(message)
sqs.delete_message(message)
assert cassette.all_played
request = get_sqs_request_data(cassette)
assert get_request_attribute(request, 'ReceiptHandle') == receipt_handle
assert get_queue_url_from_request(request) == sqs.queue_url
def test_receive_many_messages_with_mixed_payloads(queue_url):
sqs = SqsReceiver(queue_url=queue_url, max_number_of_messages=10)
with vcr.use_cassette(
'receive_many_messages_with_mixed_payloads') as cassette:
messages = sqs.receive_messages()
assert cassette.all_played
response = get_sqs_receive_message_response_data(cassette)
assert len(messages) == len(response['Message']) == 7
for i, message in enumerate(messages):
assert_is_sqs_message(message)
message_from_response = response['Message'][i]
assert message.message_id == message_from_response['MessageId']
assert message.receipt_handle == message_from_response['ReceiptHandle']
deserialized_response = deserialize_sqs_message(message_from_response)
assert message.payload == deserialized_response.payload
assert message.payload_type == deserialized_response.payload_type
assert_request_match_sqs_receiver_attributes(cassette, sqs)
def test_receive_message_with_base64_encoded_binary_payload(queue_url):
sqs = SqsReceiver(queue_url=queue_url, max_number_of_messages=1)
with vcr.use_cassette(
'receive_message_with_base64_encoded_binary_payload') as cassette:
messages = sqs.receive_messages()
assert cassette.all_played
response = get_sqs_receive_message_response_data(cassette)
assert len(messages) == 1
message = messages[0]
assert_is_sqs_message(message)
assert_message_attributes_match_response(message, response)
assert message.payload_type == PayloadType.BINARY
assert message.payload == b64decode(response['Message']['Body'])
assert message.embedded_message is None
assert_request_match_sqs_receiver_attributes(cassette, sqs)
def test_receive_message_with_plaintext_mega_payload(queue_url):
sqs = SqsReceiver(queue_url=queue_url, max_number_of_messages=1)
with vcr.use_cassette(
'receive_message_with_plaintext_mega_payload') as cassette:
messages = sqs.receive_messages()
assert cassette.all_played
response = get_sqs_receive_message_response_data(cassette)
assert len(messages) == 1
message = messages[0]
assert_is_sqs_message(message)
assert_message_attributes_match_response(message, response)
assert message.payload_type == PayloadType.MEGA
assert message.payload == deserialize_payload(
json.loads(response['Message']['Body']))
assert message.embedded_message is None
assert_request_match_sqs_receiver_attributes(cassette, sqs)
def test_log_received_messages(queue_url, caplog):
sqs = SqsReceiver(queue_url=queue_url, max_number_of_messages=1)
with caplog.at_level(logging.DEBUG, logger=logger.name):
with vcr.use_cassette(
'receive_message_with_plaintext_json_payload') as cassette:
sqs.receive_messages()
response = get_sqs_receive_message_response_data(cassette)
message_id = response['Message']['MessageId']
message_body = response['Message']['Body']
records = caplog.records
assert len(records) == 3
assert records[0].levelno == logging.DEBUG
assert records[0].message == '[{}] Querying messages. ' \
'MaxNumberOfMessages=1; WaitTimeSeconds=20; VisibilityTimeout=30'.format(sqs.queue_url)
assert records[1].levelno == logging.INFO
assert records[1].message == '[{}][{}] Received message'.format(
sqs.queue_url, message_id)
assert records[2].levelno == logging.DEBUG
assert records[2].message == '[{}][{}] {}'.format(sqs.queue_url,
message_id, message_body)
def test_receive_message_with_base64_encoded_binary_mega_payload_over_sns(
queue_url):
sqs = SqsReceiver(queue_url=queue_url, max_number_of_messages=1)
with vcr.use_cassette(
'receive_message_with_base64_encoded_binary_mega_payload_over_sns'
) as cassette:
messages = sqs.receive_messages()
assert cassette.all_played
response = get_sqs_receive_message_response_data(cassette)
assert len(messages) == 1
message = messages[0]
assert_is_sqs_message(message)
assert_message_attributes_match_response(message, response)
assert message.payload_type == PayloadType.MEGA
assert message.payload == deserialize_payload(
bson.loads(
b64decode(json.loads(response['Message']['Body'])['Message'])))
assert_is_sns_notification(message.embedded_message)
assert_request_match_sqs_receiver_attributes(cassette, sqs)