Coverage for 'gcloud._apitools.http_wrapper.GetHttp'.

tseaver · tseaver · commit f97ede251c84 · 2015-10-26T13:13:27.000-04:00
diff --git a/gcloud/_apitools/test_http_wrapper.py b/gcloud/_apitools/test_http_wrapper.py
@@ -726,7 +726,7 @@ def _retry(args):
             self.assertEqual(retry.max_retry_wait, WAIT)
 
 
-class Test_MakeRequest(unittest2.TestCase):
+class Test__RegisterHttpFactory(unittest2.TestCase):
 
     def _callFUT(self, *args, **kw):
         from gcloud._apitools.http_wrapper import _RegisterHttpFactory
@@ -736,12 +736,59 @@ def test_it(self):
         from gcloud._testing import _Monkey
         from gcloud._apitools import http_wrapper as MUT
         _factories = []
+
         def _factory(**kw):
             pass
+
         with _Monkey(MUT, _HTTP_FACTORIES=_factories):
             self._callFUT(_factory)
             self.assertEqual(_factories, [_factory])
 
+
+class Test_GetHttp(unittest2.TestCase):
+
+    def _callFUT(self, *args, **kw):
+        from gcloud._apitools.http_wrapper import GetHttp
+        return GetHttp(*args, **kw)
+
+    def test_wo_registered_factories(self):
+        from httplib2 import Http
+        from gcloud._testing import _Monkey
+        from gcloud._apitools import http_wrapper as MUT
+        _factories = []
+
+        with _Monkey(MUT, _HTTP_FACTORIES=_factories):
+            http = self._callFUT()
+
+        self.assertTrue(isinstance(http, Http))
+
+    def test_w_registered_factories(self):
+        from gcloud._testing import _Monkey
+        from gcloud._apitools import http_wrapper as MUT
+
+        FOUND = object()
+
+        _misses = []
+
+        def _miss(**kw):
+            _misses.append(kw)
+            return None
+
+        _hits = []
+
+        def _hit(**kw):
+            _hits.append(kw)
+            return FOUND
+
+        _factories = [_miss, _hit]
+
+        with _Monkey(MUT, _HTTP_FACTORIES=_factories):
+            http = self._callFUT(foo='bar')
+
+        self.assertTrue(http is FOUND)
+        self.assertEqual(_misses, [{'foo': 'bar'}])
+        self.assertEqual(_hits, [{'foo': 'bar'}])
+
 class _Dummy(object):
     def __init__(self, **kw):
         self.__dict__.update(kw)
