Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -7,30 +7,32 @@
from uuid import uuid4

import pytest
from duckdb import ColumnExpression, ConstantExpression, DuckDBPyConnection
from duckdb import ColumnExpression, ConstantExpression, DuckDBPyConnection, connect

from dve.core_engine.backends.implementations.duckdb.auditing import DDBAuditingManager
from dve.core_engine.models import ProcessingStatusRecord, SubmissionInfo, SubmissionStatisticsRecord
from dve.pipeline.utils import SubmissionStatus

from .....fixtures import temp_ddb_conn # pylint: disable=unused-import


@pytest.fixture(scope="function")
def ddb_audit_manager(temp_ddb_conn) -> Iterator[DDBAuditingManager]:
db_file: Path
conn: DuckDBPyConnection
db_file, conn = temp_ddb_conn
yield DDBAuditingManager(database_uri=db_file.as_uri(), connection=conn)
def ddb_audit_manager() -> Iterator[DDBAuditingManager]:
db = f"dve_{uuid4().hex}"
with tempfile.TemporaryDirectory(prefix="ddb_audit_testing") as tmp:
db_file = Path(tmp, db + ".duckdb")
conn = connect(database=db_file, read_only=False)

yield DDBAuditingManager(database_uri=db_file.as_uri(), connection=conn)


@pytest.fixture(scope="function")
def ddb_audit_manager_threaded(temp_ddb_conn) -> Iterator[DDBAuditingManager]:
db_file: Path
conn: DuckDBPyConnection
db_file, conn = temp_ddb_conn
with ThreadPoolExecutor(1) as pool:
yield DDBAuditingManager(database_uri=db_file.as_uri(), pool=pool, connection=conn)
def ddb_audit_manager_threaded() -> Iterator[DDBAuditingManager]:
db = f"dve_{uuid4().hex}"
with tempfile.TemporaryDirectory(prefix="ddb_audit_testing") as tmp:
db_file = Path(tmp, db + ".duckdb")
conn = connect(database=db_file, read_only=False)

with ThreadPoolExecutor(1) as pool:
yield DDBAuditingManager(database_uri=db_file.as_uri(), pool=pool, connection=conn)


@pytest.fixture
Expand Down
Loading