#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
测试 report_generator.py 的保存逻辑
"""

import sys
import os
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from lib.report_generator import save_report_history
from lib.db import execute_query

def test_save_logic():
    """测试保存逻辑"""
    print("="*60)
    print("测试 report_generator 的保存逻辑")
    print("="*60)
    
    try:
        # 1. 保存测试记录
        print("\n1. 调用 save_report_history...")
        test_file_path = '/reports/test_from_generator.xlsx'
        
        print(f"   准备保存:")
        print(f"   - file_path: '{test_file_path}'")
        print(f"   - status: 1")
        
        history_id = save_report_history(
            report_date='2025-01-01',
            start_time='2025-01-01 08:00:00',
            end_time='2025-01-02 08:00:00',
            split_type='test',
            split_value='test_value',
            split_name='测试名称',
            config_id=999999,
            total_orders=10,
            total_amount=100.50,
            sum_results={'test_field': 50.0},
            file_path=test_file_path,
            status=1
        )
        
        print(f"   ✓ 保存成功，ID: {history_id}")
        
        # 2. 查询记录
        print("\n2. 查询刚保存的记录...")
        query_sql = "SELECT * FROM t_daily_report_history WHERE id = %s"
        result = execute_query(query_sql, (history_id,))
        
        if result and len(result) > 0:
            record = result[0]
            print("   ✓ 查询成功")
            print(f"   - ID: {record.get('id')}")
            print(f"   - file_path: '{record.get('file_path')}'")
            print(f"   - status: {record.get('status')}")
            
            if record.get('file_path') == test_file_path:
                print("\n✓✓✓ save_report_history 工作正常！")
            else:
                print(f"\n❌❌❌ file_path 不匹配！")
                print(f"   期望: '{test_file_path}'")
                print(f"   实际: '{record.get('file_path')}'")
        else:
            print("   ❌ 查询失败，没有找到数据")
        
        # 3. 清理测试数据
        print("\n3. 清理测试数据...")
        from lib.db import execute_update
        execute_update("DELETE FROM t_daily_report_history WHERE id = %s", (history_id,))
        print("   ✓ 已清理")
        
        print("\n" + "="*60)
        print("测试完成")
        print("="*60)
        
    except Exception as e:
        print(f"\n❌ 测试失败: {e}")
        import traceback
        traceback.print_exc()
    
    input("\n按回车键退出...")

if __name__ == "__main__":
    test_save_logic()