digital-garbageandClaude Sonnet 5.5
dd825979f6
Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
163 lines
7.4 KiB
Python
163 lines
7.4 KiB
Python
"""/api/shortcuts and /api/prompt/expand, over real HTTP -- proves the routes
|
|
actually reach core/shortcuts.py and core/prompt_build.py, not just that
|
|
those modules are correct in isolation (test_shortcuts.py, test_prompt_build.py).
|
|
"""
|
|
|
|
import json
|
|
import socket
|
|
import threading
|
|
|
|
import pytest
|
|
|
|
pytest.importorskip("aiohttp")
|
|
|
|
from core import config, routes # noqa: E402
|
|
|
|
|
|
@pytest.fixture
|
|
def server(tmp_path, monkeypatch):
|
|
monkeypatch.setattr(config, "SHORTCUTS_FILE", tmp_path / "shortcuts.json")
|
|
monkeypatch.setattr(config, "MARKERS_FILE", tmp_path / "markers.json")
|
|
monkeypatch.setattr(config, "SHORTCUT_CATEGORIES_FILE", tmp_path / "cats.json")
|
|
monkeypatch.setattr(config, "REVOLVER_FILE", tmp_path / "rev.json")
|
|
from aiohttp import web as aioweb
|
|
|
|
app = aioweb.Application()
|
|
table = aioweb.RouteTableDef()
|
|
routes.register(table, prefix="/funpack")
|
|
app.add_routes(table)
|
|
|
|
holder = {"ready": threading.Event()}
|
|
|
|
def run():
|
|
import asyncio
|
|
loop = asyncio.new_event_loop()
|
|
asyncio.set_event_loop(loop)
|
|
runner = aioweb.AppRunner(app)
|
|
loop.run_until_complete(runner.setup())
|
|
site = aioweb.TCPSite(runner, "127.0.0.1", 0)
|
|
loop.run_until_complete(site.start())
|
|
holder["port"] = site._server.sockets[0].getsockname()[1]
|
|
holder["loop"] = loop
|
|
holder["ready"].set()
|
|
loop.run_forever()
|
|
|
|
thread = threading.Thread(target=run, daemon=True)
|
|
thread.start()
|
|
assert holder["ready"].wait(20), "the test server did not start"
|
|
yield holder["port"]
|
|
holder["loop"].call_soon_threadsafe(holder["loop"].stop)
|
|
|
|
|
|
def _request(port, method, path, body=None):
|
|
payload = b"" if body is None else json.dumps(body).encode()
|
|
head = (f"{method} {path} HTTP/1.1\r\nHost: localhost\r\n"
|
|
f"Content-Type: application/json\r\n"
|
|
f"Content-Length: {len(payload)}\r\nConnection: close\r\n\r\n").encode()
|
|
with socket.create_connection(("127.0.0.1", port), timeout=30) as sock:
|
|
sock.sendall(head + payload)
|
|
chunks = []
|
|
while True:
|
|
got = sock.recv(65536)
|
|
if not got:
|
|
break
|
|
chunks.append(got)
|
|
header, _, rest = b"".join(chunks).partition(b"\r\n\r\n")
|
|
status = int(header.split()[1])
|
|
return status, json.loads(rest.decode() or "{}")
|
|
|
|
|
|
def test_save_list_delete(server):
|
|
status, resp = _request(server, "POST", "/funpack/api/shortcuts",
|
|
{"name": "Fox", "triggers": ["fox"], "replacements": ["red fox"]})
|
|
assert status == 200, resp
|
|
assert [s["name"] for s in resp["shortcuts"]] == ["Fox"]
|
|
|
|
status, listed = _request(server, "GET", "/funpack/api/shortcuts")
|
|
assert status == 200 and [s["name"] for s in listed["shortcuts"]] == ["Fox"]
|
|
|
|
status, deleted = _request(server, "DELETE", "/funpack/api/shortcuts/Fox")
|
|
assert status == 200 and deleted["shortcuts"] == []
|
|
|
|
|
|
def test_saving_a_shortcut_with_no_triggers_is_a_400(server):
|
|
status, resp = _request(server, "POST", "/funpack/api/shortcuts",
|
|
{"name": "Nothing", "triggers": [], "replacements": ["x"]})
|
|
assert status == 400
|
|
assert resp["problems"]
|
|
|
|
|
|
def test_clear_empties_the_library(server):
|
|
_request(server, "POST", "/funpack/api/shortcuts",
|
|
{"name": "Fox", "triggers": ["fox"], "replacements": ["red fox"]})
|
|
status, resp = _request(server, "POST", "/funpack/api/shortcuts/clear")
|
|
assert status == 200 and resp["shortcuts"] == []
|
|
|
|
|
|
def test_prompt_expand_combines_anchor_scene_and_postfix(server):
|
|
status, resp = _request(server, "POST", "/funpack/api/prompt/expand",
|
|
{"text": "a fox runs", "anchor": "cinematic", "postfix": "4k"})
|
|
assert status == 200
|
|
assert resp["text"] == "cinematic a fox runs 4k"
|
|
|
|
|
|
def test_prompt_expand_uses_the_saved_shortcut_library(server):
|
|
_request(server, "POST", "/funpack/api/shortcuts",
|
|
{"name": "Fox", "triggers": ["fox"], "replacements": ["red fox"]})
|
|
status, resp = _request(server, "POST", "/funpack/api/prompt/expand", {"text": "a fox runs"})
|
|
assert status == 200 and resp["text"] == "a red fox runs"
|
|
|
|
|
|
def test_prompt_expand_with_variables_and_postfix_disabled(server):
|
|
status, resp = _request(server, "POST", "/funpack/api/prompt/expand", {
|
|
"text": "a $animal runs", "anchor": "", "postfix": "ignored",
|
|
"postfix_enabled": False, "variables": [{"name": "animal", "value": "fox"}],
|
|
})
|
|
assert status == 200 and resp["text"] == "a fox runs"
|
|
|
|
|
|
def test_prompt_expand_on_a_non_object_body_is_a_400(server):
|
|
status, resp = _request(server, "POST", "/funpack/api/prompt/expand", ["nope"])
|
|
assert status == 400 and resp["problems"]
|
|
|
|
|
|
def test_story_split_join_and_markers_over_http(server):
|
|
_, joined = _request(server, "POST", "/funpack/api/story/join", {"scenes": ["a", "", "b"]})
|
|
status, split = _request(server, "POST", "/funpack/api/story/split", {"text": joined["text"]})
|
|
assert status == 200 and split["scenes"] == ["a", "", "b"]
|
|
|
|
status, saved = _request(server, "POST", "/funpack/api/story/markers", {"markers": ["cut"]})
|
|
assert status == 200 and saved["markers"] == ["cut"]
|
|
_, split = _request(server, "POST", "/funpack/api/story/split", {"text": "x cut y qcut z"})
|
|
assert split["scenes"] == ["x", "y qcut z"]
|
|
|
|
|
|
def test_story_rejects_bad_bodies_and_an_empty_marker_list(server):
|
|
assert _request(server, "POST", "/funpack/api/story/markers", {"markers": []})[0] == 400
|
|
assert _request(server, "POST", "/funpack/api/story/split", {"text": 5})[0] == 400
|
|
assert _request(server, "POST", "/funpack/api/story/join", {"scenes": [1]})[0] == 400
|
|
|
|
|
|
def test_categories_export_import_over_http(server):
|
|
_, r = _request(server, "POST", "/funpack/api/shortcuts/category", {"category": "Camera", "sub_category": "Moves"})
|
|
assert r["categories"] == [{"name": "Camera", "sub_categories": ["Moves"]}]
|
|
assert _request(server, "POST", "/funpack/api/shortcuts/category", {"category": ""})[0] == 400
|
|
_request(server, "POST", "/funpack/api/shortcuts", {"name": "Fox", "triggers": ["fox"], "replacements": ["red fox"]})
|
|
status, exported = _request(server, "GET", "/funpack/api/shortcuts/export")
|
|
assert status == 200 and [s["name"] for s in exported["shortcuts"]] == ["Fox"]
|
|
_request(server, "POST", "/funpack/api/shortcuts/clear")
|
|
status, back = _request(server, "POST", "/funpack/api/shortcuts/import", {"data": exported, "mode": "replace"})
|
|
assert status == 200 and back["imported"] == 1 and back["categories"] == exported["categories"]
|
|
assert _request(server, "POST", "/funpack/api/shortcuts/import", {"data": {"shortcuts": 5}})[0] == 400
|
|
assert _request(server, "POST", "/funpack/api/shortcuts/import", {"mode": "merge"})[0] == 400
|
|
|
|
|
|
def test_revolver_over_http_only_a_real_run_commits(server):
|
|
_request(server, "POST", "/funpack/api/shortcuts", {"name": "L", "triggers": ["lt"], "replacements": ["A", "B"]})
|
|
_, st = _request(server, "POST", "/funpack/api/shortcuts/revolver", {"enabled": True})
|
|
assert st == {"enabled": True, "random": False}
|
|
ask = lambda commit: _request(server, "POST", "/funpack/api/prompt/expand", {"text": "lt", "commit": commit})[1]["text"]
|
|
assert [ask(False), ask(False), ask(True), ask(True), ask(True)] == ["A", "A", "A", "B", "A"]
|
|
assert _request(server, "GET", "/funpack/api/shortcuts/revolver")[1]["enabled"] is True
|
|
assert _request(server, "POST", "/funpack/api/shortcuts/revolver", [])[0] == 400
|