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)
