import os

import boto3
import pytest
from moto import mock_secretsmanager


ENV_VARS = ['RECEIPTS_QUEUE', 'SENTRY_DSN']


@pytest.fixture
def aws_credentials():
    """Mocked AWS Credentials for boto3."""
    os.environ['AWS_ACCESS_KEY_ID'] = 'testing'
    os.environ['AWS_SECRET_ACCESS_KEY'] = 'testing'
    os.environ['AWS_SECURITY_TOKEN'] = 'testing'
    os.environ['AWS_SESSION_TOKEN'] = 'testing'


@pytest.fixture
def mocked_secrets_manager(aws_credentials):
    """Mocked AWS Secret Manager."""
    with mock_secretsmanager():
        secrets_manager = boto3.client(
            'secretsmanager', region_name='us-east-1')
        for env in ENV_VARS:
            value = os.environ[env]
            secrets_manager.create_secret(Name=value, SecretString=value)
        yield secrets_manager