import pytest

from slz_downloader.worker.appreciationengine import Worker as AEWorker


@pytest.mark.parametrize(
    'name,expected', [
        (
            'US_Columbia_20201104_20210605010101.json',
            'US_Columbia_20201104.json',
        ),
        (
            'US_Columbia_20201104_20210605010101.json.gz',
            'US_Columbia_20201104.json.gz',
        ),
        (
            'US_Columbia_20201104_20210605.json.gz',
            'US_Columbia_20201104_20210605.json.gz',
        ),
        (
            'US_Columbia_20201104.json.gz',
            'US_Columbia_20201104.json.gz',
        ),
    ]
)
def test_replace(name, expected):
    actual = AEWorker.remove_timestamp(name)
    assert actual == expected