diff options
author | R David Murray <rdmurray@bitdance.com> | 2014-06-11 13:48:58 -0400 |
---|---|---|
committer | R David Murray <rdmurray@bitdance.com> | 2014-06-11 13:48:58 -0400 |
commit | 6fe56a329dd427f2a5363b43ad24f04785d2091d (patch) | |
tree | af2b051bbdb771ad2932d6888b9405d7984ff9b6 /Lib/test/mock_socket.py | |
parent | 1144da5821553449739744f08d041b4ba7114ecb (diff) | |
download | cpython-git-6fe56a329dd427f2a5363b43ad24f04785d2091d.tar.gz |
#14758: add IPv6 support to smtpd.
Patch by Milan Oberkirch.
Diffstat (limited to 'Lib/test/mock_socket.py')
-rw-r--r-- | Lib/test/mock_socket.py | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/Lib/test/mock_socket.py b/Lib/test/mock_socket.py index e36724f54b..a4fbca6369 100644 --- a/Lib/test/mock_socket.py +++ b/Lib/test/mock_socket.py @@ -35,8 +35,9 @@ class MockFile: class MockSocket: """Mock socket object used by smtpd and smtplib tests. """ - def __init__(self): + def __init__(self, family=None): global _reply_data + self.family = family self.output = [] self.lines = [] if _reply_data: @@ -108,8 +109,7 @@ class MockSocket: def socket(family=None, type=None, proto=None): - return MockSocket() - + return MockSocket(family) def create_connection(address, timeout=socket_module._GLOBAL_DEFAULT_TIMEOUT, source_address=None): @@ -144,13 +144,16 @@ def gethostname(): def gethostbyname(name): return "" +def getaddrinfo(host, port): + return socket_module.getaddrinfo(host, port) gaierror = socket_module.gaierror error = socket_module.error # Constants -AF_INET = None +AF_INET = socket_module.AF_INET +AF_INET6 = socket_module.AF_INET6 SOCK_STREAM = None SOL_SOCKET = None SO_REUSEADDR = None |