"""Model tests."""

import pytest
from lambdacommon.models.lambda_msk_event_source_message import \
    LambdaMSKEventSourceMessage
from lambdacommon.models.lambda_msk_event_source_message import \
    MSKMessageException


def test_event_source_message(mock_event):
    """Test LambdaMSKEventSourceMessage."""
    mock_event_source_message = LambdaMSKEventSourceMessage(mock_event)
    for key, event in mock_event_source_message:
        assert 'cdc.artRelations.releases-' in key
        assert 'payload' in event.value


def test_invalid_event_source_message():
    """Test LambdaMSKEventSourceMessage raises with an invalid event."""
    with pytest.raises(MSKMessageException):
        LambdaMSKEventSourceMessage({})


def test_empty_event_source_message(mock_empty_event):
    """Test generator returns None for empty messages."""
    mock_event_source_message = LambdaMSKEventSourceMessage(mock_empty_event)
    for key, event in mock_event_source_message:
        assert 'cdc.artRelations.releases-' in key
        assert event.value is None