#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
工单查询增强功能测试脚本
"""
import json
from pathlib import Path

def test_config_loading():
    """测试配置加载功能"""
    print("=" * 60)
    print("测试1: 配置加载")
    print("=" * 60)
    
    from work_order_query import load_db_config, DEFAULT_CONFIG
    
    config = load_db_config()
    
    # 验证target_process_name字段存在
    assert 'target_process_name' in config, "配置中缺少 target_process_name 字段"
    print(f"✓ target_process_name 字段存在: {config['target_process_name']}")
    
    # 验证默认值
    if config['target_process_name'] == '泵空跑合':
        print(f"✓ 默认值正确: {config['target_process_name']}")
    else:
        print(f"✓ 使用配置文件中的值: {config['target_process_name']}")
    
    print()

def test_debug_mode():
    """测试调试模式"""
    print("=" * 60)
    print("测试2: 调试模式")
    print("=" * 60)
    
    from work_order_query import query_work_order, WORK_ORDER_DB_CONFIG
    
    # 临时启用debug模式
    original_debug = WORK_ORDER_DB_CONFIG.get('debug_mode', False)
    WORK_ORDER_DB_CONFIG['debug_mode'] = True
    
    try:
        result = query_work_order("TEST-001")
        
        assert result is not None, "调试模式应该返回数据"
        print(f"✓ 调试模式返回数据")
        
        # 验证所有必需字段
        required_fields = ['work_order_no', 'process_no', 'process_name', 'part_no', 'executor']
        for field in required_fields:
            assert field in result, f"缺少字段: {field}"
            print(f"✓ 字段存在: {field} = {result[field]}")
        
    finally:
        # 恢复原始debug模式设置
        WORK_ORDER_DB_CONFIG['debug_mode'] = original_debug
    
    print()

def test_config_file_structure():
    """测试配置文件结构"""
    print("=" * 60)
    print("测试3: 配置文件结构")
    print("=" * 60)
    
    config_file = Path(__file__).parent / 'work_order_db_config.json'
    
    if config_file.exists():
        with open(config_file, 'r', encoding='utf-8') as f:
            config = json.load(f)
        
        # 验证target_process_name字段
        if 'target_process_name' in config:
            print(f"✓ 配置文件包含 target_process_name: {config['target_process_name']}")
        else:
            print("⚠ 配置文件缺少 target_process_name 字段（将使用默认值）")
        
        # 显示配置文件内容
        print("\n配置文件内容:")
        print(json.dumps(config, ensure_ascii=False, indent=2))
    else:
        print("⚠ 配置文件不存在，将在首次运行时创建")
    
    print()

def test_database_schema():
    """测试数据库schema"""
    print("=" * 60)
    print("测试4: 数据库Schema")
    print("=" * 60)
    
    import sqlite3
    from pathlib import Path
    
    db_path = Path(__file__).parent / 'experiments.db'
    
    if db_path.exists():
        conn = sqlite3.connect(str(db_path))
        cursor = conn.cursor()
        
        # 获取表结构
        cursor.execute("PRAGMA table_info(experiments)")
        columns = cursor.fetchall()
        
        column_names = [col[1] for col in columns]
        
        # 验证新字段
        if 'process_name' in column_names:
            print("✓ process_name 字段存在")
        else:
            print("⚠ process_name 字段不存在（将在程序启动时自动添加）")
        
        if 'part_no' in column_names:
            print("✓ part_no 字段存在")
        else:
            print("⚠ part_no 字段不存在（将在程序启动时自动添加）")
        
        print("\n数据库表结构:")
        for col in columns:
            print(f"  {col[1]}: {col[2]}")
        
        conn.close()
    else:
        print("⚠ 数据库文件不存在，将在程序首次运行时创建")
    
    print()

def main():
    """运行所有测试"""
    print("\n" + "=" * 60)
    print("工单查询增强功能测试")
    print("=" * 60 + "\n")
    
    try:
        test_config_loading()
        test_debug_mode()
        test_config_file_structure()
        test_database_schema()
        
        print("=" * 60)
        print("✓ 所有测试通过！")
        print("=" * 60)
        print("\n下一步:")
        print("1. 启动程序: python main.py")
        print("2. 点击「开始工单」按钮")
        print("3. 输入工单号进行测试")
        print("4. 查看实验历史列表中的新列（工序名称、零件号）")
        print()
        
    except Exception as e:
        print(f"\n✗ 测试失败: {e}")
        import traceback
        traceback.print_exc()
        return 1
    
    return 0

if __name__ == "__main__":
    exit(main())