"""Fixtures for Integration test."""

import os

import boto3
import pytest


SQS_URI = os.environ.get('SQS_URI')


@pytest.fixture(scope='session')
def boto_session():
    """Boto session."""
    return boto3.Session()


@pytest.fixture(scope='session')
def sqs_client(boto_session):
    """SQS client."""
    return boto_session.client('sqs')


@pytest.yield_fixture(autouse=True, scope='function')
def purge_messages(sqs_client):
    """Purge messages from sqs queue."""
    yield
    messages = sqs_client.receive_message(QueueUrl=SQS_URI, MaxNumberOfMessages=10)
    if 'Messages' in messages:
        for message in messages['Messages']:
            sqs_client.delete_message(QueueUrl=SQS_URI, ReceiptHandle=message['ReceiptHandle'])