async def test_retry(fake_zipkin, loop):
endpoint = az.create_endpoint('simple_service', ipv4='127.0.0.1', port=80)
tr = azt.Transport(fake_zipkin.url,
send_interval=0.01,
send_max_size=100,
send_attempt_count=3,
send_timeout=ClientTimeout(total=1))
fake_zipkin.next_errors.append('disconnect')
fake_zipkin.next_errors.append('timeout')
waiter = fake_zipkin.wait_data(1)
tracer = await az.create_custom(endpoint, tr)
with tracer.new_trace(sampled=True) as span:
span.name('root_span')
span.kind(az.CLIENT)
await waiter
await tracer.close()
data = fake_zipkin.get_received_data()
trace_id = span.context.trace_id
assert any(s['traceId'] == trace_id for trace in data for s in trace), data
async def test_batches(fake_zipkin, loop):
endpoint = az.create_endpoint('simple_service', ipv4='127.0.0.1', port=80)
tr = azt.Transport(fake_zipkin.url,
send_interval=0.01,
send_max_size=2,
send_timeout=ClientTimeout(total=1))
tracer = await az.create_custom(endpoint, tr)
with tracer.new_trace(sampled=True) as span:
span.name('root_span')
span.kind(az.CLIENT)
with span.new_child('child_1', az.CLIENT):
pass
with span.new_child('child_2', az.CLIENT):
pass
# close forced sending data to server regardless of send interval
await tracer.close()
data = fake_zipkin.get_received_data()
trace_id = span.context.trace_id
assert len(data[0]) == 2
assert len(data[1]) == 1
assert data[0][0]['name'] == 'child_1'
assert data[0][1]['name'] == 'child_2'
assert data[1][0]['name'] == 'root_span'
assert any(s['traceId'] == trace_id for trace in data for s in trace), data
async def test_send_full_batch(fake_zipkin, loop):
endpoint = az.create_endpoint('simple_service', ipv4='127.0.0.1', port=80)
tr = azt.Transport(fake_zipkin.url,
send_interval=60,
send_max_size=2,
send_timeout=ClientTimeout(total=1))
tracer = await az.create_custom(endpoint, tr)
waiter = fake_zipkin.wait_data(1)
with tracer.new_trace(sampled=True) as span:
span.name('root_span')
span.kind(az.CLIENT)
await asyncio.sleep(1, loop=loop)
data = fake_zipkin.get_received_data()
assert len(data) == 0
with tracer.new_trace(sampled=True) as span:
span.name('root_span')
span.kind(az.CLIENT)
# batch is full here
await waiter
data = fake_zipkin.get_received_data()
assert len(data) == 1
# close forced sending data to server regardless of send interval
await tracer.close()
async def test_batches(fake_zipkin: Any,
loop: asyncio.AbstractEventLoop) -> None:
endpoint = az.create_endpoint("simple_service", ipv4="127.0.0.1", port=80)
tr = azt.Transport(
fake_zipkin.url,
send_interval=0.01,
send_max_size=2,
send_timeout=ClientTimeout(total=1),
)
tracer = await az.create_custom(endpoint, tr)
with tracer.new_trace(sampled=True) as span:
span.name("root_span")
span.kind(az.CLIENT)
with span.new_child("child_1", az.CLIENT):
pass
with span.new_child("child_2", az.CLIENT):
pass
# close forced sending data to server regardless of send interval
await tracer.close()
data = fake_zipkin.get_received_data()
trace_id = span.context.trace_id
assert len(data[0]) == 2
assert len(data[1]) == 1
assert data[0][0]["name"] == "child_1"
assert data[0][1]["name"] == "child_2"
assert data[1][0]["name"] == "root_span"
assert any(s["traceId"] == trace_id for trace in data for s in trace), data
async def test_lost_spans(fake_zipkin: Any,
loop: asyncio.AbstractEventLoop) -> None:
endpoint = az.create_endpoint("simple_service", ipv4="127.0.0.1", port=80)
tr = azt.Transport(
fake_zipkin.url,
send_interval=0.01,
send_max_size=100,
send_attempt_count=2,
send_timeout=ClientTimeout(total=1),
)
fake_zipkin.next_errors.append("disconnect")
fake_zipkin.next_errors.append("disconnect")
tracer = await az.create_custom(endpoint, tr)
with tracer.new_trace(sampled=True) as span:
span.name("root_span")
span.kind(az.CLIENT)
await asyncio.sleep(1)
await tracer.close()
data = fake_zipkin.get_received_data()
assert len(data) == 0
async def test_retry(fake_zipkin: Any,
loop: asyncio.AbstractEventLoop) -> None:
endpoint = az.create_endpoint("simple_service", ipv4="127.0.0.1", port=80)
tr = azt.Transport(
fake_zipkin.url,
send_interval=0.01,
send_max_size=100,
send_attempt_count=3,
send_timeout=ClientTimeout(total=1),
)
fake_zipkin.next_errors.append("disconnect")
fake_zipkin.next_errors.append("timeout")
waiter = fake_zipkin.wait_data(1)
tracer = await az.create_custom(endpoint, tr)
with tracer.new_trace(sampled=True) as span:
span.name("root_span")
span.kind(az.CLIENT)
await waiter
await tracer.close()
data = fake_zipkin.get_received_data()
trace_id = span.context.trace_id
assert any(s["traceId"] == trace_id for trace in data for s in trace), data
async def start(self, logger: 'ipapp.logger.Logger') -> None:
endpoint = az.create_endpoint(self.cfg.name)
sampler = az.Sampler(sample_rate=self.cfg.sample_rate)
transport = azt.Transport(
self.cfg.addr,
send_interval=self.cfg.send_interval,
)
self.tracer = az.Tracer(transport, sampler, endpoint)
async def test_lost_spans(fake_zipkin, loop):
endpoint = az.create_endpoint('simple_service', ipv4='127.0.0.1', port=80)
tr = azt.Transport(fake_zipkin.url,
send_interval=0.01,
send_max_size=100,
send_attempt_count=2,
send_timeout=ClientTimeout(total=1))
fake_zipkin.next_errors.append('disconnect')
fake_zipkin.next_errors.append('disconnect')
tracer = await az.create_custom(endpoint, tr)
with tracer.new_trace(sampled=True) as span:
span.name('root_span')
span.kind(az.CLIENT)
await asyncio.sleep(1, loop=loop)
await tracer.close()
data = fake_zipkin.get_received_data()
assert len(data) == 0