def setUp(self): super(MySQLModeTestCase, self).setUp() self.engine = session.create_engine(self.engine.url, mysql_sql_mode=self.mysql_mode) self.connection = self.engine.connect() meta = MetaData() meta.bind = self.engine self.test_table = Table(_TABLE_NAME + "mode", meta, Column('id', Integer, primary_key=True), Column('bar', String(255))) self.test_table.create() self.addCleanup(self.test_table.drop) self.addCleanup(self.connection.close)
def setUp(self): super(DbFixture, self).setUp() self.test.engine = session.create_engine(self._get_uri()) self.test.sessionmaker = session.get_maker(self.test.engine)
def version(self): engine = db_session.create_engine(self.db_url) with engine.connect() as conn: context = alembic_migration.MigrationContext.configure(conn) return context.get_current_revision()
def __init__(self, migration_config): self.repository = migration_config.get('migration_repo_path', '') self.init_version = migration_config.get('init_version', 0) self.db_url = migration_config['db_url'] self.engine = db_session.create_engine(self.db_url)