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