"""Fixtures for lambda testing."""

import pytest
from kafka import KafkaConsumer
from kafka import KafkaProducer
import config  # noqa:I202

import os

TEST_EVENT = 'tests/sample_event.json'
AUTH0_URL = 'https://qa-orchard.auth0.com/oauth/token'
AUDIENCE_URL = 'https://qa-gda-gateway-auth0-jwt-authorizer'
GROUP_ID = '123456c-2b7e-4a74-98ce-d1a0dd22d360'
CLIENT_ID = 'qa-integration-test'
BOOTSTRAP_SERVERS = os.environ.get('BOOTSTRAP_SERVERS')


@pytest.fixture()
def get_consumer():
    """Initiate Consumer factory."""
    def _get_consumer(topic_name):
        """Initiate Kafka consumer."""
        return KafkaConsumer(topic_name, **{
            'bootstrap_servers': config.BOOTSTRAP_SERVERS,
            'security_protocol': 'SSL',
            'client_id': CLIENT_ID,
            'group_id': GROUP_ID,
            'auto_offset_reset': 'latest',
            'consumer_timeout_ms': 1000
        })

    return _get_consumer


@pytest.fixture()
def get_producer():
    """Initiate Kafka producer."""
    return KafkaProducer(
        client_id=CLIENT_ID,
        bootstrap_servers=BOOTSTRAP_SERVERS,
        security_protocol='SSL',
        linger_ms=0,
        api_version='2.6.0'
    )