mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-10-27 08:12:22 +01:00
62 lines
2.4 KiB
Python
62 lines
2.4 KiB
Python
import re
|
|
from pathlib import Path
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
JS = (ROOT / "static" / "js" / "galleryEditor.js").read_text()
|
|
ROUTES = (ROOT / "routes" / "gallery" / "gallery_routes.py").read_text()
|
|
SW = (ROOT / "static" / "sw.js").read_text()
|
|
|
|
|
|
def test_client_gen_mode_generates():
|
|
assert "if (mode === 'gen')" in JS
|
|
assert "_runQuickControlOp('ai.generateImage'" in JS
|
|
assert "Generating an image…" in JS
|
|
assert "Generated an image." in JS
|
|
assert "Could not generate an image:" in JS
|
|
|
|
|
|
def test_client_gen_mode_runs_after_harmonize():
|
|
harm = JS.index("_isHarmonizeIntent(prompt))")
|
|
img = JS.index("if (mode === 'gen')")
|
|
assert harm < img
|
|
|
|
|
|
def test_editing_status_is_distinct_from_generating_status():
|
|
# Image mode makes a brand-new picture; every other mode edits the canvas.
|
|
assert "_setAiCommandStatus('Editing the image…', 'running');" in JS
|
|
assert "_setAiCommandStatus('Generating an image…', 'running');" in JS
|
|
assert "_setAiCommandStatus('Generating…', 'running');" not in JS
|
|
|
|
|
|
def test_server_generate_route_exists():
|
|
assert '@router.post("/api/image/generate")' in ROUTES
|
|
assert "do_generate_image" in ROUTES
|
|
assert "can_generate_images" in ROUTES.split('@router.post("/api/image/generate")', 1)[1][:1200]
|
|
|
|
|
|
def test_sw_cache_bumped():
|
|
m = re.search(r"CACHE_NAME = '(odysseus-v\d+-pr40-editor-precache)'", SW)
|
|
assert m and re.fullmatch(r"odysseus-v\d+-pr40-editor-precache", m.group(1))
|
|
|
|
|
|
def test_generate_route_forwards_the_picked_model():
|
|
"""The model picker's choice must reach do_generate_image.
|
|
|
|
Settings defaults may otherwise point at the local Qwen server and the
|
|
user's chosen OpenRouter model is never used.
|
|
"""
|
|
route = ROUTES.split('@router.post("/api/image/generate")', 1)[1][:2600]
|
|
assert 'model_choice = str((body or {}).get("model") or "").strip()' in route
|
|
assert 'payload["model"] = model_choice' in route
|
|
assert "content = _json.dumps(payload)" in route
|
|
assert "localStorage.getItem('ge-ai-command-model')" in JS
|
|
|
|
|
|
def test_cloud_image_model_is_resolved_before_the_local_fallback():
|
|
"""A gateway model must not be hijacked onto the local image endpoint."""
|
|
ai = (ROOT / "src" / "ai_interaction.py").read_text()
|
|
block = ai.split("async def _resolve_image_model(model_name: str):", 1)[1][:900]
|
|
plain = block.index("_resolve_model(model_name, owner=owner)")
|
|
typed = block.index('model_type="image"')
|
|
assert plain < typed
|