"""Test integration of an app with SQS.""" from unittest import mock import boto3 from moto import mock_sqs from sales.connectors import sqs @mock_sqs def test_get_queue_success(): """Test if get_queue() returns a Queue.""" response = boto3.client('sqs').create_queue(QueueName='test_queue_name') queue = sqs.get_queue(response['QueueUrl']) assert queue.url == response['QueueUrl'] @mock.patch('boto3.resource') def test_get_queue_failure(mock_boto_resource): """Test failure if get_queue() returns None.""" mock_boto_resource.return_value = None queue = sqs.get_queue('test_queue_url') assert queue is None