# pylint: disable=unused-argument,protected-access,too-many-arguments
import pytest

from slz_spotify_charts_data_extender.entities import Payload
from slz_spotify_charts_data_extender.utils import get_lambda_payload_from_event


@pytest.mark.parametrize(
    'event, expected', [
        (
            {
                'report_type': 'charts_weekly_viral',
                'uow_id': 'some uow id',
                'dsp': 'spotify',
                'version': 'v1',
                'report_date': '2021-01-02',
                'licensor': 'sme',
                'info': {
                    'execution_name': 'job 1'
                },
            },
            Payload(
                uow_id='some uow id',
                dsp='spotify',
                report_type='charts_weekly_viral',
                application='charts',
                version='v1',
                licensor='sme',
                job_id='job 1',
                report_date='2021-01-02',
                recurrence_type='weekly',
                chart_type='viral',
            )
        )
    ]
)
def test_get_lambda_payload_from_event(event, expected):
    got = get_lambda_payload_from_event(event)
    assert got == expected