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