"""Unit tests for Utils."""

from yt_conflict_elasticsearch.util import file_util


def test_get_s3_csv_gz_file_key():
    """Test get s3 url."""
    file_key = file_util.get_s3_csv_gz_file_key('2018-01-01', 'resolved')
    assert file_key == 'folder/2018_01_01/file_resolved.csv.gz'


def test_get_json_filename():
    """Test JSON filename."""
    file_key = file_util.get_json_filename('resolved')
    assert file_key == 'file_resolved.json'


def test_get_json_filename_with_gz():
    """Test get gzipped JSON filename."""
    file_key = file_util.get_json_filename('resolved', gzipped=True)
    assert file_key == 'file_resolved.json.gz'


def test_get_s3_json_gz_file_key():
    """Test get s3 JSON gz file key."""
    file_key = file_util.get_s3_json_gz_file_key('2018-01-01', 'resolved')
    assert file_key == 'folder/2018_01_01/file_resolved.json.gz'