mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-06 19:42:34 +08:00
11ca1995a3
- Remove unused os imports in conftest.py and test_ingest.py - Remove unused Tag import in test_asset_management.py - Remove unused ensure_tags_exist import in test_ingest.py - Fix unused info2 variable in test_asset_management.py Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
import tempfile
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.assets.database.models import Base
|
|
|
|
|
|
@pytest.fixture
|
|
def db_engine():
|
|
"""In-memory SQLite engine for fast unit tests."""
|
|
engine = create_engine("sqlite:///:memory:")
|
|
Base.metadata.create_all(engine)
|
|
return engine
|
|
|
|
|
|
@pytest.fixture
|
|
def session(db_engine):
|
|
"""Session fixture for tests that need direct DB access."""
|
|
with Session(db_engine) as sess:
|
|
yield sess
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_create_session(db_engine):
|
|
"""Patch create_session to use our in-memory database."""
|
|
from contextlib import contextmanager
|
|
from sqlalchemy.orm import Session as SASession
|
|
|
|
@contextmanager
|
|
def _create_session():
|
|
with SASession(db_engine) as sess:
|
|
yield sess
|
|
|
|
with patch("app.assets.services.ingest.create_session", _create_session), \
|
|
patch("app.assets.services.asset_management.create_session", _create_session), \
|
|
patch("app.assets.services.tagging.create_session", _create_session):
|
|
yield _create_session
|
|
|
|
|
|
@pytest.fixture
|
|
def temp_dir():
|
|
"""Temporary directory for file operations."""
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
yield Path(tmpdir)
|