"""Test for the Chart Handler."""
import datetime
from unittest.mock import patch

from charts.logic import track


def test_get_track_aggregated_rankings_by_public_sound_recording_id(mock_get_all_feature_flags):
    """Test get_charts."""

    mock_fetchall_result = [
        ('8068b0d9-67a4-46d2-aebe-7c7199437c76', '72b1e864-8ec2-4cc2-b700-94729f7484a4',
         'b1e864-8ec2-4cc2-b700-94729f7484', 'B027514'),
        ('a010c5ea-351c-4a13-a84f-6117b447b869', '72b1e864-8ec2-4cc2-b700-94729f7484a4',
         None, '8ec2-4cc2-b700-94729f7484'),
    ]
    expected_response = [
        {
            'chartId': '8068b0d9-67a4-46d2-aebe-7c7199437c76',
            'publicSoundRecordingId': '72b1e864-8ec2-4cc2-b700-94729f7484a4',
            'publicProductId': 'b1e864-8ec2-4cc2-b700-94729f7484',
            'trackId': 'B027514',
        },
        {
            'chartId': 'a010c5ea-351c-4a13-a84f-6117b447b869',
            'publicSoundRecordingId': '72b1e864-8ec2-4cc2-b700-94729f7484a4',
            'publicProductId': None,
            'trackId': '8ec2-4cc2-b700-94729f7484',
        },
    ]

    with patch.object(track.snowflake, 'fetchall', return_value=mock_fetchall_result):
        response = track.get_track_aggregated_rankings(
            public_sound_recording_id='72b1e864-8ec2-4cc2-b700-94729f7484a4', appeared_on_date=None
        )
        assert response == expected_response