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 base_from_models(self):
assert not self.db_exist
new_db(self.uri)
self.engine = create_engine(self.uri)
self.engine.execute("CREATE EXTENSION pgcrypto;")
db = self.app.extensions.get('sqlalchemy').db
db.metadata.create_all(self.engine)
self.metadata = MetaData(bind=self.engine)
self.metadata.reflect()
self.metadata.bind = None
self.db_exist = True
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 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 new_db_right(uri_right):
new_db(uri_right)
yield
destroy_database(uri_right)
def new_db_left(uri_left):
new_db(uri_left)
yield
destroy_database(uri_left)
