From 72b5c98e8da0fa3d4b6f879aeb1887228645ee48 Mon Sep 17 00:00:00 2001 From: James Rowe Date: Thu, 30 Jun 2011 12:24:07 -0700 Subject: [PATCH] Python 2.7 create_connection compatibility fix. --- httpretty/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/httpretty/__init__.py b/httpretty/__init__.py index 6fff814..4fcf314 100644 --- a/httpretty/__init__.py +++ b/httpretty/__init__.py @@ -255,11 +255,12 @@ def fake_wrap_socket(s, *args, **kw): return s -def create_fake_connection(address, timeout=socket._GLOBAL_DEFAULT_TIMEOUT): +def create_fake_connection(address, timeout=socket._GLOBAL_DEFAULT_TIMEOUT, source_address=None): s = fakesock.socket(socket.AF_INET, socket.SOCK_STREAM, socket.IPPROTO_TCP) if timeout is not socket._GLOBAL_DEFAULT_TIMEOUT: s.settimeout(timeout) - + if source_address: + s.bind(source_address) s.connect(address) return s