"""Lambda test module."""

from unittest.mock import patch

from kafka_utils.consumer.deserializer.simple_json import JSONDeserializer
from kafka_utils.consumer.source.mapping import EventSourceMessage

from src.constants.graphql import EMAIL_NOTIFICATION_MUTATION
from src.logic.event_handling import processing_logic


@patch('src.logic.event_handling.gql_conn.execute')
def test_processing_logic(
    mock_gql,
    mock_event,
    mock_graphql_response
):
    """Test processing logic."""
    mock_gql.return_value = mock_graphql_response
    msk_message = EventSourceMessage(mock_event)
    mock_message = None
    for _, message in msk_message:
        mock_message = message.value
        break
    response = processing_logic(
        'event.owsContentReview.review',
        mock_message,
        JSONDeserializer()
    )
    assert response == mock_graphql_response
    mock_gql.assert_called_once_with(
        EMAIL_NOTIFICATION_MUTATION, {'reviewQueueId': 10168})