def setUp(self):
uri = ("sqlite:///:memory:")
self.uri_left = get_temporary_uri(uri)
self.uri_right = get_temporary_uri(uri)
self.alembic_config_left = make_alembic_config(self.uri_left,
alembic_root)
self.alembic_config_right = make_alembic_config(
self.uri_right, alembic_root)
new_db(self.uri_left)
new_db(self.uri_right)
def setUp(self):
uri = (
"mysql+mysqlconnector://root:password@localhost:3306/alembicverify"
)
self.uri_left = get_temporary_uri(uri)
self.uri_right = get_temporary_uri(uri)
self.alembic_config_left = make_alembic_config(self.uri_left,
alembic_root)
self.alembic_config_right = make_alembic_config(
self.uri_right, alembic_root)
new_db(self.uri_left)
new_db(self.uri_right)
def setUp(self):
# from aiida.backends.sqlalchemy.tests.migration_test import versions
from aiida.backends.sqlalchemy.migrations import versions
self.migr_method_dir_path = os.path.dirname(
os.path.realpath(utils.__file__))
# Set the alembic script directory location
self.alembic_dpath = os.path.join(self.migr_method_dir_path,
utils.ALEMBIC_REL_PATH)
# Constructing the versions directory
versions_dpath = os.path.join(os.path.dirname(versions.__file__))
# Setting dynamically the the path to the alembic configuration
# (this is where the env.py file can be found)
self.alembic_cfg_left = Config()
self.alembic_cfg_left.set_main_option('script_location',
self.alembic_dpath)
# Setting dynamically the versions directory. These are the
# migration scripts to pass from one version to the other. The
# default ones are overridden with test-specific migrations.
self.alembic_cfg_left.set_main_option('version_locations',
versions_dpath)
# The correction URL to the SQLA database of the current
# AiiDA connection
curr_db_url = sa.engine.url
# Create new urls for the two new databases
self.db_url_left = get_temporary_uri(str(curr_db_url))
self.db_url_right = get_temporary_uri(str(curr_db_url))
# Put the correct database url to the database used by alembic
self.alembic_cfg_left.set_main_option("sqlalchemy.url",
self.db_url_left)
# Database creation
new_database(self.db_url_left)
new_database(self.db_url_right)
def setUp(self):
"""Test setup."""
uri = (get_config('SQLALCHEMY_DATABASE_URI'))
alembic_ini = os.path.abspath('migrations/alembic.ini')
self.uri_left = get_temporary_uri(uri)
self.uri_right = get_temporary_uri(uri)
self.alembic_config_left = Config(alembic_ini)
self.alembic_config_left.set_main_option("script_location",
alembic_root)
self.alembic_config_left.set_main_option("sqlalchemy.url",
self.uri_left)
self.alembic_config_right = Config(alembic_ini)
self.alembic_config_right.set_main_option("script_location",
alembic_root)
self.alembic_config_right.set_main_option("sqlalchemy.url",
self.uri_right)
new_db(self.uri_left)
new_db(self.uri_right)
def create_temp_config(app):
"""Create alembic config with temp DB."""
uri = app.config['SQLALCHEMY_DATABASE_URI']
config_dir = os.path.abspath(os.path.dirname('migrations/'))
config = Config(os.path.join(config_dir, 'alembic.ini'))
uri_temp = get_temporary_uri(uri)
config.set_main_option("script_location", config_dir)
config.set_main_option("sqlalchemy.url", uri_temp)
return config
def create_app(self):
"""
Переопределние метода create_app из LiveServerTestCase.
Для того чтобы скормить тестам наше приложение.
"""
uri = (get_config('DB_CONFIG'))
self.testdb_uri = get_temporary_uri(uri)
new_db(self.testdb_uri)
engine = create_engine(self.testdb_uri)
engine.execute("CREATE EXTENSION pgcrypto;")
prepare_schema_from_models(self.testdb_uri, Base)
app = create_app(self.testdb_uri)
app.config['TESTING'] = True
app.config['DEBUG'] = True
app.config['LIVESERVER_PORT'] = 8943
app.config['LIVESERVER_TIMEOUT'] = 10
return app
def uri_right(db_uri):
return get_temporary_uri(db_uri)
def uri_left(db_uri):
return get_temporary_uri(db_uri)
