def test_in_memory_reporter():
reporter = jaeger_client.reporter.InMemoryReporter()
reporter.report_span({})
f = reporter.close()
f.result()
spans = reporter.get_spans()
assert [{}] == spans
def test_composite_reporter():
reporter = jaeger_client.reporter.CompositeReporter(
jaeger_client.reporter.NullReporter(),
jaeger_client.reporter.LoggingReporter())
with mock.patch('jaeger_client.reporter.NullReporter.set_process') \
as null_mock:
with mock.patch('jaeger_client.reporter.LoggingReporter.set_process') \
as log_mock:
reporter.set_process('x', {}, 123)
null_mock.assert_called_with('x', {}, 123)
log_mock.assert_called_with('x', {}, 123)
with mock.patch('jaeger_client.reporter.NullReporter.report_span') \
as null_mock:
with mock.patch('jaeger_client.reporter.LoggingReporter.report_span') \
as log_mock:
reporter.report_span({})
null_mock.assert_called_with({})
log_mock.assert_called_with({})
with mock.patch('jaeger_client.reporter.NullReporter.close') \
as null_mock:
with mock.patch('jaeger_client.reporter.LoggingReporter.close') \
as log_mock:
f1 = Future()
f2 = Future()
null_mock.return_value = f1
log_mock.return_value = f2
f = reporter.close()
null_mock.assert_called_once()
log_mock.assert_called_once()
assert not f.done()
f1.set_result(True)
f2.set_result(True)
assert f.done()
def test_close_drains_queue(self):
reporter, sender = self._new_reporter(batch_size=1, flush=0.050)
reporter.report_span(self._new_span('0'))
yield self._wait_for(lambda: len(sender.futures) > 0)
assert 1 == len(sender.futures)
# now that the consumer is blocked on the first future.
# let's reset Send to actually respond right away
# and flood the queue with messages
count = [0]
def send(_):
count[0] += 1
return future_result(True)
reporter._send = send
reporter.batch_size = 3
for i in range(10):
reporter.report_span(self._new_span('%s' % i))
assert reporter.queue.qsize() == 10, 'queued 10 spans'
# now unblock consumer
sender.futures[0].set_result(1)
yield self._wait_for(lambda: count[0] > 2)
assert count[0] == 3, '9 out of 10 spans submitted in 3 batches'
assert reporter.queue._unfinished_tasks == 1, 'one span still pending'
yield reporter.close()
assert reporter.queue.qsize() == 0, 'all spans drained'
assert count[0] == 4, 'last span submitted in one extrac batch'
def test_close_drains_queue(self):
reporter, sender = self._new_reporter(batch_size=1, flush=0.050)
reporter.report_span(self._new_span('0'))
yield self._wait_for(lambda: len(sender.futures) > 0)
assert 1 == len(sender.futures)
# now that the consumer is blocked on the first future.
# let's reset Send to actually respond right away
# and flood the queue with messages
count = [0]
def send(_):
count[0] += 1
return future_result(True)
reporter._send = send
reporter.batch_size = 3
for i in range(10):
reporter.report_span(self._new_span('%s' % i))
assert reporter.queue.qsize() == 10, 'queued 10 spans'
# now unblock consumer
sender.futures[0].set_result(1)
yield self._wait_for(lambda: count[0] > 2)
assert count[0] == 3, '9 out of 10 spans submitted in 3 batches'
assert reporter.queue._unfinished_tasks == 1, 'one span still pending'
yield reporter.close()
assert reporter.queue.qsize() == 0, 'all spans drained'
assert count[0] == 4, 'last span submitted in one extrac batch'
def test_in_memory_reporter():
reporter = jaeger_client.reporter.InMemoryReporter()
reporter.report_span({})
f = reporter.close()
f.result()
spans = reporter.get_spans()
assert [{}] == spans
def test_composite_reporter():
reporter = jaeger_client.reporter.CompositeReporter(
jaeger_client.reporter.NullReporter(),
jaeger_client.reporter.LoggingReporter())
with mock.patch('jaeger_client.reporter.NullReporter.set_process') \
as null_mock:
with mock.patch('jaeger_client.reporter.LoggingReporter.set_process') \
as log_mock:
reporter.set_process('x', {}, 123)
null_mock.assert_called_with('x', {}, 123)
log_mock.assert_called_with('x', {}, 123)
with mock.patch('jaeger_client.reporter.NullReporter.report_span') \
as null_mock:
with mock.patch('jaeger_client.reporter.LoggingReporter.report_span') \
as log_mock:
reporter.report_span({})
null_mock.assert_called_with({})
log_mock.assert_called_with({})
with mock.patch('jaeger_client.reporter.NullReporter.close') \
as null_mock:
with mock.patch('jaeger_client.reporter.LoggingReporter.close') \
as log_mock:
f1 = Future()
f2 = Future()
null_mock.return_value = f1
log_mock.return_value = f2
f = reporter.close()
null_mock.assert_called_once()
log_mock.assert_called_once()
assert not f.done()
f1.set_result(True)
f2.set_result(True)
assert f.done()
def test_submit_failure(self):
reporter, sender = self._new_reporter(batch_size=1)
reporter.error_reporter = ErrorReporter(metrics=reporter.metrics)
# simulate exception in send
reporter._send = mock.MagicMock(side_effect=ValueError())
reporter.report_span(self._new_span('1'))
yield tornado.gen.sleep(0.001)
assert 1 == reporter.metrics.counters[Metrics.REPORTER_FAILURE]
# silly test, for code coverage only
yield reporter._submit([])
def test_submit_batch_size_1(self):
reporter, sender = self._new_reporter(batch_size=1)
reporter.report_span(self._new_span('1'))
yield tornado.gen.sleep(0.001)
assert 1 == len(sender.futures)
sender.futures[0].set_result(1)
yield reporter.close()
assert 1 == len(sender.futures)
# send after close
assert Metrics.REPORTER_DROPPED not in reporter.metrics.counters
reporter.report_span(self._new_span('1'))
assert 1 == reporter.metrics.counters[Metrics.REPORTER_DROPPED]
def test_submit_batch_size_1(self):
reporter, sender = self._new_reporter(batch_size=1)
reporter.report_span(self._new_span('1'))
yield self._wait_for(lambda: len(sender.futures) > 0)
assert 1 == len(sender.futures)
sender.futures[0].set_result(1)
yield reporter.close()
assert 1 == len(sender.futures)
# send after close
assert Metrics.REPORTER_DROPPED not in reporter.metrics.counters
reporter.report_span(self._new_span('1'))
assert 1 == reporter.metrics.counters[Metrics.REPORTER_DROPPED]
def test_submit_batch_size_1(self):
reporter, sender = self._new_reporter(batch_size=1)
reporter.report_span(self._new_span('1'))
yield self._wait_for(lambda: len(sender.futures) > 0)
assert 1 == len(sender.futures)
sender.futures[0].set_result(1)
yield reporter.close()
assert 1 == len(sender.futures)
# send after close
span_dropped_key = 'jaeger:reporter_spans.result_dropped'
assert span_dropped_key not in reporter.metrics_factory.counters
reporter.report_span(self._new_span('1'))
assert 1 == reporter.metrics_factory.counters[span_dropped_key]
def test_submit_batch_size_1(self):
reporter, sender = self._new_reporter(batch_size=1)
reporter.report_span(self._new_span('1'))
yield self._wait_for(lambda: len(sender.futures) > 0)
assert 1 == len(sender.futures)
sender.futures[0].set_result(1)
yield reporter.close()
assert 1 == len(sender.futures)
# send after close
span_dropped_key = 'jaeger:reporter_spans.result_dropped'
assert span_dropped_key not in reporter.metrics_factory.counters
reporter.report_span(self._new_span('1'))
assert 1 == reporter.metrics_factory.counters[span_dropped_key]
def test_submit_failure(self):
reporter, sender = self._new_reporter(batch_size=1)
reporter.error_reporter = ErrorReporter(metrics=reporter.metrics,
logger=logging.getLogger())
assert Metrics.REPORTER_FAILURE not in reporter.metrics.counters
# simulate exception in send
reporter._send = mock.MagicMock(side_effect=ValueError())
reporter.report_span(self._new_span('1'))
yield self._wait_for(
lambda: Metrics.REPORTER_FAILURE in reporter.metrics.counters)
assert 1 == reporter.metrics.counters.get(Metrics.REPORTER_FAILURE)
# silly test, for code coverage only
yield reporter._submit([])
def test_submit_failure(self):
reporter, sender = self._new_reporter(batch_size=1)
reporter.error_reporter = ErrorReporter(metrics=Metrics(),
logger=logging.getLogger())
reporter_failure_key = 'jaeger:reporter_spans.result_err'
assert reporter_failure_key not in reporter.metrics_factory.counters
# simulate exception in send
reporter._send = mock.MagicMock(side_effect=ValueError())
reporter.report_span(self._new_span('1'))
yield self._wait_for(
lambda: reporter_failure_key in reporter.metrics_factory.counters)
assert 1 == reporter.metrics_factory.counters.get(reporter_failure_key)
# silly test, for code coverage only
yield reporter._submit([])
def test_submit_failure(self):
reporter, sender = self._new_reporter(batch_size=1)
reporter.error_reporter = ErrorReporter(
metrics=Metrics(), logger=logging.getLogger())
reporter_failure_key = 'jaeger:reporter_spans.result_err'
assert reporter_failure_key not in reporter.metrics_factory.counters
# simulate exception in send
reporter._send = mock.MagicMock(side_effect=ValueError())
reporter.report_span(self._new_span('1'))
yield self._wait_for(
lambda: reporter_failure_key in reporter.metrics_factory.counters)
assert 1 == reporter.metrics_factory.counters.get(reporter_failure_key)
# silly test, for code coverage only
yield reporter._submit([])
def test_submit_queue_full_batch_size_1(self):
reporter, sender = self._new_reporter(batch_size=1, queue_cap=1)
reporter.report_span(self._new_span('1'))
yield tornado.gen.sleep(0.001)
assert 1 == len(sender.futures)
# the consumer is blocked on a future, so won't drain the queue
reporter.report_span(self._new_span('2'))
assert Metrics.REPORTER_DROPPED not in reporter.metrics.counters
reporter.report_span(self._new_span('3'))
assert 1 == reporter.metrics.counters[Metrics.REPORTER_DROPPED]
# let it drain the queue
sender.futures[0].set_result(1)
yield tornado.gen.sleep(0.001)
assert 2 == len(sender.futures)
sender.futures[1].set_result(1)
yield reporter.close()
def test_submit_queue_full_batch_size_1(self):
reporter, sender = self._new_reporter(batch_size=1, queue_cap=1)
reporter.report_span(self._new_span('1'))
yield self._wait_for(lambda: len(sender.futures) > 0)
assert 1 == len(sender.futures)
# the consumer is blocked on a future, so won't drain the queue
reporter.report_span(self._new_span('2'))
assert Metrics.REPORTER_DROPPED not in reporter.metrics.counters
reporter.report_span(self._new_span('3'))
yield self._wait_for(
lambda: Metrics.REPORTER_DROPPED in reporter.metrics.counters)
assert 1 == reporter.metrics.counters.get(Metrics.REPORTER_DROPPED)
# let it drain the queue
sender.futures[0].set_result(1)
yield self._wait_for(lambda: len(sender.futures) > 1)
assert 2 == len(sender.futures)
sender.futures[1].set_result(1)
yield reporter.close()
def test_submit_queue_full_batch_size_1(self):
reporter, sender = self._new_reporter(batch_size=1, queue_cap=1)
reporter.report_span(self._new_span('1'))
yield self._wait_for(lambda: len(sender.futures) > 0)
assert 1 == len(sender.futures)
# the consumer is blocked on a future, so won't drain the queue
reporter.report_span(self._new_span('2'))
span_dropped_key = 'jaeger:reporter_spans.result_dropped'
assert span_dropped_key not in reporter.metrics_factory.counters
reporter.report_span(self._new_span('3'))
yield self._wait_for(
lambda: span_dropped_key in reporter.metrics_factory.counters)
assert 1 == reporter.metrics_factory.counters.get(span_dropped_key)
# let it drain the queue
sender.futures[0].set_result(1)
yield self._wait_for(lambda: len(sender.futures) > 1)
assert 2 == len(sender.futures)
sender.futures[1].set_result(1)
yield reporter.close()
def test_submit_queue_full_batch_size_1(self):
reporter, sender = self._new_reporter(batch_size=1, queue_cap=1)
reporter.report_span(self._new_span('1'))
yield self._wait_for(lambda: len(sender.futures) > 0)
assert 1 == len(sender.futures)
# the consumer is blocked on a future, so won't drain the queue
reporter.report_span(self._new_span('2'))
span_dropped_key = 'jaeger:reporter_spans.result_dropped'
assert span_dropped_key not in reporter.metrics_factory.counters
reporter.report_span(self._new_span('3'))
yield self._wait_for(
lambda: span_dropped_key in reporter.metrics_factory.counters
)
assert 1 == reporter.metrics_factory.counters.get(span_dropped_key)
# let it drain the queue
sender.futures[0].set_result(1)
yield self._wait_for(lambda: len(sender.futures) > 1)
assert 2 == len(sender.futures)
sender.futures[1].set_result(1)
yield reporter.close()
def test_submit_batch_size_2(self):
reporter, sender = self._new_reporter(batch_size=2, flush=0.005)
reporter.report_span(self._new_span('1'))
yield tornado.gen.sleep(0.001)
assert 0 == len(sender.futures)
reporter.report_span(self._new_span('2'))
yield tornado.gen.sleep(0.001)
assert 1 == len(sender.futures)
sender.futures[0].set_result(1)
# 3rd span will not be submitted right away, but after `flush` interval
reporter.report_span(self._new_span('3'))
yield tornado.gen.sleep(0.001)
assert 1 == len(sender.futures)
yield tornado.gen.sleep(0.001)
assert 1 == len(sender.futures)
yield tornado.gen.sleep(0.005)
assert 2 == len(sender.futures)
sender.futures[1].set_result(1)
yield reporter.close()
def test_submit_batch_size_2(self):
reporter, sender = self._new_reporter(batch_size=2, flush=0.01)
reporter.report_span(self._new_span('1'))
yield tornado.gen.sleep(0.001)
assert 0 == len(sender.futures)
reporter.report_span(self._new_span('2'))
yield self._wait_for(lambda: len(sender.futures) > 0)
assert 1 == len(sender.futures)
assert 2 == len(sender.requests[0].spans)
sender.futures[0].set_result(1)
# 3rd span will not be submitted right away, but after `flush` interval
reporter.report_span(self._new_span('3'))
yield tornado.gen.sleep(0.001)
assert 1 == len(sender.futures)
yield tornado.gen.sleep(0.001)
assert 1 == len(sender.futures)
yield tornado.gen.sleep(0.01)
assert 2 == len(sender.futures)
sender.futures[1].set_result(1)
yield reporter.close()
def test_logging_reporter():
log_mock = mock.MagicMock()
reporter = jaeger_client.reporter.LoggingReporter(logger=log_mock)
reporter.report_span({})
log_mock.info.assert_called_with('Reporting span %s', {})
reporter.close().result()
def test_logging_reporter():
log_mock = mock.MagicMock()
reporter = jaeger_client.reporter.LoggingReporter(logger=log_mock)
reporter.report_span({})
log_mock.info.assert_called_with('Reporting span %s', {})
reporter.close().result()
def test_null_reporter():
reporter = jaeger_client.reporter.NullReporter()
reporter.report_span({})
f = reporter.close()
f.result()
def test_null_reporter():
reporter = jaeger_client.reporter.NullReporter()
reporter.report_span({})
f = reporter.close()
f.result()
