summaryrefslogtreecommitdiff
path: root/Lib/test/mock_socket.py
diff options
context:
space:
mode:
authorR David Murray <rdmurray@bitdance.com>2014-06-11 13:48:58 -0400
committerR David Murray <rdmurray@bitdance.com>2014-06-11 13:48:58 -0400
commit6fe56a329dd427f2a5363b43ad24f04785d2091d (patch)
treeaf2b051bbdb771ad2932d6888b9405d7984ff9b6 /Lib/test/mock_socket.py
parent1144da5821553449739744f08d041b4ba7114ecb (diff)
downloadcpython-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.py11
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