import pytest

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


@pytest.mark.asyncio
async def test_mv1(results_as_df):
    expected_columns = [
        "facts_prod_staging_raw_youtube_video_report.channel_display_name",
        "facts_prod_staging_raw_youtube_video_report.time_uploaded",
        "facts_prod_staging_raw_youtube_video_report.video_title",
        "facts_prod_staging_raw_youtube_video_report.video_id",
        "facts_prod_staging_raw_youtube_video_report.video_privacy_status",
        "facts_prod_staging_raw_youtube_video_report.video_length",
        "facts_prod_dim_label.label_name",
        "facts_prod_staging_raw_youtube_video_report.isrc",
        "facts_prod_staging_raw_youtube_video_report.custom_id",
        "facts_prod_staging_raw_youtube_video_report.asset_id",
        "facts_prod_staging_raw_youtube_video_report.other_owners_claiming",
        "facts_prod_staging_raw_youtube_video_report.views",
    ]
    result = await results_as_df(mv.mv1)
    assert result.columns.tolist() == expected_columns


@pytest.mark.asyncio
async def test_mv2(results_as_df):
    expected_columns = [
        "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.conflicting_owner",
        "yt_asset_report.conflicting_territories",
    ]
    result = await results_as_df(mv.mv2)
    assert result.columns.tolist() == expected_columns