"""Lambda test module."""
from unittest.mock import MagicMock

import pytest
from src.common.app_logic import _get_records
from src.common.app_logic import _is_response_successful


@pytest.fixture
def msk_event():
    """MSK event example."""
    return {'records': {'my-topic': [{'value': 'eyJ0ZXN0IjogInZhbHVlIn0='}]}}


def test_get_records(msk_event):
    """Test _get_records helper function."""
    records = list(_get_records(msk_event))
    assert records == [{'test': 'value'}]


@pytest.mark.parametrize(
    'response, expected_result',
    [
        (MagicMock(status_code=200), True),
        (MagicMock(status_code=500), False),
    ]
)
def test_is_response_successful(response, expected_result):
    """Test _is_response_successful helper."""
    assert _is_response_successful(response) == expected_result