"""Test for Kafka Producer."""

import json

from config import KAFKA_TARGET_TOPIC
from src.connectors.kafka_producer import produce_for_id


def test_produce_for_id(kafka_mock):
    """Test Kafka Producing message."""
    message = {'key': 'value'}
    id = 'ID_1'
    produce_for_id(message, id)
    kafka_mock.assert_call_count('produce', 1)
    kafka_mock.assert_messages([{
        'topic': KAFKA_TARGET_TOPIC,
        'value': json.dumps(message).encode(),
        'key': id.encode()
    }])