mirror of
https://github.com/langchain-ai/langgraph.git
synced 2026-08-21 07:02:25 +02:00
b37f78942d
* checkpoint-postgres: new library for postgres checkpointer implementation --------- Co-authored-by: Nuno Campos <nuno@langchain.dev>
26 lines
775 B
Python
26 lines
775 B
Python
import pytest
|
|
from psycopg import AsyncConnection
|
|
from psycopg.errors import UndefinedTable
|
|
from psycopg.rows import dict_row
|
|
|
|
DEFAULT_URI = "postgres://postgres:postgres@localhost:5432/postgres?sslmode=disable"
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
async def conn():
|
|
async with await AsyncConnection.connect(
|
|
DEFAULT_URI, autocommit=True, prepare_threshold=0, row_factory=dict_row
|
|
) as conn:
|
|
yield conn
|
|
|
|
|
|
@pytest.fixture(scope="function", autouse=True)
|
|
async def clear_test_db(conn):
|
|
"""Delete all tables before each test."""
|
|
try:
|
|
await conn.execute("DELETE FROM checkpoints")
|
|
await conn.execute("DELETE FROM checkpoint_blobs")
|
|
await conn.execute("DELETE FROM checkpoint_writes")
|
|
except UndefinedTable:
|
|
pass
|