def test_migrations(self):
schema = {0: OldInitialSchema, # sickbeard.mainDB.InitialSchema,
31: sickbeard.mainDB.AddAnimeTVShow,
32: sickbeard.mainDB.AddAbsoluteNumbering,
33: sickbeard.mainDB.AddSceneAbsoluteNumbering,
34: sickbeard.mainDB.AddAnimeBlacklistWhitelist,
35: sickbeard.mainDB.AddSceneAbsoluteNumbering2,
36: sickbeard.mainDB.AddXemRefresh,
37: sickbeard.mainDB.AddSceneToTvShows,
38: sickbeard.mainDB.AddIndexerMapping,
39: sickbeard.mainDB.AddVersionToTvEpisodes,
41: AddDefaultEpStatusToTvShows
}
count = 1
while count < len(schema.keys()):
my_db = db.DBConnection()
for version in sorted(schema.keys())[:count]:
update = schema[version](my_db)
update.execute()
sleep(0.1)
db.MigrationCode(my_db)
my_db.close()
for filename in glob.glob(os.path.join(test.TESTDIR, test.TESTDBNAME) + '*'):
os.remove(filename)
sleep(0.1)
count += 1
def setUp_test_db():
"""upgrades the db to the latest version
"""
# upgrading the db
db.MigrationCode(db.DBConnection())
# fix up any db problems
db.sanityCheckDatabase(db.DBConnection(), mainDB.MainSanityCheck)
# and for cachedb too
db.upgradeDatabase(db.DBConnection('cache.db'), cache_db.InitialSchema)
# and for faileddb too
db.upgradeDatabase(db.DBConnection('failed.db'), failed_db.InitialSchema)