"""Kafka producer."""

from unittest.mock import MagicMock

from src import config
from confluent_kafka.serialization import StringSerializer
from kafka_utils.producer.event import EventProducer


def setup_producer():
    """Return MSK producer."""
    if config.ENV == 'test':
        return MagicMock()
    return EventProducer(
        config.BOOTSTRAP_SERVERS,
        StringSerializer(),
        StringSerializer(),
        'SSL')


kafka_producer = setup_producer()