mirror of https://github.com/python/cpython
Add simple unittests for Request
This commit is contained in:
parent
16a2a1d2ce
commit
1868d7c221
|
@ -1104,6 +1104,51 @@ class MiscTests(unittest.TestCase):
|
||||||
else:
|
else:
|
||||||
self.assert_(False)
|
self.assert_(False)
|
||||||
|
|
||||||
|
class RequestTests(unittest.TestCase):
|
||||||
|
|
||||||
|
def setUp(self):
|
||||||
|
self.get = urllib2.Request("http://www.python.org/~jeremy/")
|
||||||
|
self.post = urllib2.Request("http://www.python.org/~jeremy/",
|
||||||
|
"data",
|
||||||
|
headers={"X-Test": "test"})
|
||||||
|
|
||||||
|
def test_method(self):
|
||||||
|
self.assertEqual("POST", self.post.get_method())
|
||||||
|
self.assertEqual("GET", self.get.get_method())
|
||||||
|
|
||||||
|
def test_add_data(self):
|
||||||
|
self.assert_(not self.get.has_data())
|
||||||
|
self.assertEqual("GET", self.get.get_method())
|
||||||
|
self.get.add_data("spam")
|
||||||
|
self.assert_(self.get.has_data())
|
||||||
|
self.assertEqual("POST", self.get.get_method())
|
||||||
|
|
||||||
|
def test_get_full_url(self):
|
||||||
|
self.assertEqual("http://www.python.org/~jeremy/",
|
||||||
|
self.get.get_full_url())
|
||||||
|
|
||||||
|
def test_selector(self):
|
||||||
|
self.assertEqual("/~jeremy/", self.get.get_selector())
|
||||||
|
req = urllib2.Request("http://www.python.org/")
|
||||||
|
self.assertEqual("/", req.get_selector())
|
||||||
|
|
||||||
|
def test_get_type(self):
|
||||||
|
self.assertEqual("http", self.get.get_type())
|
||||||
|
|
||||||
|
def test_get_host(self):
|
||||||
|
self.assertEqual("www.python.org", self.get.get_host())
|
||||||
|
|
||||||
|
def test_get_host_unquote(self):
|
||||||
|
req = urllib2.Request("http://www.%70ython.org/")
|
||||||
|
self.assertEqual("www.python.org", req.get_host())
|
||||||
|
|
||||||
|
def test_proxy(self):
|
||||||
|
self.assert_(not self.get.has_proxy())
|
||||||
|
self.get.set_proxy("www.perl.org", "http")
|
||||||
|
self.assert_(self.get.has_proxy())
|
||||||
|
self.assertEqual("www.python.org", self.get.get_origin_req_host())
|
||||||
|
self.assertEqual("www.perl.org", self.get.get_host())
|
||||||
|
|
||||||
|
|
||||||
def test_main(verbose=None):
|
def test_main(verbose=None):
|
||||||
from test import test_urllib2
|
from test import test_urllib2
|
||||||
|
@ -1112,7 +1157,8 @@ def test_main(verbose=None):
|
||||||
tests = (TrivialTests,
|
tests = (TrivialTests,
|
||||||
OpenerDirectorTests,
|
OpenerDirectorTests,
|
||||||
HandlerTests,
|
HandlerTests,
|
||||||
MiscTests)
|
MiscTests,
|
||||||
|
RequestTests)
|
||||||
test_support.run_unittest(*tests)
|
test_support.run_unittest(*tests)
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|
Loading…
Reference in New Issue