check_memory: stop using long-deprecated twisted internals
authorBrian Warner <warner@lothar.com>
Wed, 3 Sep 2014 01:22:59 +0000 (18:22 -0700)
committerBrian Warner <warner@lothar.com>
Wed, 3 Sep 2014 01:22:59 +0000 (18:22 -0700)
src/allmydata/test/check_memory.py

index ce6b33a7c380c6baf01ade6f0752d1ea093433fa..2cca811277928ffc0dfb9a97ea9e64dc113d928d 100644 (file)
@@ -1,4 +1,4 @@
-import os, shutil, sys, urllib, time, stat
+import os, shutil, sys, urllib, time, stat, urlparse
 from cStringIO import StringIO
 from twisted.internet import defer, reactor, protocol, error
 from twisted.application import service, internet
@@ -49,10 +49,14 @@ def discardPage(url, stall=False, *args, **kwargs):
     # adapted from twisted.web.client.getPage . We can't just wrap or
     # subclass because it provides no way to override the HTTPClientFactory
     # that it creates.
-    scheme, host, port, path = tw_client._parse(url)
+    scheme, netloc, path, params, query, fragment = urlparse.urlparse(url)
+    assert scheme == 'http'
+    host, port = netloc, 80
+    if ":" in host:
+        host, port = host.split(":")
+        port = int(port)
     factory = StallableDiscardingHTTPClientFactory(url, *args, **kwargs)
     factory.do_stall = stall
-    assert scheme == 'http'
     reactor.connectTCP(host, port, factory)
     return factory.deferred