import pytest

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


@pytest.mark.asyncio
async def test_at1(results_as_df):
    expected_columns = [
        "releases_snowflake.upc",
        "track_snowflake.isrc",
        "track_snowflake.track_name",
        "artist_info_snowflake.name",
        "releases_snowflake.release_name",
        "genre_snowflake.genre",
        "track_snowflake.duration",
        "vendor_snowflake.vendor_name",
        "releases_snowflake.is_active_release",
        "delivery_history_snowflake_aggregated.delivered_status",
    ]
    result = await results_as_df(at.at1)
    assert result.columns.tolist() == expected_columns


@pytest.mark.asyncio
async def test_at2(results_as_df):
    expected_columns = [
        "releases_snowflake.upc",
    ]
    result = await results_as_df(at.at2)
    assert result.columns.tolist() == expected_columns


@pytest.mark.asyncio
async def test_at3(results_as_df):
    expected_columns = [
        "releases_snowflake.upc",
        "vtr_country.vtr_countries",
        "rtr_country.rtr_countries",
        "str_country.str_countries",
    ]
    result = await results_as_df(at.at3)
    assert result.columns.tolist() == expected_columns


@pytest.mark.asyncio
async def test_at4(results_as_df):
    expected_columns = [
        "dim_track.UPC",
        "yt_asset_report.isrc",
        "yt_asset_report.asset_id",
        "yt_asset_report.ownership",
        "yt_asset_report.artist",
    ]
    result = await results_as_df(at.at4)
    assert result.columns.tolist() == expected_columns


@pytest.mark.asyncio
async def test_at5(results_as_df):
    expected_columns = [
        "yt_conflict_report_final.asset_id",
        "yt_conflict_report_final.conflicting_owner",
    ]
    result = await results_as_df(at.at5)
    assert result.columns.tolist() == expected_columns


@pytest.mark.asyncio
async def test_at6(results_as_df):
    expected_columns = [
        "facts_prod_staging_raw_youtube_claim.asset_id",
        "facts_prod_staging_raw_youtube_claim.channel_display_name",
        "facts_prod_staging_raw_youtube_claim.content_type",
        "facts_prod_staging_raw_youtube_claim.claim_type",
    ]
    result = await results_as_df(at.at6)
    assert result.columns.tolist() == expected_columns