Python SocketServer.wait_until_queuing_requests示例

示例#1

文件： test_collector.py 项目： farzadghanei/navdoon

class SocketServerTestCaseMixIn(object):
    def setup_socket_server(self, socket_type):
        self.host = '127.0.0.1'
        self.port = find_open_port(self.host, socket_type)
        self.server = SocketServer()
        self.server.socket_type = socket_type
        self.server.host = self.host
        self.server.port = self.port
        self.server_thread = threading.Thread(target=self.server.start)
        self.server_thread.daemon = True

    def start_server_send_data_and_consume_queue(self, data_set, socket_type):
        self.server_thread.start()
        self.server.wait_until_queuing_requests()
        client_sock = socket.socket(socket.AF_INET, socket_type)
        client_sock.connect((self.host, self.port))
        socket_sendall_close(client_sock, data_set)
        in_queue = consume_queue(self.server.queue, len(data_set))
        return in_queue

示例#2

class SocketServerTestCaseMixIn(object):
    def setup_socket_server(self, socket_type):
        self.host = '127.0.0.1'
        self.port = find_open_port(self.host, socket_type)
        self.server = SocketServer()
        self.server.socket_type = socket_type
        self.server.host = self.host
        self.server.port = self.port
        self.server_thread = threading.Thread(target=self.server.start)
        self.server_thread.daemon = True

    def start_server_send_data_and_consume_queue(self, data_set, socket_type):
        self.server_thread.start()
        self.server.wait_until_queuing_requests()
        client_sock = socket.socket(socket.AF_INET, socket_type)
        client_sock.connect((self.host, self.port))
        socket_sendall_close(client_sock, data_set)
        in_queue = consume_queue(self.server.queue, len(data_set))
        return in_queue