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
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
def test_constructor_args(self): conf = dict(user='******', port=9876, group='thatgroup', host='example.org') server = SocketServer(**conf) self.assertEqual(server.host, 'example.org') self.assertEqual(server.port, 9876) self.assertEqual(server.user, 'thisuser') self.assertEqual(server.group, 'thatgroup')
def _create_socket_servers(self, addresses, socket_type): # type: (str, int) -> List[SocketServer] collectors = [] socket_addresses = self.get_addresses_with_unique_ports(addresses) for (host, port) in socket_addresses: socket_server = self._configure_socket_server(SocketServer(), host=host, port=port) socket_server.socket_type = socket_type collectors.append(socket_server) return collectors
def test_constructor_args(self): conf = dict(socket_type=socket.SOCK_STREAM) server = SocketServer(**conf) self.assertEqual(server.socket_type, socket.SOCK_STREAM)