- services/kb-postgres: pgvector/pgvector:pg16 na SOLARIA (:5433), named volume, init/001_envelope.sql (CREATE EXTENSION vector + zamrożona tabela envelope: id/source/ts/geo/raw_ref/entities), service.yaml, healthcheck, README z poprawnym mechanizmem deploy (deploy-node.sh składa dwa -f) - hosts/solaria/runtime/kb-postgres/docker-compose.override.yml: mem_limit 4g - inventory/topology.yaml + hosts/solaria/services.yaml: kb-postgres wpisany - packages/kb-mail: nowa konwencja shared lib (pip install /repo/packages/<lib>/) envelope.py — @dataclass Envelope, walidacja tz-aware ts db.py — insert_envelope / get_envelope (asyncpg, ON CONFLICT DO NOTHING) archive.py — save_eml append-only (asyncio.to_thread, FileExistsError na dup) tests: 15 unit pass + 5 integration (@pytest.mark.integration, wymaga KB_TEST_DSN) Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
19 lines
533 B
Python
19 lines
533 B
Python
import os
|
|
import pytest
|
|
import asyncpg
|
|
|
|
|
|
@pytest.fixture
|
|
async def db_conn():
|
|
"""Async asyncpg connection wrapped in a transaction that rolls back after each test.
|
|
|
|
Requires KB_TEST_DSN env var or a kb-postgres running on localhost:5433.
|
|
Only used by tests marked @pytest.mark.integration.
|
|
"""
|
|
dsn = os.environ.get("KB_TEST_DSN", "postgresql://kb:kb@localhost:5433/kb")
|
|
conn = await asyncpg.connect(dsn)
|
|
await conn.execute("BEGIN")
|
|
yield conn
|
|
await conn.execute("ROLLBACK")
|
|
await conn.close()
|