from src.search.ranking import rank_search_results


def test_news_queries_prefer_news_sources_over_sports_and_social_results():
    results = [
        {
            "title": "Chicago Stars fire GM Richard Feuz",
            "url": "https://www.reuters.com/sports/soccer/chicago-stars-fire-gm-richard-feuz--flm-2026-05-27/",
            "snippet": "The Chicago Stars fired their general manager.",
        },
        {
            "title": "United States Eliminates Canada In Quarterfinals",
            "url": "https://sports.yahoo.com/articles/united-states-vs-canada-live-updates-170747222.html",
            "snippet": "United States eliminated Canada in hockey.",
        },
        {
            "title": "Canada - AP News",
            "url": "https://apnews.com/hub/canada",
            "snippet": "Stay up to date on the latest Canada news coverage from AP News.",
        },
        {
            "title": "CBC News - Canada",
            "url": "https://www.cbc.ca/news/canada",
            "snippet": "Your source for Canadian news in English.",
        },
        {
            "title": "CTV News - Canada",
            "url": "https://www.ctvnews.ca/canada",
            "snippet": "Latest news, travel, politics, money, jobs and more.",
        },
    ]

    ranked = rank_search_results("Canada news today", results)
    top_urls = [item["url"] for item in ranked[:3]]

    assert "https://apnews.com/hub/canada" in top_urls
    assert "https://www.cbc.ca/news/canada" in top_urls
    assert "https://www.ctvnews.ca/canada" in top_urls
    assert ranked[-1]["url"].startswith("https://sports.yahoo.com/")
