import pytest

from src.backend.logic.look_data.looks import sr_ugc


@pytest.mark.asyncio
@pytest.mark.slow
async def test_sr_ugc1(results_as_df):
    expected_columns = [
        "facts_prod_staging_raw_youtube_claim.asset_id",
        "facts_prod_staging_raw_youtube_claim.isrc",
        "facts_prod_staging_raw_youtube_claim.video_id",
        "facts_prod_staging_raw_youtube_claim.views",
        "facts_prod_staging_raw_youtube_claim.claim_type",
        "dim_track.trackname",
        "dim_artists.track_artists",
    ]
    result = await results_as_df(sr_ugc.sr_ugc1)
    assert result.columns.tolist() == expected_columns


@pytest.mark.asyncio
@pytest.mark.slow
async def test_sr_ugc2(results_as_df):
    expected_columns = [
        "facts_prod_staging_raw_youtube_claim.asset_id",
        "total_videos",
        "total_claims",
        "total_views",
    ]
    result = await results_as_df(sr_ugc.sr_ugc2)
    assert result.columns.tolist() == expected_columns


@pytest.mark.asyncio
async def test_sr_ugc3(results_as_df):
    expected_columns = ["dim_track.isrc", "yt_asset_report.recent_daily_average"]
    result = await results_as_df(sr_ugc.sr_ugc3)
    assert result.columns.tolist() == expected_columns