tests/test_integration_db.py

import os
import importlib

def has_db():
    return bool(os.environ.get("DATABASE_URL"))

def test_init_db_and_ingest_retrieve():
    if not has_db():
        return
    app = importlib.import_module("app")
    ok = app.init_db()
    assert ok
    status, _, _ = app.ingest_files([b"conteudo de teste para inserir no banco"])
    assert "Ingeridos" in status
    res = app.retrieve("conteudo", k=1)
    assert isinstance(res, list)
    assert len(res) >= 0