tests/test_document_store.py

from pathlib import Path

from pydantic import BaseModel

from rag_app.services.document_store import read_jsonl, write_jsonl


class ExampleRow(BaseModel):
    name: str


def test_jsonl_round_trip(tmp_path: Path) -> None:
    path = tmp_path / "rows.jsonl"
    rows = [ExampleRow(name="alpha"), ExampleRow(name="beta")]

    write_jsonl(path, rows)
    loaded = read_jsonl(path, ExampleRow)

    assert [row.name for row in loaded] == ["alpha", "beta"]