def test_context(self):
mock_observer = mock.Mock(spec=SpanObserver)
span = Span(1, 2, 3, None, 0, "name")
span.register(mock_observer)
with span:
self.assertEqual(mock_observer.on_start.call_count, 1)
self.assertEqual(mock_observer.on_finish.call_count, 1)
def test_context(self):
mock_observer = mock.Mock(spec=SpanObserver)
span = Span(1, 2, 3, "name")
span.register(mock_observer)
with span:
self.assertEqual(mock_observer.on_start.call_count, 1)
self.assertEqual(mock_observer.on_finish.call_count, 1)
def test_events(self):
mock_observer = mock.Mock(spec=SpanObserver)
span = Span(1, 2, 3, "name")
span.register(mock_observer)
span.start()
self.assertEqual(mock_observer.on_start.call_count, 1)
span.annotate("key", "value")
mock_observer.on_annotate("key", "value")
span.stop()
mock_observer.on_stop(error=None)
def test_events(self):
mock_observer = mock.Mock(spec=SpanObserver)
span = Span(1, 2, 3, "name")
span.register(mock_observer)
span.start()
self.assertEqual(mock_observer.on_start.call_count, 1)
span.annotate("key", "value")
mock_observer.on_annotate("key", "value")
span.stop()
mock_observer.on_stop(error=None)
def test_context_with_exception(self):
mock_observer = mock.Mock(spec=SpanObserver)
span = Span(1, 2, 3, "name")
span.register(mock_observer)
class TestException(Exception):
pass
exc = TestException()
with self.assertRaises(TestException):
with span:
raise exc
self.assertEqual(mock_observer.on_stop.call_count, 1)
self.assertEqual(mock_observer.on_stop.call_args, mock.call(error=exc))
def test_context_with_exception(self):
mock_observer = mock.Mock(spec=SpanObserver)
span = Span(1, 2, 3, "name")
span.register(mock_observer)
class TestException(Exception):
pass
exc = TestException()
with self.assertRaises(TestException):
with span:
raise exc
self.assertEqual(mock_observer.on_stop.call_count, 1)
self.assertEqual(mock_observer.on_stop.call_args, mock.call(error=exc))
def test_context_with_exception(self):
mock_observer = mock.Mock(spec=SpanObserver)
span = Span(1, 2, 3, "name")
span.register(mock_observer)
class TestException(Exception):
pass
exc = TestException()
with self.assertRaises(TestException):
with span:
raise exc
self.assertEqual(mock_observer.on_finish.call_count, 1)
_, captured_exc, _ = mock_observer.on_finish.call_args[0][0]
self.assertEqual(captured_exc, exc)
def test_context_with_exception(self):
mock_observer = mock.Mock(spec=SpanObserver)
span = Span(1, 2, 3, None, 0, "name")
span.register(mock_observer)
class TestException(Exception):
pass
exc = TestException()
with self.assertRaises(TestException):
with span:
raise exc
self.assertEqual(mock_observer.on_finish.call_count, 1)
_, captured_exc, _ = mock_observer.on_finish.call_args[0][0]
self.assertEqual(captured_exc, exc)
def test_events(self):
mock_observer = mock.Mock(spec=SpanObserver)
span = Span(1, 2, 3, "name")
span.register(mock_observer)
span.start()
self.assertEqual(mock_observer.on_start.call_count, 1)
span.set_tag("key", "value")
mock_observer.on_set_tag("key", "value")
span.log("name", "payload")
mock_observer.on_log("name", "payload")
span.finish()
mock_observer.on_finish(exc_info=None)
def test_events(self):
mock_observer = mock.Mock(spec=SpanObserver)
span = Span(1, 2, 3, None, 0, "name")
span.register(mock_observer)
span.start()
self.assertEqual(mock_observer.on_start.call_count, 1)
span.set_tag("key", "value")
mock_observer.on_set_tag("key", "value")
span.log("name", "payload")
mock_observer.on_log("name", "payload")
span.finish()
mock_observer.on_finish(exc_info=None)
