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