wornvirtue
/
premiere-libtorrent
forked from premiere/premiere-libtorrent
1
0
Fork 0
Code Pull Requests Activity

fix web_seed unit test

This commit is contained in:
Arvid Norberg 2015-01-09 14:02:48 +00:00
parent fb058f602f
commit 7761cba38a
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
test/web_server.py
View File

@ -73,9 +73,10 @@ class http_handler(SimpleHTTPServer.SimpleHTTPRequestHandler):
piece = int(args['piece'])
ranges = args['ranges'].split('-')
filename = ''
try:
filename = s.path[1:s.path.find('seed?') + 4].replace('/', '\\')
#print 'filename = %s' % filename
filename = os.path.normpath(s.path[1:s.path.find('seed?') + 4])
print 'filename = %s' % filename
f = open(filename, 'rb')
f.seek(piece * 64 * 1024 + int(ranges[0]))
data = f.read(int(ranges[1]) - int(ranges[0]) + 1)
@ -87,13 +88,14 @@ class http_handler(SimpleHTTPServer.SimpleHTTPRequestHandler):
s.end_headers()
s.wfile.write(data);
except Exception, e:
print 'FILE NOT FOUND: ', os.getcwd(), filename
print 'FILE ERROR: ', filename, e
s.send_response(404)
s.send_header("Content-Length", "0")
s.end_headers()
else:
filename = ''
try:
filename = file_path[1:].replace('/', '\\')
filename = os.path.normpath(file_path[1:])
# serve file by invoking default handler
f = open(filename, 'rb')
size = int(os.stat(filename).st_size)
@ -145,7 +147,7 @@ class http_handler(SimpleHTTPServer.SimpleHTTPRequestHandler):
if chunked_encoding:
s.wfile.write('0\r\n\r\n')
except Exception, e:
print 'FILE NOT FOUND: ', os.getcwd(), e
print 'FILE ERROR: ', filename, e
s.send_response(404)
s.send_header("Content-Length", "0")
s.end_headers()