]> granicus.if.org Git - python/commitdiff
Issue12541 - Add UserWarning for unquoted realms
authorSenthil Kumaran <senthil@uthcode.com>
Tue, 15 May 2012 15:59:19 +0000 (23:59 +0800)
committerSenthil Kumaran <senthil@uthcode.com>
Tue, 15 May 2012 15:59:19 +0000 (23:59 +0800)
Lib/test/test_urllib2.py
Lib/urllib2.py

index ad558dac300c52224ee0c7062ed3e141225fe253..fc76a6ccb923591d6240f0bd9a5e2105c0399ab7 100644 (file)
@@ -1106,8 +1106,8 @@ class HandlerTests(unittest.TestCase):
         self._test_basic_auth(opener, auth_handler, "Authorization",
                               realm, http_handler, password_manager,
                               "http://acme.example.com/protected",
-                              "http://acme.example.com/protected",
-                              )
+                              "http://acme.example.com/protected"
+                             )
 
     def test_basic_auth_with_single_quoted_realm(self):
         self.test_basic_auth(quote_char="'")
@@ -1121,11 +1121,13 @@ class HandlerTests(unittest.TestCase):
             401, 'WWW-Authenticate: Basic realm=%s\r\n\r\n' % realm)
         opener.add_handler(auth_handler)
         opener.add_handler(http_handler)
-        self._test_basic_auth(opener, auth_handler, "Authorization",
-                              realm, http_handler, password_manager,
-                              "http://acme.example.com/protected",
-                              "http://acme.example.com/protected",
-                              )
+        msg = "Basic Auth Realm was unquoted"
+        with test_support.check_warnings((msg, UserWarning)):
+            self._test_basic_auth(opener, auth_handler, "Authorization",
+                                  realm, http_handler, password_manager,
+                                  "http://acme.example.com/protected",
+                                  "http://acme.example.com/protected"
+                                 )
 
 
     def test_proxy_basic_auth(self):
index 21b5262fea943ffe213f2d5dac4c1c9f6d7dd5a0..d4596cdf57746bf6778af1d5be570b12d848e849 100644 (file)
@@ -102,6 +102,7 @@ import sys
 import time
 import urlparse
 import bisect
+import warnings
 
 try:
     from cStringIO import StringIO
@@ -861,6 +862,9 @@ class AbstractBasicAuthHandler:
             mo = AbstractBasicAuthHandler.rx.search(authreq)
             if mo:
                 scheme, quote, realm = mo.groups()
+                if quote not in ['"', "'"]:
+                    warnings.warn("Basic Auth Realm was unquoted",
+                                  UserWarning, 2)
                 if scheme.lower() == 'basic':
                     response = self.retry_http_basic_auth(host, req, realm)
                     if response and response.code != 401: