class TestSeeding(TestAsServer):
"""
Testing seeding via new tribler API:
"""
def setUp(self):
""" override TestAsServer """
TestAsServer.setUp(self)
self.session2 = None
self.seeding_event = threading.Event()
self.downloading_event = threading.Event()
def setUpPreSession(self):
""" override TestAsServer """
TestAsServer.setUpPreSession(self)
self.config.set_libtorrent(True)
self.config2 = self.config.copy() # not really necess
self.config2.set_state_dir(self.getStateDir(2))
self.config2.set_listen_port(4810)
self.dscfg2 = DownloadStartupConfig()
self.dscfg2.set_dest_dir(self.getDestDir(2))
def setUpPostSession(self):
pass
def tearDown(self):
if self.session2:
self._shutdown_session(self.session2)
time.sleep(10)
TestAsServer.tearDown(self)
def setup_seeder(self, merkle, filename='file.wmv'):
self.tdef = TorrentDef()
self.sourcefn = os.path.join(BASE_DIR, "API", filename)
self.tdef.add_content(self.sourcefn)
self.tdef.set_tracker("http://fake.net/announce")
self.tdef.set_create_merkle_torrent(merkle)
self.tdef.finalize()
self.torrentfn = os.path.join(self.session.get_state_dir(),
"gen.torrent")
self.tdef.save(self.torrentfn)
print >> sys.stderr, "test: setup_seeder: name is", self.tdef.metainfo[
'info']['name']
self.dscfg = DownloadStartupConfig()
self.dscfg.set_dest_dir(os.path.join(
BASE_DIR, "API")) # basedir of the file we are seeding
d = self.session.start_download(self.tdef, self.dscfg)
d.set_state_callback(self.seeder_state_callback)
print >> sys.stderr, "test: setup_seeder: starting to wait for download to reach seeding state"
assert self.seeding_event.wait(60)
def seeder_state_callback(self, ds):
d = ds.get_download()
print >> sys.stderr, "test: seeder:", repr(
d.get_def().get_name()), dlstatus_strings[
ds.get_status()], ds.get_progress()
if ds.get_status() == DLSTATUS_SEEDING:
self.seeding_event.set()
return (1.0, False)
def test_normal_torrent(self):
self.setup_seeder(False)
self.subtest_is_seeding()
self.subtest_download()
# def test_merkle_torrent(self):
# self.setup_seeder(True)
# self.subtest_is_seeding()
# self.subtest_download()
def subtest_is_seeding(self):
infohash = self.tdef.get_infohash()
s = BTConnection('localhost', self.hisport, user_infohash=infohash)
s.read_handshake_medium_rare()
s.send(CHOKE)
try:
s.s.settimeout(10.0)
resp = s.recv()
self.assert_(len(resp) > 0)
self.assert_(resp[0] == EXTEND)
except socket.timeout:
print >> sys.stderr, "test: Timeout, peer didn't reply"
self.assert_(False)
s.close()
def subtest_download(self):
""" Now download the file via another Session """
self.session2 = Session(self.config2, ignore_singleton=True)
self.session2.start()
time.sleep(5)
tdef2 = TorrentDef.load(self.torrentfn)
d = self.session2.start_download(tdef2, self.dscfg2)
d.set_state_callback(self.downloader_state_callback)
time.sleep(5)
d.add_peer(("127.0.0.1", self.hisport))
assert self.downloading_event.wait(60)
def downloader_state_callback(self, ds):
d = ds.get_download()
print >> sys.stderr, "test: download:", repr(
d.get_def().get_name()), dlstatus_strings[
ds.get_status()], ds.get_progress()
if ds.get_status() == DLSTATUS_SEEDING:
# File is in
destfn = os.path.join(self.getDestDir(2), "file.wmv")
f = open(destfn, "rb")
realdata = f.read()
f.close()
f = open(self.sourcefn, "rb")
expdata = f.read()
f.close()
self.assert_(realdata == expdata)
self.downloading_event.set()
return (1.0, True)
return (1.0, False)
class TestSeeding(TestAsServer):
"""
Testing seeding via new tribler API:
"""
def setUp(self):
""" override TestAsServer """
TestAsServer.setUp(self)
self.session2 = None
self.seeding_event = threading.Event()
self.downloading_event = threading.Event()
def setUpPreSession(self):
""" override TestAsServer """
TestAsServer.setUpPreSession(self)
self.config.set_libtorrent(True)
self.config2 = self.config.copy() # not really necess
self.config2.set_state_dir(self.getStateDir(2))
self.config2.set_listen_port(4810)
self.dscfg2 = DownloadStartupConfig()
self.dscfg2.set_dest_dir(self.getDestDir(2))
def setUpPostSession(self):
pass
def tearDown(self):
if self.session2:
self._shutdown_session(self.session2)
time.sleep(10)
TestAsServer.tearDown(self)
def setup_seeder(self, merkle, filename='file.wmv'):
self.tdef = TorrentDef()
self.sourcefn = os.path.join(BASE_DIR, "API", filename)
self.tdef.add_content(self.sourcefn)
self.tdef.set_tracker("http://fake.net/announce")
self.tdef.set_create_merkle_torrent(merkle)
self.tdef.finalize()
self.torrentfn = os.path.join(self.session.get_state_dir(), "gen.torrent")
self.tdef.save(self.torrentfn)
print >> sys.stderr, "test: setup_seeder: name is", self.tdef.metainfo['info']['name']
self.dscfg = DownloadStartupConfig()
self.dscfg.set_dest_dir(os.path.join(BASE_DIR, "API")) # basedir of the file we are seeding
d = self.session.start_download(self.tdef, self.dscfg)
d.set_state_callback(self.seeder_state_callback)
print >> sys.stderr, "test: setup_seeder: starting to wait for download to reach seeding state"
assert self.seeding_event.wait(60)
def seeder_state_callback(self, ds):
d = ds.get_download()
print >> sys.stderr, "test: seeder:", repr(d.get_def().get_name()), dlstatus_strings[ds.get_status()], ds.get_progress()
if ds.get_status() == DLSTATUS_SEEDING:
self.seeding_event.set()
return (1.0, False)
def test_normal_torrent(self):
self.setup_seeder(False)
self.subtest_is_seeding()
self.subtest_download()
# def test_merkle_torrent(self):
# self.setup_seeder(True)
# self.subtest_is_seeding()
# self.subtest_download()
def subtest_is_seeding(self):
infohash = self.tdef.get_infohash()
s = BTConnection('localhost', self.hisport, user_infohash=infohash)
s.read_handshake_medium_rare()
s.send(CHOKE)
try:
s.s.settimeout(10.0)
resp = s.recv()
self.assert_(len(resp) > 0)
self.assert_(resp[0] == EXTEND)
except socket.timeout:
print >> sys.stderr, "test: Timeout, peer didn't reply"
self.assert_(False)
s.close()
def subtest_download(self):
""" Now download the file via another Session """
self.session2 = Session(self.config2, ignore_singleton=True)
self.session2.start()
time.sleep(5)
tdef2 = TorrentDef.load(self.torrentfn)
d = self.session2.start_download(tdef2, self.dscfg2)
d.set_state_callback(self.downloader_state_callback)
time.sleep(5)
d.add_peer(("127.0.0.1", self.hisport))
assert self.downloading_event.wait(60)
def downloader_state_callback(self, ds):
d = ds.get_download()
print >> sys.stderr, "test: download:", repr(d.get_def().get_name()), dlstatus_strings[ds.get_status()], ds.get_progress()
if ds.get_status() == DLSTATUS_SEEDING:
# File is in
destfn = os.path.join(self.getDestDir(2), "file.wmv")
f = open(destfn, "rb")
realdata = f.read()
f.close()
f = open(self.sourcefn, "rb")
expdata = f.read()
f.close()
self.assert_(realdata == expdata)
self.downloading_event.set()
return (1.0, True)
return (1.0, False)
class TestAsServer(AbstractServer):
"""
Parent class for testing the server-side of Tribler
"""
def setUp(self):
self.setUpCleanup()
self.setUpPreSession()
self.quitting = False
self.session = Session(self.config)
self.session.start()
self.hisport = self.session.get_listen_port()
while not self.session.lm.initComplete:
time.sleep(1)
self.annotate(self._testMethodName, start=True)
def setUpPreSession(self):
""" Should set self.config_path and self.config """
self.config = SessionStartupConfig()
self.config.set_state_dir(self.getStateDir())
self.config.set_listen_port(random.randint(10000, 60000))
self.config.set_torrent_checking(False)
self.config.set_multicast_local_peer_discovery(False)
self.config.set_megacache(False)
self.config.set_dispersy(False)
self.config.set_swift_proc(False)
self.config.set_mainline_dht(False)
self.config.set_torrent_collecting(False)
self.config.set_libtorrent(False)
self.config.set_dht_torrent_collecting(False)
def tearDown(self):
self.annotate(self._testMethodName, start=False)
""" unittest test tear down code """
if self.session is not None:
self._shutdown_session(self.session)
Session.del_instance()
time.sleep(10)
gc.collect()
ts = enumerate_threads()
print >> sys.stderr, "test_as_server: Number of threads still running", len(ts)
for t in ts:
print >> sys.stderr, "test_as_server: Thread still running", t.getName(), "daemon", t.isDaemon(), "instance:", t
if SQLiteCacheDB.hasInstance():
SQLiteCacheDB.getInstance().close_all()
SQLiteCacheDB.delInstance()
self.tearDownCleanup()
def _shutdown_session(self, session):
session_shutdown_start = time.time()
waittime = 60
session.shutdown()
while not session.has_shutdown():
diff = time.time() - session_shutdown_start
assert diff < waittime, "test_as_server: took too long for Session to shutdown"
print >> sys.stderr, "test_as_server: ONEXIT Waiting for Session to shutdown, will wait for an additional %d seconds" % (waittime - diff)
time.sleep(1)
print >> sys.stderr, "test_as_server: Session is shutdown"
def assert_(self, boolean, reason=None, do_assert = True):
if not boolean:
self.quit()
assert boolean, reason
def startTest(self, callback):
self.quitting = False
callback()
def Call(self, seconds, callback):
if not self.quitting:
if seconds:
time.sleep(seconds)
callback()
def CallConditional(self, timeout, condition, callback, assertMsg=None):
t = time.time()
def DoCheck():
if not self.quitting:
if time.time() - t < timeout:
if condition():
print >> sys.stderr, "test_as_server: condition satisfied after %d seconds, calling callback" % (time.time() - t)
callback()
else:
self.Call(0.5, DoCheck)
else:
print >> sys.stderr, "test_as_server: quitting, condition was not satisfied in %d seconds (%s)" % (timeout, assertMsg or "no-assert-msg")
self.assert_(False, assertMsg if assertMsg else "Condition was not satisfied in %d seconds" % timeout, do_assert=False)
self.Call(0, DoCheck)
def quit(self):
self.quitting = True
