"""Tests for warming event handler module."""

import pytest

from lambdacommon.warmer_util import catch_warmer_event


@pytest.mark.parametrize(
    'input_event, warming_event, expected_result', [
        ({}, {}, {}), ({}, None, {}), ({'1': '1'}, None, {'1': '1'}),
        ({'1': '1'}, {'2': '2'}, {'1': '1'}), ({'1': '1'}, {'1': '1'}, {})
    ])
def test_get_transcoding_type(input_event, warming_event, expected_result):
    """Test for successful call of catch_warmer_event decorator."""
    # data preparation
    @catch_warmer_event(warming_event=warming_event)
    def fake_handler(event, context):
        return event
    # test function call
    result = fake_handler(input_event, None)
    # checking
    assert expected_result == result