Add accumulator retake workflow restore
This commit is contained in:
+14
-1
@@ -8004,7 +8004,13 @@ def smoke_server_route_payload_policy() -> None:
|
||||
|
||||
key = "smoke_route_payload"
|
||||
loop_nodes._ACCUMULATOR_STORES[key] = [
|
||||
{"id": "first", "value": "alpha", "_sxcp_preview_key": "first-key"},
|
||||
{
|
||||
"id": "first",
|
||||
"value": "alpha",
|
||||
"_sxcp_preview_key": "first-key",
|
||||
"prompt": {"api": "prompt"},
|
||||
"extra_pnginfo": {"workflow": {"nodes": [{"id": 1, "type": "SmokeNode"}]}},
|
||||
},
|
||||
{"id": "second", "value": "beta", "_sxcp_preview_key": "second-key"},
|
||||
]
|
||||
try:
|
||||
@@ -8012,6 +8018,13 @@ def smoke_server_route_payload_policy() -> None:
|
||||
_expect(listed.get("count") == 2, "Accumulator list payload lost stored entries")
|
||||
_expect(listed["entries"][0].get("value") == "alpha", "Accumulator list payload lost value summary")
|
||||
|
||||
retake = server_routes.accumulator_retake_payload({"store_key": key, "preview_key": "first-key"})
|
||||
_expect(
|
||||
retake.get("workflow", {}).get("nodes", [{}])[0].get("type") == "SmokeNode",
|
||||
"Accumulator retake payload lost workflow metadata",
|
||||
)
|
||||
_expect(retake.get("prompt", {}).get("api") == "prompt", "Accumulator retake payload lost prompt metadata")
|
||||
|
||||
moved = server_routes.accumulator_move_payload({"store_key": key, "entry_id": "second", "target_index": "1"})
|
||||
_expect(moved.get("moved") is True, "Accumulator move payload did not report movement")
|
||||
_expect(moved.get("from_index") == 2 and moved.get("to_index") == 1, "Accumulator move payload changed indices")
|
||||
|
||||
Reference in New Issue
Block a user