def _walk_versions(self, conf):
# Determine latest version script from the repo, then
# upgrade from 1 through to the latest, with no data
# in the databases. This just checks that the schema itself
# upgrades successfully.
# Assert we are not under version control...
self.assertRaises(exception.DatabaseMigrationError,
migration_api.db_version,
conf)
# Place the database under version control
migration_api.version_control(conf)
cur_version = migration_api.db_version(conf)
self.assertEqual(0, cur_version)
for version in xrange(1, TestMigrations.REPOSITORY.latest + 1):
migration_api.upgrade(conf, version)
cur_version = migration_api.db_version(conf)
self.assertEqual(cur_version, version)
# Now walk it back down to 0 from the latest, testing
# the downgrade paths.
for version in reversed(
xrange(0, TestMigrations.REPOSITORY.latest)):
migration_api.downgrade(conf, version)
cur_version = migration_api.db_version(conf)
self.assertEqual(cur_version, version)
def _walk_versions(self, conf):
# Determine latest version script from the repo, then
# upgrade from 1 through to the latest, with no data
# in the databases. This just checks that the schema itself
# upgrades successfully.
# Assert we are not under version control...
self.assertRaises(exception.DatabaseMigrationError,
migration_api.db_version, conf)
# Place the database under version control
migration_api.version_control(conf)
cur_version = migration_api.db_version(conf)
self.assertEqual(0, cur_version)
for version in xrange(1, TestMigrations.REPOSITORY.latest + 1):
migration_api.upgrade(conf, version)
cur_version = migration_api.db_version(conf)
self.assertEqual(cur_version, version)
# Now walk it back down to 0 from the latest, testing
# the downgrade paths.
for version in reversed(xrange(0, TestMigrations.REPOSITORY.latest)):
migration_api.downgrade(conf, version)
cur_version = migration_api.db_version(conf)
self.assertEqual(cur_version, version)
def _no_data_loss_2_to_3_to_2(self, engine, conf):
migration_api.version_control(conf)
migration_api.upgrade(conf, 2)
cur_version = migration_api.db_version(conf)
self.assertEquals(2, cur_version)
# We are now on version 2. Check that the images table does
# not contain the type column...
images_table = Table('images', MetaData(), autoload=True,
autoload_with=engine)
image_properties_table = Table('image_properties', MetaData(),
autoload=True,
autoload_with=engine)
self.assertTrue('type' in images_table.c,
"'type' column found in images table columns! "
"images table columns: %s"
% images_table.c.keys())
conn = engine.connect()
sel = select([func.count("*")], from_obj=[images_table])
orig_num_images = conn.execute(sel).scalar()
sel = select([func.count("*")], from_obj=[image_properties_table])
orig_num_image_properties = conn.execute(sel).scalar()
now = datetime.datetime.now()
inserter = images_table.insert()
conn.execute(inserter, [
{'deleted': False, 'created_at': now,
'updated_at': now, 'type': 'kernel',
'status': 'active', 'is_public': True},
{'deleted': False, 'created_at': now,
'updated_at': now, 'type': 'ramdisk',
'status': 'active', 'is_public': True}])
sel = select([func.count("*")], from_obj=[images_table])
num_images = conn.execute(sel).scalar()
self.assertEqual(orig_num_images + 2, num_images)
conn.close()
# Now let's upgrade to 3. This should move the type column
# to the image_properties table as type properties.
migration_api.upgrade(conf, 3)
cur_version = migration_api.db_version(conf)
self.assertEquals(3, cur_version)
images_table = Table('images', MetaData(), autoload=True,
autoload_with=engine)
self.assertTrue('type' not in images_table.c,
"'type' column not found in images table columns! "
"images table columns reported by metadata: %s\n"
% images_table.c.keys())
image_properties_table = Table('image_properties', MetaData(),
autoload=True,
autoload_with=engine)
conn = engine.connect()
sel = select([func.count("*")], from_obj=[image_properties_table])
num_image_properties = conn.execute(sel).scalar()
self.assertEqual(orig_num_image_properties + 2, num_image_properties)
conn.close()
# Downgrade to 2 and check that the type properties were moved
# to the main image table
migration_api.downgrade(conf, 2)
images_table = Table('images', MetaData(), autoload=True,
autoload_with=engine)
self.assertTrue('type' in images_table.c,
"'type' column found in images table columns! "
"images table columns: %s"
% images_table.c.keys())
image_properties_table = Table('image_properties', MetaData(),
autoload=True,
autoload_with=engine)
conn = engine.connect()
sel = select([func.count("*")], from_obj=[image_properties_table])
last_num_image_properties = conn.execute(sel).scalar()
self.assertEqual(num_image_properties - 2, last_num_image_properties)
def _no_data_loss_2_to_3_to_2(self, engine, conf):
migration_api.version_control(conf)
migration_api.upgrade(conf, 2)
cur_version = migration_api.db_version(conf)
self.assertEquals(2, cur_version)
# We are now on version 2. Check that the images table does
# not contain the type column...
images_table = Table('images',
MetaData(),
autoload=True,
autoload_with=engine)
image_properties_table = Table('image_properties',
MetaData(),
autoload=True,
autoload_with=engine)
self.assertTrue(
'type' in images_table.c,
"'type' column found in images table columns! "
"images table columns: %s" % images_table.c.keys())
conn = engine.connect()
sel = select([func.count("*")], from_obj=[images_table])
orig_num_images = conn.execute(sel).scalar()
sel = select([func.count("*")], from_obj=[image_properties_table])
orig_num_image_properties = conn.execute(sel).scalar()
now = datetime.datetime.now()
inserter = images_table.insert()
conn.execute(inserter, [{
'deleted': False,
'created_at': now,
'updated_at': now,
'type': 'kernel',
'status': 'active',
'is_public': True
}, {
'deleted': False,
'created_at': now,
'updated_at': now,
'type': 'ramdisk',
'status': 'active',
'is_public': True
}])
sel = select([func.count("*")], from_obj=[images_table])
num_images = conn.execute(sel).scalar()
self.assertEqual(orig_num_images + 2, num_images)
conn.close()
# Now let's upgrade to 3. This should move the type column
# to the image_properties table as type properties.
migration_api.upgrade(conf, 3)
cur_version = migration_api.db_version(conf)
self.assertEquals(3, cur_version)
images_table = Table('images',
MetaData(),
autoload=True,
autoload_with=engine)
self.assertTrue(
'type' not in images_table.c,
"'type' column not found in images table columns! "
"images table columns reported by metadata: %s\n" %
images_table.c.keys())
image_properties_table = Table('image_properties',
MetaData(),
autoload=True,
autoload_with=engine)
conn = engine.connect()
sel = select([func.count("*")], from_obj=[image_properties_table])
num_image_properties = conn.execute(sel).scalar()
self.assertEqual(orig_num_image_properties + 2, num_image_properties)
conn.close()
# Downgrade to 2 and check that the type properties were moved
# to the main image table
migration_api.downgrade(conf, 2)
images_table = Table('images',
MetaData(),
autoload=True,
autoload_with=engine)
self.assertTrue(
'type' in images_table.c,
"'type' column found in images table columns! "
"images table columns: %s" % images_table.c.keys())
image_properties_table = Table('image_properties',
MetaData(),
autoload=True,
autoload_with=engine)
conn = engine.connect()
sel = select([func.count("*")], from_obj=[image_properties_table])
last_num_image_properties = conn.execute(sel).scalar()
self.assertEqual(num_image_properties - 2, last_num_image_properties)