1 files changed, 43 insertions, 1 deletions
diff --git a/python3/httplib2test.py b/python3/httplib2test.py
index a3e29b2..1a92726 100755
--- a/python3/httplib2test.py
+++ b/python3/httplib2test.py
@@ -1535,4 +1535,46 @@ class HttpPrivateTest(unittest.TestCase):
         end2end = httplib2._get_end2end_headers(response)
         self.assertEqual(0, len(end2end))
 
-unittest.main()
+
+class TestProxyInfo(unittest.TestCase):
+    def setUp(self):
+        self.orig_env = dict(os.environ)
+
+    def tearDown(self):
+        os.environ.clear()
+        os.environ.update(self.orig_env)
+
+    def test_from_url(self):
+        pi = httplib2.proxy_info_from_url('http://myproxy.example.com')
+        self.assertEqual(pi.proxy_host, 'myproxy.example.com')
+        self.assertEqual(pi.proxy_port, 80)
+        self.assertEqual(pi.proxy_user, None)
+
+    def test_from_url_ident(self):
+        pi = httplib2.proxy_info_from_url('http://zoidberg:fish@someproxy:99')
+        self.assertEqual(pi.proxy_host, 'someproxy')
+        self.assertEqual(pi.proxy_port, 99)
+        self.assertEqual(pi.proxy_user, 'zoidberg')
+        self.assertEqual(pi.proxy_pass, 'fish')
+
+    def test_from_env(self):
+        os.environ['http_proxy'] = 'http://myproxy.example.com:8080'
+        pi = httplib2.proxy_info_from_environment()
+        self.assertEqual(pi.proxy_host, 'myproxy.example.com')
+        self.assertEqual(pi.proxy_port, 8080)
+
+    def test_from_env_no_proxy(self):
+        os.environ['http_proxy'] = 'http://myproxy.example.com:80'
+        os.environ['https_proxy'] = 'http://myproxy.example.com:81'
+        pi = httplib2.proxy_info_from_environment('https')
+        self.assertEqual(pi.proxy_host, 'myproxy.example.com')
+        self.assertEqual(pi.proxy_port, 81)
+
+    def test_from_env_none(self):
+        os.environ.clear()
+        pi = httplib2.proxy_info_from_environment()
+        self.assertEqual(pi, None)
+
+
+if __name__ == '__main__':
+    unittest.main()