Viral Shorts

"""YouTube 模块单元测试"""
import pytest
from datetime import datetime, timezone, timedelta
from src.models.video import VideoData
from src.youtube.analyzer import ViralAnalyzer


class TestViralAnalyzer:
    """爆款分析器测试"""
    
    def test_calculate_hours_since_publish(self):
        """测试发布时长计算"""
        # 2 小时前发布
        published_at = datetime.now(timezone.utc) - timedelta(hours=2)
        hours = ViralAnalyzer.calculate_hours_since_publish(published_at)
        
        assert 1.9 <= hours <= 2.1  # 允许小误差
    
    def test_calculate_vph(self):
        """测试 VPH 计算"""
        # 10,000 播放量,2 小时
        vph = ViralAnalyzer.calculate_vph(10000, 2)
        assert vph == 5000
        
        # 边界情况:0 小时
        vph = ViralAnalyzer.calculate_vph(1000, 0)
        assert vph == 10000  # 至少算作 0.1 小时
    
    def test_calculate_engagement_rate(self):
        """测试互动率计算"""
        # 1000 播放,50 点赞,10 评论
        rate = ViralAnalyzer.calculate_engagement_rate(1000, 50, 10)
        assert rate == 6.0  # (50+10)/1000 * 100 = 6%
        
        # 边界情况:0 播放
        rate = ViralAnalyzer.calculate_engagement_rate(0, 10, 5)
        assert rate == 0.0
    
    def test_calculate_viral_score(self):
        """测试爆款指数计算"""
        # VPH=5000, 2小时前, 互动率=5%
        score = ViralAnalyzer.calculate_viral_score(5000, 2, 5)
        
        # 时效权重 = 1/(2+1) = 0.333
        # 互动权重 = 1 + (5/100*10) = 1.5
        # score = 5000 * 0.333 * 1.5 ≈ 2500
        assert 2400 <= score <= 2600
    
    def test_categorize_potential(self):
        """测试潜力等级分类"""
        assert "超级爆款" in ViralAnalyzer.categorize_potential(15000)
        assert "高潜力" in ViralAnalyzer.categorize_potential(6000)
        assert "潜力视频" in ViralAnalyzer.categorize_potential(2000)
        assert "普通视频" in ViralAnalyzer.categorize_potential(500)
    
    def test_analyze_video(self):
        """测试视频分析"""
        # 创建测试视频
        video = VideoData(
            video_id="test123",
            title="测试视频",
            channel_name="测试频道",
            channel_id="channel123",
            views=10000,
            likes=500,
            comments=100,
            published_at=datetime.now(timezone.utc) - timedelta(hours=2),
            duration="PT30S",
            url="https://youtube.com/shorts/test123"
        )
        
        # 分析视频
        analyzed = ViralAnalyzer.analyze_video(video)
        
        # 验证指标已填充
        assert analyzed.vph > 0
        assert analyzed.engagement_rate > 0
        assert analyzed.viral_score > 0
        assert analyzed.hours_since_publish > 0
    
    def test_rank_videos(self):
        """测试视频排序"""
        videos = [
            VideoData(
                video_id=f"test{i}",
                title=f"视频{i}",
                channel_name="频道",
                channel_id="ch1",
                views=1000,
                likes=10,
                comments=5,
                published_at=datetime.now(timezone.utc),
                duration="PT30S",
                url=f"https://youtube.com/shorts/test{i}",
                viral_score=float(i * 100)
            )
            for i in range(5)
        ]
        
        ranked = ViralAnalyzer.rank_videos(videos)
        
        # 验证降序排列
        assert ranked[0].viral_score == 400
        assert ranked[-1].viral_score == 0
    
    def test_filter_by_vph(self):
        """测试 VPH 筛选"""
        videos = [
            VideoData(
                video_id=f"test{i}",
                title=f"视频{i}",
                channel_name="频道",
                channel_id="ch1",
                views=1000,
                likes=10,
                comments=5,
                published_at=datetime.now(timezone.utc),
                duration="PT30S",
                url=f"https://youtube.com/shorts/test{i}",
                vph=float(i * 1000)
            )
            for i in range(5)
        ]
        
        filtered = ViralAnalyzer.filter_by_vph(videos, 2000)
        
        # 应该只保留 VPH >= 2000 的视频
        assert len(filtered) == 3  # vph=2000, 3000, 4000
        assert all(v.vph >= 2000 for v in filtered)


class TestVideoData:
    """视频数据模型测试"""
    
    def test_to_markdown_row(self):
        """测试 Markdown 行生成"""
        video = VideoData(
            video_id="test123",
            title="测试视频标题",
            channel_name="测试频道",
            channel_id="ch1",
            views=10000,
            likes=500,
            comments=100,
            published_at=datetime.now(timezone.utc),
            duration="PT30S",
            url="https://youtube.com/shorts/test123",
            vph=5000,
            engagement_rate=6.0,
            viral_score=2500,
            hours_since_publish=2.0
        )
        
        row = video.to_markdown_row()
        
        assert "测试视频标题" in row
        assert "测试频道" in row
        assert "10,000" in row
        assert "5,000" in row
        assert "6.00%" in row
        assert "2,500" in row
    
    def test_markdown_header(self):
        """测试 Markdown 表头"""
        header = VideoData.markdown_header()
        
        assert "标题" in header
        assert "VPH" in header
        assert "互动率" in header
        assert "爆款指数" in header


if __name__ == "__main__":
    pytest.main([__file__, "-v"])