def test_tcollector_submit(trace_server):
tchannel = TChannel(name='test', known_peers=[trace_server.hostport])
trace = Trace(endpoint=Endpoint("1.0.0.1", 1111, "tcollector"))
anns = [client_send()]
results = yield TChannelZipkinTracer(tchannel).record([(trace, anns)])
assert results[0].ok
def __init__(
self,
description=None,
id=None,
tracing=None,
):
super(TChannelError, self).__init__(description)
self.tracing = tracing or Trace()
self.id = id
self.description = description
def test_build_raw_response_message():
message_factory = MessageFactory()
resp = Response(
flags=FlagsType.none,
code=StatusCode.ok,
headers={},
tracing=Trace(),
id=1111,
)
resp.state = StreamState.init
message = message_factory.build_raw_response_message(resp, None, True)
assert message.code == resp.code
assert message.flags == resp.flags
assert message.id == resp.id
assert message.headers == resp.headers
def test_span_host_field():
tchannel = TChannel('test')
tchannel.listen()
tracer = TChannelZipkinTracer(tchannel)
host_span = tracer.parse_host_port()
thrift_obj = thrift_formatter(
trace=Trace(name='test', endpoint=Endpoint('0.0.0.1', 90, 'test1')),
annotations=[annotation.client_send()],
isbased64=False,
span_host=host_span,
)
assert thrift_obj.spanHost.ipv4 == ipv4_to_int(host_span.ipv4)
assert thrift_obj.spanHost.port == host_span.port
assert thrift_obj.spanHost.serviceName == host_span.service_name