#!/usr/bin/env python3
"""
测试添加"如何投资"的优化记录
"""

import requests
import json

BASE_URL = "http://localhost:5002"

def test_add_investment_record():
    """测试添加投资相关的优化记录"""
    print("🧪 测试添加投资优化记录...")
    
    test_data = {
        "original_text": "如何投资",
        "optimized_text": """以下是针对不同投资需求的优化提示词，涵盖主要资产类别与策略方向：

【股票投资】
"生成A股沪深300成分股近三年的基本面分析报告，需包含以下维度：
1. 估值指标（PE/PB/PS分位数）
2. 财务健康度（资产负债率/现金流覆盖率）
3. 行业景气度（营收复合增长率/毛利率变动）
4. 技术面分析（相对强弱指数/布林带位置）
*重点标注当前估值处于历史30%分位以下的标的"

【基金配置】
"设计平衡型基金投资组合方案（股债比例6:4），要求：
1. 列举3只近5年夏普比率＞1.2的主动型权益基金
2. 配置2只久期在3-5年的国债ETF
3. 加入15%的黄金ETF作为避险资产
4. 测算该组合在2018-2023年期间的最大回撤率
*需说明美联储加息周期中的调整策略"

【固收产品】
"对比分析当前中美债券市场投资机会：
1. 中国10年期国债收益率与CPI的利差空间
2. 美国高收益企业债违约率预期（2024Q2）
3. 人民币汇率波动对跨境债券收益的影响系数
4. 提供久期免疫策略的具体操作方案"

【另类投资】
"编制大宗商品CTA策略回测报告（2020-2024）：
1. 能化/黑色/有色三大板块的趋势性指标
2. 跨品种套利年化收益率测算
3. 波动率聚集效应的风控方案
4. 建议仓位管理模型（凯利公式变体）"

【风险提示要素】
所有方案必须包含：
1. 历史波动率与VaR(95%)测算
2. 流动性风险预警指标
3. 跨市场传染效应分析
4. 符合《证券期货投资者适当性管理办法》的风险等级评定
请补充说明您的：
- 投资期限（短期/中期/长期）
- 风险承受等级（R1-R5）
- 特定行业偏好
- 资金规模区间
我将据此提供定制化方案。""",
        "optimization_type": "投资策略",
        "industry": "金融",
        "profession": "投资顾问",
        "tags": ["投资策略", "风险控制", "资产配置"]
    }
    
    try:
        response = requests.post(
            f"{BASE_URL}/api/optimization-history",
            json=test_data,
            headers={'Content-Type': 'application/json'}
        )
        
        if response.status_code == 200:
            result = response.json()
            if result.get('success'):
                new_id = result.get('data', {}).get('id')
                print(f"   ✅ 投资优化记录添加成功，ID: {new_id}")
                return True
            else:
                print(f"   ❌ 投资优化记录添加失败: {result.get('message')}")
                return False
        else:
            print(f"   ❌ 投资优化记录添加请求失败: {response.status_code}")
            print(f"   响应内容: {response.text}")
            return False
            
    except Exception as e:
        print(f"   ❌ 投资优化记录添加异常: {str(e)}")
        return False

def test_get_all_records():
    """测试获取所有记录"""
    print("\n🧪 测试获取所有优化记录...")
    
    try:
        response = requests.get(f"{BASE_URL}/api/optimization-history")
        
        if response.status_code == 200:
            result = response.json()
            if result.get('success'):
                history = result.get('data', {}).get('history', [])
                total = result.get('data', {}).get('pagination', {}).get('total', 0)
                print(f"   ✅ 成功获取 {total} 条记录")
                
                # 查找投资相关的记录
                investment_records = [r for r in history if '投资' in r.get('original_text', '') or '投资' in r.get('optimization_type', '')]
                if investment_records:
                    print(f"   ✅ 找到 {len(investment_records)} 条投资相关记录")
                    for record in investment_records:
                        print(f"      - ID: {record['id']}, 类型: {record['optimization_type']}, 时间: {record['created_at']}")
                else:
                    print("   ⚠️ 未找到投资相关记录")
                
                return True
            else:
                print(f"   ❌ 获取记录失败: {result.get('message')}")
                return False
        else:
            print(f"   ❌ 获取记录请求失败: {response.status_code}")
            return False
            
    except Exception as e:
        print(f"   ❌ 获取记录异常: {str(e)}")
        return False

def main():
    """主测试函数"""
    print("🚀 测试投资优化记录功能")
    print("=" * 50)
    
    # 测试添加投资记录
    add_success = test_add_investment_record()
    
    # 测试获取所有记录
    get_success = test_get_all_records()
    
    # 输出结果
    print("\n" + "=" * 50)
    print("📊 测试结果汇总")
    print("=" * 50)
    
    if add_success and get_success:
        print("🎉 所有测试通过！投资优化记录功能正常！")
        print("\n💡 现在您可以：")
        print("   1. 在优化历史页面看到投资相关的记录")
        print("   2. 搜索和筛选投资类型的记录")
        print("   3. 查看详细的投资策略内容")
    else:
        print("⚠️ 部分测试失败，请检查相关功能")
    
    return add_success and get_success

if __name__ == '__main__':
    success = main()
    exit(0 if success else 1)