"""Model tests."""

from kafka_utils.consumer.message.product import ProductEventMessage
from kafka_utils.consumer.source.mapping import EventSourceMessage


def test_event_producer_event(mock_event_producer_event, mock_avro_serializer_id_79):
    """Test EventProducerMessage."""
    message = EventSourceMessage(
        mock_event_producer_event)
    for key, event in message:
        message = ProductEventMessage(
            message=event.value,
            topic=event.topic,
            value_deserializer=mock_avro_serializer_id_79)
        assert 'operation' in message.message
        assert 'payload' in message.message
        assert key