import pytest

from dapd_exporter_lambda.utils import is_manifest_file


@pytest.mark.parametrize(
    'filename, expected', [
        (
            's3://data_source/_success',
            True,
        ),
        (
            's3://data_source/_success_pg',
            True,
        ),
        (
            's3://data_source/_success_bt',
            False,
        ),
        (
            's3://data_source/dim_artist.parquet',
            False,
        ),
        (
            's3://data_source/_success_success',
            False,
        ),
        (
            's3://data_source/success',
            False,
        ),
    ]
)
def test_is_manifest_file(filename, expected):
    assert is_manifest_file(filename) == expected