"""Test for Kafka Producer."""

import json

import pytest
from src.connectors.kafka_producer import produce_for_id


def test_produce_for_id(kafka_mock):
    """Test Kafka Producing message."""
    message = {'key': 'value'}
    key = 'ID_1'
    topic = 'my-topic'
    produce_for_id(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_produce_for_id_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):
        produce_for_id(message, key, topic)
        kafka_mock.assert_call_count('produce', 1)