mirror of
https://github.com/galaxyproject/galaxy.git
synced 2026-11-04 23:43:20 +08:00
John Davis
b57ec18991
- Do not explicitly import fixtures (a little duplication is safer/cleaner) - Move dangerous fixture into module to prevent accidental usage from other modules - Ensure only one database is used for all tests in module
42 lines
919 B
Python
42 lines
919 B
Python
import tempfile
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import Session
|
|
|
|
if TYPE_CHECKING:
|
|
from sqlalchemy.engine import Engine
|
|
|
|
from galaxy.model.unittest_utils.model_testing_utils import (
|
|
_generate_unique_database_name,
|
|
_make_sqlite_db_url,
|
|
)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def sqlite_url_factory():
|
|
"""Return a function that generates a sqlite url"""
|
|
|
|
def url():
|
|
database = _generate_unique_database_name()
|
|
return _make_sqlite_db_url(tmp_dir, database)
|
|
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
yield url
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def db_url(sqlite_url_factory): # noqa: F811
|
|
return sqlite_url_factory()
|
|
|
|
|
|
@pytest.fixture()
|
|
def engine(db_url: str) -> "Engine":
|
|
return create_engine(db_url)
|
|
|
|
|
|
@pytest.fixture
|
|
def session(engine: "Engine") -> Session:
|
|
return Session(engine)
|