nihalashetty
ae67bff5a3
Self-hosted platform for building, testing, and shipping LangChain/LangGraph agents. Deep-agent sub-agents on the canvas, a live tracing/observability timeline, auto-provisioned built-in tools with import/export, per-environment tool variables, streamed evaluations, and per-user auth token forwarding.
51 lines
2.0 KiB
Python
51 lines
2.0 KiB
Python
"""Shared test fixtures.
|
|
|
|
CRITICAL: tests must NOT touch the dev database. We point every persistence path at
|
|
a throwaway temp dir *before* any `forge` module imports (the engine, settings, secret
|
|
store, and Chroma path are all bound at import time from these env vars).
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
_TMP = Path(tempfile.mkdtemp(prefix="forge_tests_"))
|
|
os.environ.setdefault("FORGE_DATABASE_URL", f"sqlite+aiosqlite:///{(_TMP / 'test.db').as_posix()}")
|
|
os.environ.setdefault("FORGE_CHROMA_PATH", (_TMP / "chroma").as_posix())
|
|
os.environ.setdefault("FORGE_CHECKPOINT_DB", "memory")
|
|
os.environ.setdefault("FORGE_SECRET_KEY_FILE", (_TMP / "master.key").as_posix())
|
|
os.environ.setdefault("FORGE_SEED_DEMO", "false")
|
|
# Tests hit MockTransport / fake hosts; don't do real DNS in the SSRF guard. The
|
|
# guard's blocking logic is covered explicitly in test_ssrf.py (explicit policies).
|
|
os.environ.setdefault("FORGE_EGRESS_BLOCK_PRIVATE", "false")
|
|
# Auth defaults to ON in the real app; the test suite mostly calls services directly
|
|
# without tokens, so run permissive. test_auth forces auth_required=True where it matters.
|
|
os.environ.setdefault("FORGE_AUTH_REQUIRED", "false")
|
|
|
|
import pytest # noqa: E402
|
|
|
|
from forge.db.base import init_db # noqa: E402
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
async def _ensure_tables():
|
|
# create_all is idempotent; cheap to run per-test for an isolated DB state.
|
|
await init_db()
|
|
yield
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _reset_sse_appstatus():
|
|
# sse_starlette caches a module-level `should_exit_event` bound to the FIRST event loop it
|
|
# runs in. pytest-asyncio gives each test a fresh loop, so a *second* streaming test in the
|
|
# process would await that stale event -> "bound to a different event loop". Reset it per
|
|
# test so each SSE response recreates the event in its own loop.
|
|
try:
|
|
from sse_starlette.sse import AppStatus
|
|
except ImportError:
|
|
return
|
|
AppStatus.should_exit = False
|
|
AppStatus.should_exit_event = None
|