Add test for proxy info

skimbrel · skimbrel · commit d13d45e8d62c · 2013-09-23T17:54:07.000-07:00
diff --git a/tests/test_make_request.py b/tests/test_make_request.py
@@ -4,10 +4,11 @@
 Uses the awesome httpbin.org to validate responses
 """
 import twilio
-from nose.tools import raises
-from mock import patch, Mock
+from nose.tools import assert_equal, raises
+from mock import patch, Mock, ANY
 from twilio import TwilioRestException
 from twilio.rest.resources.base import make_request, make_twilio_request
+from twilio.rest.resources.connection import Connection
 
 get_headers = {
     "User-Agent": "twilio-python/%s" % (twilio.__version__),
@@ -90,3 +91,19 @@ def test_make_twilio_request_bad_data(mock):
     make_twilio_request("POST", url)
     mock.assert_called_with("POST", "http://random/url.json",
                             headers=post_headers)
+
+
+@patch('twilio.rest.resources.base.Response')
+@patch('httplib2.Http')
+def test_proxy_info(http_mock, resp_mock):
+    http = Mock()
+    http.request.return_value = (Mock(), Mock())
+    http_mock.return_value = http
+    Connection.set_proxy_info('http://example.com/proxy', '8080')
+    make_request("GET", "http://httpbin.org/get")
+    http_mock.assert_called_with(timeout=None, ca_certs=ANY, proxy_info=ANY)
+    http.request.assert_called_with("http://httpbin.org/get", "GET",
+                                    body=None, headers=None)
+    proxy_info = http_mock.call_args[1]['proxy_info']
+    assert_equal(proxy_info.proxy_host, 'http://example.com/proxy')
+    assert_equal(proxy_info.proxy_port, '8080')
