def setUp(self):
yield super(AbstractTorrentUpgrade63to64, self).setUp()
self.torrent_collecting_dir = os.path.join(self.session_base_dir,
"torrent_collecting")
self.sqlite_path = os.path.join(self.session_base_dir, "sqlite")
os.mkdir(self.torrent_collecting_dir)
os.mkdir(os.path.join(self.torrent_collecting_dir, "test_dir"))
os.mkdir(self.sqlite_path)
# write and create files
self.write_data_to_file(
os.path.join(self.session_base_dir, "upgradingdb.txt"))
self.write_data_to_file(
os.path.join(self.torrent_collecting_dir, "test1.mbinmap"))
self.write_data_to_file(
os.path.join(self.torrent_collecting_dir, "test2.mhash"))
self.write_data_to_file(
os.path.join(self.torrent_collecting_dir, "tmp_test3"))
self.write_data_to_file(
os.path.join(self.torrent_collecting_dir, "torrent1.torrent"))
self.write_data_to_file(
os.path.join(self.torrent_collecting_dir, "torrent2.torrent"))
os.mkdir(os.path.join(self.torrent_collecting_dir, "swift_reseeds"))
shutil.copyfile(
TORRENT_FILE,
os.path.join(self.torrent_collecting_dir, "torrent3.torrent"))
shutil.copyfile(
os.path.join(self.DB_DATA_DIR, "torrent_upgrade_64_dispersy.db"),
os.path.join(self.sqlite_path, "dispersy.db"))
self.torrent_upgrader = TorrentMigrator64(self.torrent_collecting_dir,
self.session_base_dir)
def test_upgrade_temp_torrent_dir_is_file(self):
self.write_data_to_file(
os.path.join(self.session_base_dir, ".tmp_migration_v64"))
self.torrent_upgrader = TorrentMigrator64(self.torrent_collecting_dir,
self.session_base_dir)
self.torrent_upgrader.start_migrate()
def test_upgrade_no_valid_torrent_collecting_dir(self):
self.torrent_upgrader = TorrentMigrator64(
os.path.join(self.torrent_collecting_dir, "bla"),
self.session_base_dir)
self.torrent_upgrader.start_migrate()