]> granicus.if.org Git - python/commitdiff
Merged revisions 76040 via svnmerge from
authorAntoine Pitrou <solipsis@pitrou.net>
Sun, 1 Nov 2009 22:28:14 +0000 (22:28 +0000)
committerAntoine Pitrou <solipsis@pitrou.net>
Sun, 1 Nov 2009 22:28:14 +0000 (22:28 +0000)
svn+ssh://pythondev@svn.python.org/python/branches/py3k

................
  r76040 | antoine.pitrou | 2009-11-01 23:13:48 +0100 (dim., 01 nov. 2009) | 9 lines

  Merged revisions 76037 via svnmerge from
  svn+ssh://pythondev@svn.python.org/python/trunk

  ........
    r76037 | antoine.pitrou | 2009-11-01 23:02:03 +0100 (dim., 01 nov. 2009) | 3 lines

    Use a custom timeout in test_support.open_urlresource.
  ........
................

Lib/test/support.py

index 5b4cc5b5329615eea18b5821ff7d8ff129df2915..5672a20c9a70ffe8c4a425d465dca382f0cd407f 100644 (file)
@@ -455,9 +455,16 @@ def open_urlresource(url, *args, **kw):
             return open(fn, *args, **kw)
 
     print('\tfetching %s ...' % url, file=get_original_stdout())
-    fn, _ = urllib.request.urlretrieve(url, filename)
-    return open(fn, *args, **kw)
-
+    f = urllib.request.urlopen(url, timeout=15)
+    try:
+        with open(filename, "wb") as out:
+            s = f.read()
+            while s:
+                out.write(s)
+                s = f.read()
+    finally:
+        f.close()
+    return open(filename, *args, **kw)
 
 class WarningsRecorder(object):
     """Convenience wrapper for the warnings list returned on