summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_urllibnet.py
diff options
context:
space:
mode:
authorBrett Cannon <bcannon@gmail.com>2003-05-14 02:18:31 (GMT)
committerBrett Cannon <bcannon@gmail.com>2003-05-14 02:18:31 (GMT)
commita71319eebbc2a35ab8df2c78ced364583f3608f3 (patch)
tree10f4f07f4248c92965cd5103ba4382ab72fbb941 /Lib/test/test_urllibnet.py
parentabe8eb0a3044fd1a23a2a46cc4bb61e4e022b7c0 (diff)
downloadcpython-a71319eebbc2a35ab8df2c78ced364583f3608f3.zip
cpython-a71319eebbc2a35ab8df2c78ced364583f3608f3.tar.gz
cpython-a71319eebbc2a35ab8df2c78ced364583f3608f3.tar.bz2
Fleshed out tests for urllib requiring a network connection.
Diffstat (limited to 'Lib/test/test_urllibnet.py')
-rw-r--r--Lib/test/test_urllibnet.py126
1 files changed, 123 insertions, 3 deletions
diff --git a/Lib/test/test_urllibnet.py b/Lib/test/test_urllibnet.py
index 03728e1..f38a2d0 100644
--- a/Lib/test/test_urllibnet.py
+++ b/Lib/test/test_urllibnet.py
@@ -4,8 +4,10 @@ import unittest
from test import test_support
import socket
-import urllib2
+import urllib
import sys
+import os
+import mimetools
class URLTimeoutTest(unittest.TestCase):
@@ -18,12 +20,130 @@ class URLTimeoutTest(unittest.TestCase):
socket.setdefaulttimeout(None)
def testURLread(self):
- f = urllib2.urlopen("http://www.python.org/")
+ f = urllib.urlopen("http://www.python.org/")
x = f.read()
+class urlopenNetworkTests(unittest.TestCase):
+ """Tests urllib.urlopen using the network.
+
+ These tests are not exhaustive. Assuming that testing using files does a
+ good job overall of some of the basic interface features. There are no
+ tests exercising the optional 'data' and 'proxies' arguments. No tests
+ for transparent redirection have been written.
+
+ setUp is not used for always constructing a connection to
+ http://www.python.org/ since there a few tests that don't use that address
+ and making a connection is expensive enough to warrant minimizing unneeded
+ connections.
+
+ """
+
+ def test_basic(self):
+ # Simple test expected to pass.
+ open_url = urllib.urlopen("http://www.python.org/")
+ for attr in ("read", "readline", "readlines", "fileno", "close",
+ "info", "geturl"):
+ self.assert_(hasattr(open_url, attr), "object returned from "
+ "urlopen lacks the %s attribute" % attr)
+ try:
+ self.assert_(open_url.read(), "calling 'read' failed")
+ finally:
+ open_url.close()
+
+ def test_readlines(self):
+ # Test both readline and readlines.
+ open_url = urllib.urlopen("http://www.python.org/")
+ try:
+ self.assert_(isinstance(open_url.readline(), basestring),
+ "readline did not return a string")
+ self.assert_(isinstance(open_url.readlines(), list),
+ "readlines did not return a list")
+ finally:
+ open_url.close()
+
+ def test_info(self):
+ # Test 'info'.
+ open_url = urllib.urlopen("http://www.python.org/")
+ try:
+ info_obj = open_url.info()
+ finally:
+ open_url.close()
+ self.assert_(isinstance(info_obj, mimetools.Message),
+ "object returned by 'info' is not an instance of "
+ "mimetools.Message")
+ self.assertEqual(info_obj.getsubtype(), "html")
+
+ def test_geturl(self):
+ # Make sure same URL as opened is returned by geturl.
+ URL = "http://www.python.org/"
+ open_url = urllib.urlopen(URL)
+ try:
+ gotten_url = open_url.geturl()
+ finally:
+ open_url.close()
+ self.assertEqual(gotten_url, URL)
+
+ def test_fileno(self):
+ # Make sure fd returned by fileno is valid.
+ if hasattr(os, 'fdopen'):
+ open_url = urllib.urlopen("http://www.python.org/")
+ fd = open_url.fileno()
+ FILE = os.fdopen(fd)
+ try:
+ self.assert_(FILE.read(), "reading from file created using fd "
+ "returned by fileno failed")
+ finally:
+ FILE.close()
+
+ def test_bad_address(self):
+ # Make sure proper exception is raised when connecting to a bogus
+ # address.
+ self.assertRaises(IOError,
+ urllib.urlopen, "http://www.sadflkjsasadf.com/")
+
+class urlretrieveNetworkTests(unittest.TestCase):
+ """Tests urllib.urlretrieve using the network."""
+
+ def test_basic(self):
+ # Test basic functionality.
+ file_location,info = urllib.urlretrieve("http://www.python.org/")
+ self.assert_(os.path.exists(file_location), "file location returned by"
+ " urlretrieve is not a valid path")
+ FILE = file(file_location)
+ try:
+ self.assert_(FILE.read(), "reading from the file location returned "
+ "by urlretrieve failed")
+ finally:
+ FILE.close()
+ os.unlink(file_location)
+
+ def test_specified_path(self):
+ # Make sure that specifying the location of the file to write to works.
+ file_location,info = urllib.urlretrieve("http://www.python.org/",
+ test_support.TESTFN)
+ self.assertEqual(file_location, test_support.TESTFN)
+ self.assert_(os.path.exists(file_location))
+ FILE = file(file_location)
+ try:
+ self.assert_(FILE.read(), "reading from temporary file failed")
+ finally:
+ FILE.close()
+ os.unlink(file_location)
+
+ def test_header(self):
+ # Make sure header returned as 2nd value from urlretrieve is good.
+ file_location, header = urllib.urlretrieve("http://www.python.org/")
+ os.unlink(file_location)
+ self.assert_(isinstance(header, mimetools.Message),
+ "header is not an instance of mimetools.Message")
+
+
+
def test_main():
test_support.requires('network')
- test_support.run_unittest(URLTimeoutTest)
+ test_support.run_unittest(URLTimeoutTest,
+ urlopenNetworkTests,
+ urlretrieveNetworkTests)
if __name__ == "__main__":
test_main()