"""Lambda test module."""

from datetime import date, datetime

from freezegun import freeze_time
import transform


@freeze_time('2020-01-01T00:00:00.100000+00:00')
def test_transform_metadata():
    """Test metadata transformation logic."""
    test_product_metadata = {
        'productId': 12345,
        'productName': None,
        'format': None,
        'releaseDate': None,
        'genreId': None,
        'subgenreId': None,
        'tracks': []
    }
    expected_transformed_metadata = {
        'message_thread_id': 12345,
        'message_id': '12345_20200101_000000',
        'message_created_date_time': '2020-01-01T00:00:00.100000+00:00',
        'message_notification_period_start': date(2020, 1, 1),
        'message_notification_period_end': date(2020, 1, 1)
    }
    result = transform.transform_metadata(test_product_metadata)

    assert result == expected_transformed_metadata


def test_build_message_id():
    """Test that the message id is returned in the correct format."""
    test_product_id = 12345
    test_release_created_datetime = datetime(2020, 1, 1)
    message_id = transform.build_message_id(test_product_id, test_release_created_datetime)

    assert message_id == '12345_20200101_000000'