import requests

BASE_URL = "http://127.0.0.1:8000"


def test_create_mindmap():
    payload = {
        "session_id": "test-session-123456",
        "mindmap_json": {
            "id": "node_0",
            "label": "Python编程",
            "parent_id": None,
            "level": 0,
            "is_leaf": False,
            "children": [
                {
                    "id": "node_1",
                    "label": "基础语法",
                    "parent_id": "node_0",
                    "level": 1,
                    "is_leaf": False,
                    "children": [
                        {
                            "id": "node_5",
                            "label": "变量与数据类型",
                            "parent_id": "node_1",
                            "level": 2,
                            "is_leaf": True,
                            "children": [],
                        },
                        {
                            "id": "node_6",
                            "label": "控制流程",
                            "parent_id": "node_1",
                            "level": 2,
                            "is_leaf": True,
                            "children": [],
                        },
                        {
                            "id": "node_7",
                            "label": "函数定义",
                            "parent_id": "node_1",
                            "level": 2,
                            "is_leaf": True,
                            "children": [],
                        },
                    ],
                },
                {
                    "id": "node_2",
                    "label": "面向对象",
                    "parent_id": "node_0",
                    "level": 1,
                    "is_leaf": False,
                    "children": [
                        {
                            "id": "node_8",
                            "label": "类与对象",
                            "parent_id": "node_2",
                            "level": 2,
                            "is_leaf": True,
                            "children": [],
                        },
                        {
                            "id": "node_9",
                            "label": "继承与多态",
                            "parent_id": "node_2",
                            "level": 2,
                            "is_leaf": True,
                            "children": [],
                        },
                    ],
                },
                {
                    "id": "node_3",
                    "label": "常用库",
                    "parent_id": "node_0",
                    "level": 1,
                    "is_leaf": False,
                    "children": [
                        {
                            "id": "node_10",
                            "label": "NumPy",
                            "parent_id": "node_3",
                            "level": 2,
                            "is_leaf": True,
                            "children": [],
                        },
                        {
                            "id": "node_11",
                            "label": "Pandas",
                            "parent_id": "node_3",
                            "level": 2,
                            "is_leaf": True,
                            "children": [],
                        },
                        {
                            "id": "node_12",
                            "label": "Requests",
                            "parent_id": "node_3",
                            "level": 2,
                            "is_leaf": True,
                            "children": [],
                        },
                    ],
                },
                {
                    "id": "node_4",
                    "label": "应用场景",
                    "parent_id": "node_0",
                    "level": 1,
                    "is_leaf": False,
                    "children": [
                        {
                            "id": "node_13",
                            "label": "Web开发",
                            "parent_id": "node_4",
                            "level": 2,
                            "is_leaf": True,
                            "children": [],
                        },
                        {
                            "id": "node_14",
                            "label": "数据分析",
                            "parent_id": "node_4",
                            "level": 2,
                            "is_leaf": True,
                            "children": [],
                        },
                    ],
                },
            ],
        },
    }

    response = requests.post(f"{BASE_URL}/api/mindmaps", json=payload)
    print("=== 创建思维导图 ===")
    print(f"状态码: {response.status_code}")
    result = response.json()
    print(f"访问链接: {result['url']}")
    print(f"标题: {result['title']}")
    print(f"unique_id: {result['unique_id']}")
    return result


def test_get_mindmap(unique_id):
    response = requests.get(f"{BASE_URL}/api/mindmaps/{unique_id}")
    print("\n=== 获取思维导图 ===")
    print(f"状态码: {response.status_code}")
    print(f"标题: {response.json()['title']}")


if __name__ == "__main__":
    result = test_create_mindmap()
    test_get_mindmap(result["unique_id"])