"""
报告生成测试辅助函数
"""
import json


def create_mock_section1():
    """创建模拟的章节一数据"""
    return {
        "enterprise_background": {
            "description": "企业背景描述"
        },
        "informatization_status": {
            "overview": "概述",
            "private_cloud": {
                "title": "私有云",
                "description": "描述"
            },
            "public_cloud": {
                "title": "公有云",
                "description": "描述"
            }
        },
        "business_data_flow": {
            "overview": "概述",
            "manufacturing": {
                "title": "制造",
                "description": "描述"
            },
            "logistics": {
                "title": "物流",
                "description": "描述"
            },
            "retail": {
                "title": "零售",
                "description": "描述"
            },
            "data_aggregation": {
                "title": "数据聚合",
                "description": "描述"
            }
        }
    }


def create_mock_section2(data_source_structured=70, data_source_semi_structured=30):
    """创建模拟的章节二数据"""
    return {
        "summary": {
            "total_data_volume": "100TB",
            "total_data_objects": {
                "tables": "50 张表",
                "fields": "300 个字段"
            }
        },
        "storage_distribution": [
            {
                "category": "交易数据",
                "volume": "50TB",
                "storage_type": "MySQL",
                "color": "blue"
            }
        ],
        "data_source_structure": {
            "structured": data_source_structured,
            "semi_structured": data_source_semi_structured
        }
    }


def create_mock_section3():
    """创建模拟的章节三数据"""
    return {
        "overview": {
            "asset_count": 1,
            "high_value_assets": ["测试资产"],
            "description": "概述描述"
        },
        "assets": [
            {
                "id": "asset_001",
                "title": "测试资产",
                "subtitle": "测试副标题",
                "composition": {
                    "description": "构成描述",
                    "core_tables": ["test_table_01"]
                },
                "application_scenarios": {
                    "description": "应用场景描述"
                },
                "compliance_risks": {
                    "warnings": [
                        {
                            "type": "PII风险",
                            "content": "测试警告",
                            "highlights": []
                        }
                    ]
                }
            }
        ]
    }


def create_mock_section4():
    """创建模拟的章节四数据"""
    return {
        "compliance_remediation": {
            "title": "合规整改",
            "items": [
                {
                    "order": 1,
                    "category": "分类",
                    "description": "详细建议",
                    "code_references": []
                }
            ]
        },
        "technical_evolution": {
            "title": "技术演进",
            "description": "技术建议描述",
            "technologies": ["技术1", "技术2"]
        },
        "value_deepening": {
            "title": "价值深化",
            "items": [
                {
                    "description": "建议描述",
                    "scenarios": []
                }
            ]
        }
    }


def create_mock_llm_response_1_2(data_source_structured=70, data_source_semi_structured=30):
    """创建模拟的LLM响应（章节1和2）"""
    return json.dumps({
        "section1": create_mock_section1(),
        "section2": create_mock_section2(data_source_structured, data_source_semi_structured)
    }, ensure_ascii=False)


def create_mock_llm_response_3():
    """创建模拟的LLM响应（章节3）"""
    return json.dumps({
        "section3": create_mock_section3()
    }, ensure_ascii=False)


def create_mock_llm_response_4():
    """创建模拟的LLM响应（章节4）"""
    return json.dumps({
        "section4": create_mock_section4()
    }, ensure_ascii=False)