def test_not_only_sockets_in_list(self): from waitress.adjustments import as_socket_list sockets = [ socket.socket(socket.AF_INET, socket.SOCK_STREAM), socket.socket(socket.AF_INET6, socket.SOCK_STREAM), {'something': 'else'}] new_sockets = as_socket_list(sockets) self.assertEqual(new_sockets, [sockets[0], sockets[1]]) for sock in [sock for sock in sockets if isinstance(sock, socket.socket)]: sock.close()
def test_only_sockets_in_list(self): from waitress.adjustments import as_socket_list sockets = [ socket.socket(socket.AF_INET, socket.SOCK_STREAM), socket.socket(socket.AF_INET6, socket.SOCK_STREAM)] if hasattr(socket, 'AF_UNIX'): sockets.append(socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)) new_sockets = as_socket_list(sockets) self.assertEqual(sockets, new_sockets) for sock in sockets: sock.close()