mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-04-25 08:46:22 +02:00
a4dc84d1ab
- Introduced should_summarize parameter in TaskDispatcher and CeleryTaskDispatcher to control summary generation. - Updated InlineTaskDispatcher to support the new parameter for document processing.
61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
"""Root conftest — shared fixtures available to all test modules."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
|
|
_DEFAULT_TEST_DB = (
|
|
"postgresql+asyncpg://postgres:postgres@localhost:5432/surfsense_test"
|
|
)
|
|
TEST_DATABASE_URL = os.environ.get("TEST_DATABASE_URL", _DEFAULT_TEST_DB)
|
|
|
|
# Force the app to use the test database regardless of any pre-existing
|
|
# DATABASE_URL in the environment (e.g. from .env or shell profile).
|
|
os.environ["DATABASE_URL"] = TEST_DATABASE_URL
|
|
|
|
import pytest # noqa: E402
|
|
|
|
from app.db import DocumentType # noqa: E402
|
|
from app.indexing_pipeline.connector_document import ConnectorDocument # noqa: E402
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Unit test fixtures
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_user_id() -> str:
|
|
return "00000000-0000-0000-0000-000000000001"
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_search_space_id() -> int:
|
|
return 1
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_connector_id() -> int:
|
|
return 42
|
|
|
|
|
|
@pytest.fixture
|
|
def make_connector_document():
|
|
"""
|
|
Generic factory for unit tests. Overridden in tests/integration/conftest.py
|
|
with real DB-backed IDs for integration tests.
|
|
"""
|
|
|
|
def _make(**overrides):
|
|
defaults = {
|
|
"title": "Test Document",
|
|
"source_markdown": "## Heading\n\nSome content.",
|
|
"unique_id": "test-id-001",
|
|
"document_type": DocumentType.CLICKUP_CONNECTOR,
|
|
"search_space_id": 1,
|
|
"connector_id": 1,
|
|
"created_by_id": "00000000-0000-0000-0000-000000000001",
|
|
}
|
|
defaults.update(overrides)
|
|
return ConnectorDocument(**defaults)
|
|
|
|
return _make
|