def test_register_server_span_observer(self): baseplate_observer = TraceBaseplateObserver("test-service") context_mock = mock.Mock() span = ServerSpan("test-id", "test-parent-id", "test-span-id", True, 0, "test") baseplate_observer.on_server_span_created(context_mock, span) self.assertEqual(len(span.observers), 1) self.assertEqual(type(span.observers[0]), TraceServerSpanObserver)
def test_no_tracing_without_sampling(self): client = make_client("test-service", sample_rate=0) baseplate_observer = TraceBaseplateObserver(client) context_mock = mock.Mock() span = ServerSpan('test-id', 'test-parent-id', 'test-span-id', False, 0, 'test', self.mock_context) baseplate_observer.on_server_span_created(context_mock, span) self.assertEqual(len(span.observers), 0)
def test_register_server_span_observer(self): baseplate_observer = TraceBaseplateObserver('test-service') context_mock = mock.Mock() span = ServerSpan('test-id', 'test-parent-id', 'test-span-id', True, 0, 'test') baseplate_observer.on_server_span_created(context_mock, span) self.assertEqual(len(span.observers), 1) self.assertEqual(type(span.observers[0]), TraceServerSpanObserver)
def test_register_server_span_observer(self): client = make_client("test-service") baseplate_observer = TraceBaseplateObserver(client) context_mock = mock.Mock() span = ServerSpan("test-id", "test-parent-id", "test-span-id", True, 0, "test", self.mock_context) baseplate_observer.on_server_span_created(context_mock, span) self.assertEqual(len(span.observers), 1) self.assertEqual(type(span.observers[0]), TraceServerSpanObserver)
def test_register_server_span_observer(self): client = make_client("test-service") baseplate_observer = TraceBaseplateObserver(client) context_mock = mock.Mock() span = ServerSpan('test-id', 'test-parent-id', 'test-span-id', True, 0, 'test', self.mock_context) baseplate_observer.on_server_span_created(context_mock, span) self.assertEqual(len(span.observers), 1) self.assertEqual(type(span.observers[0]), TraceServerSpanObserver)
def test_no_tracing_without_sampling(self): baseplate_observer = TraceBaseplateObserver("test-service", sample_rate=0) context_mock = mock.Mock() span = ServerSpan("test-id", "test-parent-id", "test-span-id", False, 0, "test") baseplate_observer.on_server_span_created(context_mock, span) self.assertEqual(len(span.observers), 0)