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 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')
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 _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)
