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)