mirror of
https://github.com/galaxyproject/galaxy.git
synced 2026-09-24 16:30:27 +08:00
72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
import os
|
|
import tempfile
|
|
from typing import Optional
|
|
|
|
import pytest
|
|
|
|
try:
|
|
import psycopg
|
|
except ImportError:
|
|
psycopg = None # type: ignore[assignment, unused-ignore]
|
|
|
|
try:
|
|
import psycopg2
|
|
except ImportError:
|
|
psycopg2 = None
|
|
|
|
from galaxy.app_unittest_utils import galaxy_mock
|
|
|
|
|
|
def create_base_test(connection, amqp_type: str, amqp_connection: Optional[str] = None):
|
|
app = galaxy_mock.MockApp(database_connection=connection)
|
|
app.config.database_connection = connection
|
|
app.config.amqp_internal_connection = amqp_connection or f"sqlalchemy+{app.config.database_connection}"
|
|
app.amqp_type = amqp_type
|
|
return app
|
|
|
|
|
|
@pytest.fixture()
|
|
def sqlite_connection(request):
|
|
with tempfile.NamedTemporaryFile() as temp:
|
|
yield f"sqlite:////{temp.name}"
|
|
|
|
|
|
@pytest.fixture()
|
|
def sqlite_rabbitmq_app(sqlite_connection):
|
|
def create_app():
|
|
amqp_connection = os.environ.get("GALAXY_TEST_AMQP_INTERNAL_CONNECTION")
|
|
return create_base_test(sqlite_connection, amqp_type="rabbitmq", amqp_connection=amqp_connection)
|
|
|
|
return create_app
|
|
|
|
|
|
@pytest.fixture()
|
|
def sqlite_app(sqlite_connection):
|
|
def create_app():
|
|
return create_base_test(sqlite_connection, amqp_type="sqlite")
|
|
|
|
return create_app
|
|
|
|
|
|
@pytest.fixture()
|
|
def postgres_app(postgresql_proc):
|
|
connection = f"postgresql://{postgresql_proc.user}@{postgresql_proc.host}:{postgresql_proc.port}/"
|
|
|
|
def create_app():
|
|
return create_base_test(connection, amqp_type="postgres")
|
|
|
|
return create_app
|
|
|
|
|
|
@pytest.fixture(params=["postgres_app", "sqlite_app", "sqlite_rabbitmq_app"])
|
|
def database_app(request):
|
|
if request.param == "postgres_app":
|
|
if not psycopg:
|
|
pytest.skip("psycopg must be installed for postgresql_proc fixture")
|
|
if not psycopg2:
|
|
pytest.skip("psycopg2 must be installed for database_app fixture")
|
|
if request.param == "sqlite_rabbitmq_app":
|
|
if not os.environ.get("GALAXY_TEST_AMQP_INTERNAL_CONNECTION"):
|
|
pytest.skip("rabbitmq tests will be skipped if GALAXY_TEST_AMQP_INTERNAL_CONNECTION env var is unset")
|
|
return request.getfixturevalue(request.param)
|