forked from zhangyonghao/minimap
156 lines
5.4 KiB
Python
156 lines
5.4 KiB
Python
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"])
|