def test_inet(self): endpoint = config.EndpointConfiguration(socket.AF_INET, ("localhost", 1234)) protocol = thrift_pool._make_protocol(endpoint) socket_transport = protocol.trans.getTransport() self.assertFalse(socket_transport._unix_socket)
def serve_thrift(handler, server_span_observer=None): # create baseplate root baseplate = Baseplate() if server_span_observer: class TestBaseplateObserver(BaseplateObserver): def on_server_span_created(self, context, server_span): server_span.register(server_span_observer) baseplate.register(TestBaseplateObserver()) # set up the server's processor logger = mock.Mock(spec=logging.Logger) edge_context_factory = make_edge_context_factory() processor = TestService.Processor(handler) processor = baseplateify_processor(processor, logger, baseplate, edge_context_factory) # bind a server socket on an available port server_bind_endpoint = config.Endpoint("127.0.0.1:0") listener = make_listener(server_bind_endpoint) server = make_server({ "max_concurrency": "100", }, listener, processor) # figure out what port the server ended up on server_address = listener.getsockname() server_endpoint = config.EndpointConfiguration( family=server_bind_endpoint.family, address=server_address) server.endpoint = server_endpoint # run the server until our caller is done with it server_greenlet = gevent.spawn(server.serve_forever) try: yield server finally: server_greenlet.kill()
def test_options(self, make_file): args = server.parse_args([ "filename", "--debug", "--app-name", "app", "--server-name", "server", "--bind", "1.2.3.4:81", ]) self.assertTrue(args.debug) self.assertEqual(args.app_name, "app") self.assertEqual(args.server_name, "server") self.assertEqual( args.bind, config.EndpointConfiguration(socket.AF_INET, ("1.2.3.4", 81)))
# coding=utf8 from __future__ import absolute_import from __future__ import division from __future__ import print_function from __future__ import unicode_literals import socket import unittest from baseplate import metrics, config from .. import mock EXAMPLE_ENDPOINT = config.EndpointConfiguration(socket.AF_INET, ("127.0.0.1", 1234)) class MetricJoinTests(unittest.TestCase): def test_single_node(self): joined = metrics._metric_join(b"single") self.assertEqual(joined, b"single") def test_two_nodes(self): joined = metrics._metric_join(b"first", b"second") self.assertEqual(joined, b"first.second") def test_subpaths(self): joined = metrics._metric_join(b"first.second", b"third.fourth") self.assertEqual(joined, b"first.second.third.fourth")
def test_unknown(self): endpoint = config.EndpointConfiguration( socket.AF_UNSPEC, None) with self.assertRaises(Exception): thrift_pool._make_transport(endpoint)
def test_unix(self): endpoint = config.EndpointConfiguration( socket.AF_UNIX, "/tmp/socket") socket_transport = thrift_pool._make_transport(endpoint) self.assertTrue(socket_transport._unix_socket)