"""Test integration of an app with SQS."""
from flexmock import flexmock
from moto import mock_aws
from owslogger import logger

from salessheets.connectors import sqs


TEST_CORRELATION_ID = 'c305d169-455b-4bdd-ab69-0c49c49ae4fc'
TEST_RECEIVE_COUNT = 5

sqs_attributes_correlation_id = {
    sqs.CORRELATION_ID_ATTRIBUTE: {
        'StringValue': TEST_CORRELATION_ID
    }
}

sqs_attributes_receive_count = {
    sqs.APPROXIMATE_RECEIVE_COUNT_ATTRIBUTE: TEST_RECEIVE_COUNT
}


def test_get_connection():
    """Test if get_connection() returns a valid connection object."""
    with mock_aws():
        connection = sqs.get_connection(
            access_key_id='test', secret_access_key='test-key')
        assert connection is not None


def test_get_queue():
    """Test if get_queue() returns a Queue"""
    with mock_aws():
        connection = sqs.get_connection(
            access_key_id='test', secret_access_key='test-key')
        connection.create_queue(QueueName='dev-test-queue')
        queue = sqs.get_queue('dev-test-queue', sqs_connection=connection)
        assert queue is not None


def test_json_message_ext_context():
    """Test if JSONMessageExt context can store and return a test value."""
    json_message = sqs.JSONMessageExt()
    json_message.context.test_value = 'test'
    assert json_message.context.test_value == 'test'


def test_json_message_ext_correlation_id():
    """Test if JSONMessageExt can return correlation_id."""
    message = {
        'MessageAttributes': sqs_attributes_correlation_id
    }

    json_message = sqs.JSONMessageExt(message)
    assert json_message.correlation_id == TEST_CORRELATION_ID


def test_json_message_ext_logger():
    """Test if JSONMessageExt can return OwsLoggingAdapter."""
    message = {
        'MessageAttributes': sqs_attributes_correlation_id
    }

    json_message = sqs.JSONMessageExt(message)
    assert isinstance(json_message.logger, logger.OwsLoggingAdapter)


def test_json_message_ext_receive_count():
    """Test if JSONMessageExt can return approximate receive count."""
    message = {
        'MessageAttributes': sqs_attributes_correlation_id,
        sqs.APPROXIMATE_RECEIVE_COUNT_ATTRIBUTE: TEST_RECEIVE_COUNT
    }
    json_message = sqs.JSONMessageExt(message)
    assert json_message.receive_count == TEST_RECEIVE_COUNT