1 files changed, 22 insertions, 1 deletions
diff --git a/tests/unit/cloudsearch2/test_document.py b/tests/unit/cloudsearch2/test_document.py
index f9d7d97f..dac4aa64 100644
--- a/tests/unit/cloudsearch2/test_document.py
+++ b/tests/unit/cloudsearch2/test_document.py
@@ -1,6 +1,8 @@
 #!/usr/bin env python
+from boto.cloudsearch2.domain import Domain
+from boto.cloudsearch2.layer1 import CloudSearchConnection
 
-from tests.unit import unittest
+from tests.unit import unittest, AWSMockServiceTestCase
 from httpretty import HTTPretty
 from mock import MagicMock
 
@@ -11,6 +13,24 @@ from boto.cloudsearch2.document import CommitMismatchError, EncodingError, \
         ContentTooLongError, DocumentServiceConnection
 
 import boto
+from tests.unit.cloudsearch2 import DEMO_DOMAIN_DATA
+
+
+class CloudSearchDocumentConnectionTest(AWSMockServiceTestCase):
+    connection_class = CloudSearchConnection
+
+    def test_proxy(self):
+        conn = self.service_connection
+        conn.proxy = "127.0.0.1"
+        conn.proxy_user = "john.doe"
+        conn.proxy_pass="p4ssw0rd"
+        conn.proxy_port="8180"
+        conn.use_proxy = True
+
+        domain = Domain(conn, DEMO_DOMAIN_DATA)
+        service = DocumentServiceConnection(domain=domain)
+        self.assertEqual(service.proxy, {'http': 'http://john.doe:p4ssw0rd@127.0.0.1:8180'})
+
 
 class CloudSearchDocumentTest(unittest.TestCase):
     def setUp(self):
@@ -25,6 +45,7 @@ class CloudSearchDocumentTest(unittest.TestCase):
     def tearDown(self):
         HTTPretty.disable()
 
+
 class CloudSearchDocumentSingleTest(CloudSearchDocumentTest):
 
     response = {