mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 05:50:44 +08:00
oliver
d14e9d3596
- Add SQLAlchemy Core repos, pg adapter/compat, assistant_store factory, Alembic bootstrap and migration/cutover scripts. - Harden chat_message writes (NUL scrub for PG), turn_uuid on attempt failure, WS turn_runner fallbacks and gateway executed_turn_uuid init. - start_gateway: log paths, PS7 stderr handling via cmd, background stdout/stderr redirect; runtime assistant_runtime_log_dir export. - Ops: clear_all_chat_sessions with PG-only --postgresql and env-gated wipe; clear_postgres_chat_sessions.ps1. - Tests: SA repos, pg compat, persist fallback, smoke env isolation; CI and docs touch-ups. Co-authored-by: Cursor <cursoragent@cursor.com>
46 lines
1.9 KiB
Python
46 lines
1.9 KiB
Python
"""Guardrails for SQLite→PostgreSQL SQL rewriting used by SqliteStore."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from svc.persistence import pg_compat
|
|
from svc.persistence.pg_adapter import normalize_psycopg_conninfo
|
|
|
|
|
|
def test_normalize_psycopg_conninfo_strips_sqlalchemy_driver() -> None:
|
|
u = "postgresql+psycopg://user:pass@127.0.0.1:5432/oclaw"
|
|
assert normalize_psycopg_conninfo(u) == "postgresql://user:pass@127.0.0.1:5432/oclaw"
|
|
|
|
|
|
def test_insert_or_replace_ui_session_owner_rewritten() -> None:
|
|
sql = """INSERT OR REPLACE INTO ui_session_owner(session_id, tenant_id, user_id, created_at)
|
|
VALUES (?, ?, ?, ?)"""
|
|
adapted = pg_compat.adapt_sql_for_postgres(sql)
|
|
assert "INSERT OR REPLACE" not in adapted.upper()
|
|
assert "ON CONFLICT" in adapted.upper()
|
|
|
|
|
|
def test_llm_profile_insert_or_ignore_rewritten() -> None:
|
|
sql = """INSERT OR IGNORE INTO llm_profile
|
|
(id, name, mode, model, base_url, api_key, updated_at, is_builtin, hide_in_ui, owner_user_id)
|
|
VALUES (?, ?, 'ollama', ?, ?, NULL, ?, 1, 0, NULL)"""
|
|
adapted = pg_compat.adapt_sql_for_postgres(sql)
|
|
assert "INSERT OR IGNORE" not in adapted.upper()
|
|
assert "ON CONFLICT (id) DO NOTHING" in adapted
|
|
|
|
|
|
def test_role_permission_insert_or_ignore_rewritten() -> None:
|
|
sql = """INSERT OR IGNORE INTO role_permission(role, permission, created_at)
|
|
VALUES (?, ?, ?)"""
|
|
adapted = pg_compat.adapt_sql_for_postgres(sql)
|
|
assert "INSERT OR IGNORE" not in adapted.upper()
|
|
assert "ON CONFLICT (role, permission) DO NOTHING" in adapted
|
|
|
|
|
|
def test_scrub_nul_bytes_from_text() -> None:
|
|
assert pg_compat.scrub_nul_bytes_from_text(None) is None
|
|
assert pg_compat.scrub_nul_bytes_from_text("ok") == "ok"
|
|
assert pg_compat.scrub_nul_bytes_from_text("a\x00b") == "ab"
|
|
|
|
|
|
def test_scrub_nul_bytes_from_jsonable_nested() -> None:
|
|
assert pg_compat.scrub_nul_bytes_from_jsonable({"x": "y\x00z"}) == {"x": "yz"}
|