mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-07 12:00:44 +08:00
59 lines
2.5 KiB
Python
59 lines
2.5 KiB
Python
import os
|
|
import uuid
|
|
import pytest
|
|
|
|
pytestmark = pytest.mark.skipif(
|
|
os.environ.get("API_PROXY_SCHEME", "python") != "go",
|
|
reason="Go mode only (set API_PROXY_SCHEME=go + bin/ragflow_server on 9384)",
|
|
)
|
|
|
|
V2_DSL = {
|
|
"graph": {
|
|
"nodes": [
|
|
{"id": "begin", "type": "beginNode", "position": {"x": 50, "y": 200}, "data": {"label": "Begin", "name": "begin"}},
|
|
{"id": "answer:0", "type": "messageNode", "position": {"x": 400, "y": 200}, "data": {"label": "Answer", "name": "answer"}},
|
|
],
|
|
"edges": [
|
|
{"id": "xy-edge__begin-answer:0", "source": "begin", "target": "answer:0", "sourceHandle": "end", "targetHandle": "start"},
|
|
],
|
|
},
|
|
"components": {
|
|
"begin": {"obj": {"component_name": "Begin", "params": {}}, "downstream": ["answer:0"], "upstream": []},
|
|
"answer:0": {"obj": {"component_name": "Answer", "params": {}}, "downstream": [], "upstream": ["begin"]},
|
|
},
|
|
"retrieval": [],
|
|
"history": [],
|
|
"path": [],
|
|
"variables": [],
|
|
"globals": {"sys.query": ""},
|
|
}
|
|
|
|
|
|
@pytest.mark.p2
|
|
def test_v2_dsl_round_trip_position_preserved(rest_client):
|
|
# Unique title per run so the test is idempotent (Go returns 102 on duplicate).
|
|
title = f"go_v2_e2e_{uuid.uuid4().hex[:8]}"
|
|
|
|
# 1. POST v2 DSL
|
|
r = rest_client.post("/agents", json={"title": title, "dsl": V2_DSL})
|
|
assert r.status_code == 200 and r.json()["code"] == 0, r.text
|
|
agent_id = r.json()["data"]["id"]
|
|
|
|
# 2. GET — verify v2 shape (graph + components, NO _layout)
|
|
dsl = rest_client.get(f"/agents/{agent_id}").json()["data"]["dsl"]
|
|
assert "graph" in dsl, f"v2 must have 'graph', got keys={sorted(dsl.keys())}"
|
|
assert "_layout" not in dsl, f"v2 must NOT emit '_layout', got keys={sorted(dsl.keys())}"
|
|
assert "components" in dsl
|
|
assert len(dsl["graph"]["nodes"]) == 2
|
|
assert len(dsl["graph"]["edges"]) == 1
|
|
|
|
# 3. PUT — move begin node to (777, 888)
|
|
new_dsl = {**V2_DSL, "graph": {**V2_DSL["graph"]}}
|
|
new_dsl["graph"]["nodes"] = [{**n, "position": {"x": 777, "y": 888}} if n["id"] == "begin" else n for n in V2_DSL["graph"]["nodes"]]
|
|
r = rest_client.put(f"/agents/{agent_id}", json={"title": title, "dsl": new_dsl})
|
|
assert r.status_code == 200 and r.json()["code"] == 0, r.text
|
|
|
|
# 4. Re-GET — position preserved
|
|
pos = next(n["position"] for n in rest_client.get(f"/agents/{agent_id}").json()["data"]["dsl"]["graph"]["nodes"] if n["id"] == "begin")
|
|
assert pos == {"x": 777, "y": 888}, f"position lost: {pos}"
|