def setUp(self): super(TestSqliteUniqueConstraints, self).setUp() migration.patch_migrate() self.helper = sqlite.SQLiteHelper() sa.Table( 'test_table', sa.schema.MetaData(bind=session.get_engine()), sa.Column('a', sa.Integer), sa.Column('b', sa.String(10)), sa.Column('c', sa.Integer), sa.UniqueConstraint('a', 'b', name='unique_a_b'), sa.UniqueConstraint('b', 'c', name='unique_b_c') ).create() # NOTE(rpodolyaka): it's important to use the reflected table here # rather than original one because this is what # we actually do in db migrations code self.reflected_table = sa.Table( 'test_table', sa.schema.MetaData(bind=session.get_engine()), autoload=True )
def test_slave_engine_nomatch(self): default = session.CONF.database.connection session.CONF.database.slave_connection = default e = session.get_engine() slave_e = session.get_engine(slave_engine=True) self.assertNotEqual(slave_e, e)
def tearDown(self): super(SessionErrorWrapperTestCase, self).tearDown() meta = MetaData() meta.bind = session.get_engine() test_table = Table(_TABLE_NAME, meta, autoload=True) test_table.drop() session.cleanup()
def tearDown(self): super(RegexpFilterTestCase, self).tearDown() meta = MetaData() meta.bind = session.get_engine() test_table = Table(_REGEXP_TABLE_NAME, meta, autoload=True) test_table.drop() session.cleanup()
def setUp(self): super(RegexpFilterTestCase, self).setUp() meta = MetaData() meta.bind = session.get_engine() test_table = Table( _REGEXP_TABLE_NAME, meta, Column('id', Integer, primary_key=True, nullable=False), Column('bar', String(255))) test_table.create()
def setUp(self): super(RegexpFilterTestCase, self).setUp() meta = MetaData() meta.bind = session.get_engine() test_table = Table(_REGEXP_TABLE_NAME, meta, Column('id', Integer, primary_key=True, nullable=False), Column('bar', String(255))) test_table.create()
def setUp(self): super(SessionErrorWrapperTestCase, self).setUp() meta = MetaData() meta.bind = session.get_engine() test_table = Table( _TABLE_NAME, meta, Column('id', Integer, primary_key=True, nullable=False), Column('deleted', Integer, default=0), Column('deleted_at', DateTime), Column('updated_at', DateTime), Column('created_at', DateTime), Column('foo', Integer), UniqueConstraint('foo', name='uniq_foo')) test_table.create()
def ins_wrap(self): if not set(dialects).issubset(ALLOWED_DIALECTS): raise ValueError( "Please use allowed dialects: %s" % ALLOWED_DIALECTS) engine = session.get_engine() if engine.name not in dialects: msg = ('The test "%s" can be run ' 'only on %s. Current engine is %s.') args = (f.__name__, ' '.join(dialects), engine.name) self.skip(msg % args) else: return f(self)
def test_db_version_control(self): with contextlib.nested( mock.patch.object(migration, '_find_migrate_repo'), mock.patch.object(versioning_api, 'version_control'), ) as (mock_find_repo, mock_version_control): mock_find_repo.return_value = self.return_value version = migration.db_version_control( self.path, self.test_version) self.assertEqual(version, self.test_version) mock_version_control.assert_called_once_with( db_session.get_engine(), self.return_value, self.test_version)
def setUp(self): super(TestSqliteUniqueConstraints, self).setUp() migration.patch_migrate() self.helper = sqlite.SQLiteHelper() test_table = sa.Table( "test_table", sa.schema.MetaData(bind=session.get_engine()), sa.Column("a", sa.Integer), sa.Column("b", sa.String(10)), sa.Column("c", sa.Integer), sa.UniqueConstraint("a", "b", name="unique_a_b"), sa.UniqueConstraint("b", "c", name="unique_b_c"), ) test_table.create() self.addCleanup(test_table.drop) # NOTE(rpodolyaka): it's important to use the reflected table here # rather than original one because this is what # we actually do in db migrations code self.reflected_table = sa.Table("test_table", sa.schema.MetaData(bind=session.get_engine()), autoload=True)
def test_db_sync_downgrade(self): with contextlib.nested( mock.patch.object(migration, '_find_migrate_repo'), mock.patch.object(versioning_api, 'downgrade') ) as (mock_find_repo, mock_downgrade): mock_find_repo.return_value = self.return_value self.mock_api_db_version.return_value = self.test_version + 1 migration.db_sync(self.path, self.test_version) mock_downgrade.assert_called_once_with( db_session.get_engine(), self.return_value, self.test_version)
def setUp(self): super(SessionErrorWrapperTestCase, self).setUp() meta = MetaData() meta.bind = session.get_engine() test_table = Table(_TABLE_NAME, meta, Column('id', Integer, primary_key=True, nullable=False), Column('deleted', Integer, default=0), Column('deleted_at', DateTime), Column('updated_at', DateTime), Column('created_at', DateTime), Column('foo', Integer), UniqueConstraint('foo', name='uniq_foo')) test_table.create()
def setUp(self): super(MySQLTraditionalModeTestCase, self).setUp() self.engine = session.get_engine(mysql_traditional_mode=True) 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(session.cleanup) self.addCleanup(self.test_table.drop) self.addCleanup(self.connection.close)
def config_for_engine(argv): config.parse_args(argv) logging.setup('forest') engine = get_engine() return engine
def test_no_slave_engine_match(self): slave_e = session.get_engine() e = session.get_engine() self.assertEqual(slave_e, e)