"""Unit tests for the get_date_range task."""
from freezegun import freeze_time

from activity_detector.flows.trending_tracks.tasks.get_date_range \
    import get_date_range


@freeze_time('2018-10-14')
def test_get_date_range(monkeypatch):
    """Test getting the date range."""
    result = get_date_range(None)

    assert result == {
        'top_ten_start_date': '2018-10-05',
        'top_ten_end_date': '2018-10-11',
        'all_track_start_date': '2018-08-02',
        'all_track_end_date': '2018-10-11'
    }