Make sure we can pass the connection pool into the client.

twisted · Jan 23, 2013 · a29f5c2 · a29f5c2
1 parent 06053e4
commit a29f5c2
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/treq/client.py b/treq/client.py
@@ -71,11 +71,12 @@ def with_config(cls, **kwargs):
         if not reactor:
             from twisted.internet import reactor
 
-        agent = Agent(
-            reactor,
-            pool=HTTPConnectionPool(
-                reactor,
-                persistent=kwargs.get('persistent', True)))
+        pool = kwargs.get('pool')
+        if not pool:
+            persistent = kwargs.get('persistent', True)
+            pool = HTTPConnectionPool(reactor, persitent=persistent)
+
+        agent = Agent(reactor, pool=pool)
 
         if kwargs.get('allow_redirects', True):
             agent = RedirectAgent(agent)