import pytest

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


@pytest.mark.asyncio
async def test_sr1(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.is_delivered",
    ]
    result = await results_as_df(sr.sr1)
    assert result.columns.tolist() == expected_columns


@pytest.mark.asyncio
async def test_sr2(results_as_df):
    expected_columns = ["releases_snowflake.upc", "releases_snowflake.release_name"]
    result = await results_as_df(sr.sr2)
    assert result.columns.tolist() == expected_columns


@pytest.mark.asyncio
async def test_sr3(results_as_df):
    expected_columns = ["yt_registry.isrc", "yt_registry.territory_list"]
    result = await results_as_df(sr.sr3)
    assert result.columns.tolist() == expected_columns


@pytest.mark.asyncio
async def test_sr4(results_as_df):
    expected_columns = [
        "orch_app_ar_track.isrc",
        "facts_prod_registry_locked.lock_reason",
    ]
    result = await results_as_df(sr.sr4)
    assert result.columns.tolist() == expected_columns


@pytest.mark.asyncio
async def test_sr5(results_as_df):
    expected_columns = [
        "int_prod_internal_conflicts.isrc",
        "int_prod_internal_conflicts.matched_label_name",
        "int_prod_internal_conflicts.list_conflicting_territories",
    ]
    result = await results_as_df(sr.sr5)
    assert result.columns.tolist() == expected_columns


@pytest.mark.asyncio
async def test_sr6(results_as_df):
    expected_columns = ["releases_snowflake.upc", "vtr_country.abbrivation"]
    result = await results_as_df(sr.sr6)
    assert result.columns.tolist() == expected_columns


@pytest.mark.asyncio
async def test_sr7(results_as_df):
    expected_columns = [
        "yt_asset_report.isrc",
        "yt_asset_report.asset_id",
        "yt_asset_report.active_reference_ids",
        "yt_asset_report.inactive_reference_ids",
        "yt_asset_report.match_policy",
        "yt_asset_report.ownership",
        "yt_asset_report.song",
    ]
    result = await results_as_df(sr.sr7)
    assert result.columns.tolist() == expected_columns


@pytest.mark.asyncio
async def test_sr8(results_as_df):
    expected_columns = [
        "dim_track.isrc",
        "yt_conflict_report_final.conflicting_owners",
        "yt_conflict_report_final.territory_list",
    ]
    result = await results_as_df(sr.sr8)
    assert result.columns.tolist() == expected_columns