AutoScraperX/test/test_pydantic_upgrade.py

# test_pydantic_upgrade.py
# !/usr/bin/env python3
"""Pydantic 2.x 升级验证脚本"""

from config.base import settings
from core.models.video_item import VideoItem
from core.utils.spider_config import SpiderConfig


def test_settings():
    """测试配置加载"""
    print("=== 测试 Settings ===")
    print(f"环境: {settings.ENV}")
    print(f"数据库主机: {settings.DB_HOST}")
    print(f"Redis URL: {settings.redis_url}")
    print("✓ Settings 加载成功")


def test_video_item():
    """测试视频项模型"""
    print("\n=== 测试 VideoItem ===")
    video_data = {
        "user_id": "test_user",
        "user_name": "测试用户",
        "out_video_id": "12345",
        "video_url": "https://example.com/video.mp4",
        "cover_url": "https://example.com/cover.jpg",
        "platform": "test",
        "strategy": "recommend",
        "out_user_id":"123",
        "session":"2134",
        "video_title":"123"
    }

    item = VideoItem(**video_data)
    print(f"创建视频项: {item.video_id}")
    print("✓ VideoItem 创建成功")


def test_spider_config():
    """测试爬虫配置"""
    print("\n=== 测试 SpiderConfig ===")
    try:
        config = SpiderConfig.get_platform_config("benshanzhufurecommend")
        print(f"平台: {config.platform}")
        print(f"模式: {config.mode}")
        print(f"URL: {config.url}")
        print("✓ SpiderConfig 加载成功")
    except Exception as e:
        print(f"✗ SpiderConfig 加载失败: {e}")


if __name__ == "__main__":
    test_settings()
    test_video_item()
    test_spider_config()
    print("\n=== 升级验证完成 ===")