summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
authorGiampaolo Rodola <g.rodola@gmail.com>2011-02-10 09:57:06 +0000
committerGiampaolo Rodola <g.rodola@gmail.com>2011-02-10 09:57:06 +0000
commit6c3f51129713a3fea8239903d688e2e74b570a1b (patch)
treef147b3e65f323a83c5675bcecfdae97357817fe1 /test
parentfc3b4ab21d63209e2ee9f8c82b90d18748be9dbe (diff)
downloadpysendfile-6c3f51129713a3fea8239903d688e2e74b570a1b.tar.gz
ignore me
Diffstat (limited to 'test')
-rw-r--r--test/benchmark.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/test/benchmark.py b/test/benchmark.py
index 1740099..09d1e0c 100644
--- a/test/benchmark.py
+++ b/test/benchmark.py
@@ -73,7 +73,7 @@ class Client:
return bytes_recv
-def start_server(use_sendfile=True, keep_sending=False):
+def start_server(use_sendfile, keep_sending=False):
sock = socket.socket()
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
sock.bind((HOST, PORT))
@@ -110,7 +110,8 @@ def start_server(use_sendfile=True, keep_sending=False):
continue
else:
break
- offset += sent
+ else:
+ offset += sent
file.close()
conn.close()
@@ -120,8 +121,7 @@ def main():
print "creating big file . . ."
create_file(BIGFILE, BIGFILE_SIZE)
print "starting benchmark . . .\n"
-
- #atexit.register(lambda: os.remove(BIGFILE))
+ atexit.register(lambda: os.remove(BIGFILE))
# CPU time: use sendfile()
server = Process(target=start_server, kwargs={"use_sendfile":True})