"""Model tests."""

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


def test_product_review_events_from_msk_message(
        mock_multi_topic_msk_message, mock_avro_serializer_id_103):
    """Test ProductEventMessage."""
    message = EventSourceMessage(mock_multi_topic_msk_message)
    for _, event in message:
        if event.topic == 'event.owsContentReview.review':
            review_event = ProductEventMessage(
                message=event.value,
                topic=event.topic,
                value_deserializer=mock_avro_serializer_id_103)
            assert review_event.message['operation']
            assert review_event.message['payload']
            assert review_event.product_id == 3930276
            assert review_event.review_queue_id == 12733
            assert review_event.operation_type == 'create'
            assert review_event.operation_context in ['reject', 'approve']
            assert review_event.submission_type in ['Revision', 'New']