"""Lambda test module."""

import base64
import json

from src import app as index


def _to_msk_event(msg):
    return {
        'records': {
            'event': [{'value': base64.b64encode(json.dumps(msg).encode()).decode()}]
        }
    }


def test_handler_update(kafka_mock, fan_updated_full):
    """Test handler function for an update."""
    msk_event = _to_msk_event(fan_updated_full)
    result = index.handler(msk_event, None)
    assert result.get('status') == 'OK'


def test_handler_delete(kafka_mock, fan_deleted):
    """Test handler function for a delete."""
    msk_event = _to_msk_event(fan_deleted)
    result = index.handler(msk_event, None)
    assert result.get('status') == 'OK'


def test_handler_no_id(kafka_mock, fan_no_crmId):
    """Test handler function for message without crmId is ignored."""
    msk_event = _to_msk_event(fan_no_crmId)
    result = index.handler(msk_event, None)
    assert result.get('status') == 'IGNORED'