def test_io_get_socket_address(self):
connection = FakeConnection()
connection.parameters['hostname'] = '127.0.0.1'
connection.parameters['port'] = 5672
io = IO(connection.parameters)
addresses = io._get_socket_addresses()
sock_address_tuple = addresses[0]
self.assertEqual(sock_address_tuple[4], ('127.0.0.1', 5672))
def test_io_get_socket_address(self):
connection = FakeConnection()
connection.parameters['hostname'] = '127.0.0.1'
connection.parameters['port'] = 5672
io = IO(connection.parameters)
addresses = io._get_socket_addresses()
sock_address_tuple = addresses[0]
self.assertEqual(sock_address_tuple[4],
('127.0.0.1', 5672))
def test_io_create_socket(self):
connection = FakeConnection()
io = IO(connection.parameters)
addresses = io._get_socket_addresses()
sock_address_tuple = addresses[0]
sock = io._create_socket(socket_family=sock_address_tuple[0])
if hasattr(socket, 'socket'):
self.assertIsInstance(sock, socket.socket)
elif hasattr(socket, '_socketobject'):
self.assertIsInstance(sock, socket._socketobject)
def test_io_create_ssl_socket(self):
connection = FakeConnection()
connection.parameters['ssl'] = True
io = IO(connection.parameters)
self.assertTrue(io.use_ssl)
addresses = io._get_socket_addresses()
sock_address_tuple = addresses[0]
sock = io._create_socket(socket_family=sock_address_tuple[0])
if hasattr(socket, 'socket'):
self.assertIsInstance(sock, socket.socket)
if hasattr(ssl, 'SSLSocket'):
self.assertIsInstance(sock, ssl.SSLSocket)
def test_io_create_ssl_socket(self):
connection = FakeConnection()
connection.parameters['ssl'] = True
io = IO(connection.parameters)
self.assertTrue(io.use_ssl)
addresses = io._get_socket_addresses()
sock_address_tuple = addresses[0]
sock = io._create_socket(socket_family=sock_address_tuple[0])
if hasattr(socket, 'socket'):
self.assertIsInstance(sock, socket.socket)
if hasattr(ssl, 'SSLSocket'):
self.assertIsInstance(sock, ssl.SSLSocket)
self.assertTrue(connection.parameters['ssl_options']['ssl_version'])
def test_io_has_ipv6(self):
restore_func = socket.getaddrinfo
def mock_getaddrinfo(hostname, port, family):
return [hostname, port, family]
try:
amqpstorm.io.socket.getaddrinfo = mock_getaddrinfo
connection = FakeConnection()
connection.parameters['hostname'] = 'localhost'
connection.parameters['port'] = 1234
parameters = connection.parameters
io = IO(parameters)
result = io._get_socket_addresses()
self.assertEqual(result[2], socket.AF_UNSPEC)
finally:
amqpstorm.io.socket.getaddrinfo = restore_func
def test_io_has_ipv6(self):
restore_func = socket.getaddrinfo
def mock_getaddrinfo(hostname, port, family, socktype):
return [hostname, port, family, socktype]
try:
amqpstorm.io.socket.getaddrinfo = mock_getaddrinfo
connection = FakeConnection()
connection.parameters['hostname'] = 'localhost'
connection.parameters['port'] = 1234
parameters = connection.parameters
io = IO(parameters)
result = io._get_socket_addresses()
self.assertEqual(result[2], socket.AF_UNSPEC)
self.assertEqual(result[3], socket.SOCK_STREAM)
finally:
amqpstorm.io.socket.getaddrinfo = restore_func
def test_io_has_ipv6_is_false(self):
restore_func = socket.getaddrinfo
restore_has_ipv6 = amqpstorm.io.socket.has_ipv6
def mock_getaddrinfo(hostname, port, family, socktype):
return [hostname, port, family, socktype]
try:
amqpstorm.io.socket.getaddrinfo = mock_getaddrinfo
amqpstorm.io.socket.has_ipv6 = False
connection = FakeConnection()
connection.parameters['hostname'] = 'localhost'
connection.parameters['port'] = 1234
parameters = connection.parameters
io = IO(parameters)
result = io._get_socket_addresses()
self.assertEqual(result[2], socket.AF_INET)
self.assertEqual(result[3], socket.SOCK_STREAM)
finally:
amqpstorm.io.socket.getaddrinfo = restore_func
amqpstorm.io.socket.has_ipv6 = restore_has_ipv6
