"""Fixtures for lambda testing."""
import json

import boto3
import pytest


@pytest.fixture()
def invoke_lambda():
    """Invoke the lambda function."""
    client = boto3.client('lambda', region_name='us-east-1')
    return client.invoke(
        FunctionName='qa-lambda-av-scan-containerized',
        InvocationType='RequestResponse',
        LogType='None',
        Payload=json.dumps({
            'detail': {
                'requestParameters': {
                    'bucketName': 'qa-lambda-av-scan-containerized-integration-test',
                    'key': 'safe.pdf'
                }
            }
        }),
    )


@pytest.fixture()
def invoke_lambda_virus():
    """Invoke the lambda function."""
    client = boto3.client('lambda', region_name='us-east-1')
    return client.invoke(
        FunctionName='qa-lambda-av-scan-containerized',
        InvocationType='RequestResponse',
        LogType='None',
        Payload=json.dumps({
            'detail': {
                'requestParameters': {
                    'bucketName': 'qa-lambda-av-scan-containerized-integration-test',
                    'key': 'eicar.pdf'
                },
            },
            'delete': False
        }),
    )