diff options
Diffstat (limited to 'libpathod/test.py')
-rw-r--r-- | libpathod/test.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/libpathod/test.py b/libpathod/test.py index 00e03823..b90c8de6 100644 --- a/libpathod/test.py +++ b/libpathod/test.py @@ -6,9 +6,9 @@ import tutils IFACE = "127.0.0.1" class Daemon: - def __init__(self, staticdir=None, anchors=(), ssl=None): + def __init__(self, staticdir=None, anchors=(), ssl=None, sizelimit=None): self.q = Queue.Queue() - self.thread = PaThread(self.q, staticdir, anchors, ssl) + self.thread = PaThread(self.q, staticdir, anchors, ssl, sizelimit) self.thread.start() self.port = self.q.get(True, 5) self.urlbase = "%s://%s:%s"%("https" if ssl else "http", IFACE, self.port) @@ -43,9 +43,9 @@ class Daemon: class PaThread(threading.Thread): - def __init__(self, q, staticdir, anchors, ssl): + def __init__(self, q, staticdir, anchors, ssl, sizelimit): threading.Thread.__init__(self) - self.q, self.staticdir, self.anchors, self.ssl = q, staticdir, anchors, ssl + self.q, self.staticdir, self.anchors, self.ssl, self.sizelimit = q, staticdir, anchors, ssl, sizelimit def run(self): if self.ssl is True: @@ -59,7 +59,8 @@ class PaThread(threading.Thread): (IFACE, 0), ssloptions = ssloptions, anchors = self.anchors, - staticdir = self.staticdir + staticdir = self.staticdir, + sizelimit = self.sizelimit ) self.q.put(self.server.port) self.server.serve_forever() |