52b09becac
Fixes installation issue when older dnspython is present in system packages https://github.com/eventlet/eventlet/pull/341
65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
import os
|
|
import shutil
|
|
import sys
|
|
import tempfile
|
|
import eventlet
|
|
from eventlet.green import socket
|
|
from eventlet.support import greendns
|
|
import tests
|
|
|
|
|
|
def test_create_connection_error():
|
|
try:
|
|
socket.create_connection(('192.0.2.1', 80), timeout=0.1)
|
|
except (IOError, OSError):
|
|
pass
|
|
|
|
|
|
def test_recv_type():
|
|
# https://github.com/eventlet/eventlet/issues/245
|
|
# socket recv returning multiple data types
|
|
# For this test to work, client and server have to be in separate
|
|
# processes or OS threads. Just running two greenthreads gives
|
|
# false test pass.
|
|
threading = eventlet.patcher.original('threading')
|
|
addr = []
|
|
|
|
def server():
|
|
sock = eventlet.listen(('127.0.0.1', 0))
|
|
addr[:] = sock.getsockname()
|
|
eventlet.sleep(0.2)
|
|
|
|
server_thread = threading.Thread(target=server)
|
|
server_thread.start()
|
|
eventlet.sleep(0.1)
|
|
sock = eventlet.connect(tuple(addr))
|
|
s = sock.recv(1)
|
|
assert isinstance(s, bytes)
|
|
|
|
|
|
def test_dns_methods_are_green():
|
|
assert socket.gethostbyname is greendns.gethostbyname
|
|
assert socket.gethostbyname_ex is greendns.gethostbyname_ex
|
|
assert socket.getaddrinfo is greendns.getaddrinfo
|
|
assert socket.getnameinfo is greendns.getnameinfo
|
|
|
|
# https://github.com/eventlet/eventlet/pull/341
|
|
# mock older dnspython in system packages
|
|
mock_sys_pkg_dir = tempfile.mkdtemp('eventlet_test_dns_methods_are_green')
|
|
try:
|
|
with open(mock_sys_pkg_dir + '/dns.py', 'wb') as f:
|
|
f.write(b'raise Exception("Your IP address string is so illegal ' +
|
|
b'it prevents installing packages.")\n')
|
|
tests.run_isolated(
|
|
'socket_resolve_green.py',
|
|
env={'PYTHONPATH': os.pathsep.join(sys.path + [mock_sys_pkg_dir])},
|
|
)
|
|
finally:
|
|
shutil.rmtree(mock_sys_pkg_dir)
|
|
|
|
|
|
def test_socket_api_family():
|
|
# It was named family_or_realsock
|
|
# https://github.com/eventlet/eventlet/issues/319
|
|
socket.socket(family=socket.AF_INET)
|