Skip to content

Commit

Permalink
Merge 3.5 (issue python#28652)
Browse files Browse the repository at this point in the history
  • Loading branch information
1st1 committed Nov 21, 2016
2 parents bf87126 + dab0584 commit 5cb0c09
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 18 deletions.
22 changes: 10 additions & 12 deletions Lib/asyncio/base_events.py
Original file line number Diff line number Diff line change
Expand Up @@ -98,14 +98,6 @@ def _is_dgram_socket(sock):
return (sock.type & socket.SOCK_DGRAM) == socket.SOCK_DGRAM


def _is_ip_socket(sock):
if sock.family == socket.AF_INET:
return True
if hasattr(socket, 'AF_INET6') and sock.family == socket.AF_INET6:
return True
return False


def _ipaddr_info(host, port, family, type, proto):
# Try to skip getaddrinfo if "host" is already an IP. Users might have
# handled name resolution in their own code and pass in resolved IPs.
Expand Down Expand Up @@ -796,9 +788,15 @@ def create_connection(self, protocol_factory, host=None, port=None, *,
if sock is None:
raise ValueError(
'host and port was not specified and no sock specified')
if not _is_stream_socket(sock) or not _is_ip_socket(sock):
if not _is_stream_socket(sock):
# We allow AF_INET, AF_INET6, AF_UNIX as long as they
# are SOCK_STREAM.
# We support passing AF_UNIX sockets even though we have
# a dedicated API for that: create_unix_connection.
# Disallowing AF_UNIX in this method, breaks backwards
# compatibility.
raise ValueError(
'A TCP Stream Socket was expected, got {!r}'.format(sock))
'A Stream Socket was expected, got {!r}'.format(sock))

transport, protocol = yield from self._create_connection_transport(
sock, protocol_factory, ssl, server_hostname)
Expand Down Expand Up @@ -1055,9 +1053,9 @@ def create_server(self, protocol_factory, host=None, port=None,
else:
if sock is None:
raise ValueError('Neither host/port nor sock were specified')
if not _is_stream_socket(sock) or not _is_ip_socket(sock):
if not _is_stream_socket(sock):
raise ValueError(
'A TCP Stream Socket was expected, got {!r}'.format(sock))
'A Stream Socket was expected, got {!r}'.format(sock))
sockets = [sock]

server = Server(self, sockets)
Expand Down
10 changes: 4 additions & 6 deletions Lib/test/test_asyncio/test_base_events.py
Original file line number Diff line number Diff line change
Expand Up @@ -1047,22 +1047,20 @@ def test_create_connection_host_port_sock(self):
MyProto, 'example.com', 80, sock=object())
self.assertRaises(ValueError, self.loop.run_until_complete, coro)

@unittest.skipUnless(hasattr(socket, 'AF_UNIX'), 'no Unix sockets')
def test_create_connection_wrong_sock(self):
sock = socket.socket(socket.AF_UNIX)
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
with sock:
coro = self.loop.create_connection(MyProto, sock=sock)
with self.assertRaisesRegex(ValueError,
'A TCP Stream Socket was expected'):
'A Stream Socket was expected'):
self.loop.run_until_complete(coro)

@unittest.skipUnless(hasattr(socket, 'AF_UNIX'), 'no Unix sockets')
def test_create_server_wrong_sock(self):
sock = socket.socket(socket.AF_UNIX)
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
with sock:
coro = self.loop.create_server(MyProto, sock=sock)
with self.assertRaisesRegex(ValueError,
'A TCP Stream Socket was expected'):
'A Stream Socket was expected'):
self.loop.run_until_complete(coro)

@unittest.skipUnless(hasattr(socket, 'SOCK_NONBLOCK'),
Expand Down

0 comments on commit 5cb0c09

Please sign in to comment.