content-find-agent-new/tests/test_platform_access.py

from content_agent.business_modules import platform_access


class DuplicateVideoClient:
    def search(self, query):
        return [
            {
                "candidate_id": f"{query['query_id']}_c_001",
                "query_id": query["query_id"],
                "aweme_id": "7601814454925298994",
                "desc": "重复视频",
            }
        ]


def test_platform_access_deduplicates_same_aweme_across_queries():
    queries = [
        {
            "query_id": "q_001",
            "query": "基层公职人员贪腐案例",
            "generation_type": "item_combo",
        },
        {
            "query_id": "q_002",
            "query": "综合性腐败 警示案例",
            "generation_type": "light_extension",
        },
    ]

    results = platform_access.run(queries, DuplicateVideoClient())

    assert len(results) == 1
    assert results[0]["query_id"] == "q_001"
    assert results[0]["query"] == "基层公职人员贪腐案例"