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 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 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)
def test_no_slave_engine_match(self):
slave_e = session.get_engine()
e = session.get_engine()
self.assertEqual(slave_e, e)
