"""Test for Kafka Producer."""

import json

import pytest

from src.connectors.kafka_producer import send_message_to_kaka_topic


def test_send_message_to_kaka_topic(kafka_mock):
    """Test Kafka Producing message."""
    message = {'key': 'value'}
    key = 'ID_1'
    topic = 'my-topic'
    send_message_to_kaka_topic(message, key, topic)
    kafka_mock.assert_call_count('produce', 1)
    kafka_mock.assert_messages([{
        'topic': topic,
        'value': json.dumps(message).encode(),
        'key': key.encode()
    }])


def test_send_message_to_kaka_topic_error(kafka_mock):
    """Test Kafka Producing message."""
    message = {'key': 'value'}
    key = 'ID_1'
    topic = 'my-topic'
    kafka_mock.set_return_error(True)
    with pytest.raises(Exception):
        send_message_to_kaka_topic(message, key, topic)
        kafka_mock.assert_call_count('produce', 1)