class TorrentObjectTestCase(AbstractTorrentTestCase):
@classmethod
def setUpClass(cls):
cls.run_tracker()
cls.run_seeder()
@classmethod
def tearDownClass(cls):
cls.stop_tracker()
shutil.rmtree(settings.DOWNLOAD_DIR, True)
def setUp(self):
shutil.rmtree(settings.DOWNLOAD_DIR, True)
self.configure_session()
try:
os.makedirs(settings.DOWNLOAD_DIR)
except OSError:
pass
def tearDown(self):
self.stop_tracker()
def test_create_torrent(self):
torrent_content = TORRENT_PATH
self.obj = Torrent(
source_type="TORRENT",
source=torrent_content,
name="test movie")
self.obj.bind_session(self.session)
self.assertIsInstance(self.obj, Torrent)
self.assertIsInstance(self.obj.get_torrent_info(), lt.torrent_info)
def test_create_magnet_torrent(self):
self.obj = Torrent(
source_type="MAGNET",
source=MAGNET,
name="test movie")
self.obj.bind_session(self.session)
self.assertIsInstance(self.obj, Torrent)
# we need to wait to download metadata
self.assertRaises(TorrentHasNotMetadataYet, self.obj.get_torrent_info)
count = 0
while not self.obj.has_torrent_info():
time.sleep(0.2)
count += 1
if count > 300:
raise Exception("counter exceeded")
self.assertIsInstance(self.obj.get_torrent_info(), lt.torrent_info)
def test_get_filelist(self):
self.test_create_torrent()
files = self.obj.get_filelist()
self.assertEqual(len(files), 3)
self.assertEqual(files[0], "movie/movie.mp4")
self.assertEqual(files[1], "movie/music.mp3")
self.assertEqual(files[2], "movie/subs.txt")
def test_get_movie_filelist(self):
self.test_create_torrent()
files = self.obj.get_movies()
self.assertEqual(len(files), 1)
self.assertEqual(list(files)[0].path, "movie/movie.mp4")
def test_download_file(self):
# it works! but i have no idea why it not work with localhost tracker
return
self.test_create_torrent()
print("created. Go further")
time.sleep(0.5)
filename = "movie/movie.mp4"
# print(self.obj.get_movies()[0].__dict__)
self.obj.download_file(filename)
while True:
time.sleep(0.5)
# print(self.obj.get_movies()[0].cur_first_piece)
# print(self.obj.get_torrent_info().files()[0].size)
# print(self.__class__.h.status().state)
# print(self.__class__.h.status().progress)
# print(self.obj.get_status())
# print(self.obj.torrent_handler.status().state)
print(self.obj.torrent_handler.status().progress * 100)
class TorrentObjectTestCase(AbstractTorrentTestCase):
@classmethod
def setUpClass(cls):
cls.run_tracker()
cls.run_seeder()
@classmethod
def tearDownClass(cls):
cls.stop_tracker()
shutil.rmtree(settings.DOWNLOAD_DIR, True)
def setUp(self):
shutil.rmtree(settings.DOWNLOAD_DIR, True)
self.configure_session()
try:
os.makedirs(settings.DOWNLOAD_DIR)
except OSError:
pass
def tearDown(self):
self.stop_tracker()
def test_create_torrent(self):
torrent_content = TORRENT_PATH
self.obj = Torrent(source_type="TORRENT",
source=torrent_content,
name="test movie")
self.obj.bind_session(self.session)
self.assertIsInstance(self.obj, Torrent)
self.assertIsInstance(self.obj.get_torrent_info(), lt.torrent_info)
def test_create_magnet_torrent(self):
self.obj = Torrent(source_type="MAGNET",
source=MAGNET,
name="test movie")
self.obj.bind_session(self.session)
self.assertIsInstance(self.obj, Torrent)
# we need to wait to download metadata
self.assertRaises(TorrentHasNotMetadataYet, self.obj.get_torrent_info)
count = 0
while not self.obj.has_torrent_info():
time.sleep(0.2)
count += 1
if count > 300:
raise Exception("counter exceeded")
self.assertIsInstance(self.obj.get_torrent_info(), lt.torrent_info)
def test_get_filelist(self):
self.test_create_torrent()
files = self.obj.get_filelist()
self.assertEqual(len(files), 3)
self.assertEqual(files[0], "movie/movie.mp4")
self.assertEqual(files[1], "movie/music.mp3")
self.assertEqual(files[2], "movie/subs.txt")
def test_get_movie_filelist(self):
self.test_create_torrent()
files = self.obj.get_movies()
self.assertEqual(len(files), 1)
self.assertEqual(list(files)[0].path, "movie/movie.mp4")
def test_download_file(self):
# it works! but i have no idea why it not work with localhost tracker
return
self.test_create_torrent()
print("created. Go further")
time.sleep(0.5)
filename = "movie/movie.mp4"
# print(self.obj.get_movies()[0].__dict__)
self.obj.download_file(filename)
while True:
time.sleep(0.5)
# print(self.obj.get_movies()[0].cur_first_piece)
# print(self.obj.get_torrent_info().files()[0].size)
# print(self.__class__.h.status().state)
# print(self.__class__.h.status().progress)
# print(self.obj.get_status())
# print(self.obj.torrent_handler.status().state)
print(self.obj.torrent_handler.status().progress * 100)
