def test_should_sample_utilizes_sampled_setting(self): client = make_client("test-service", sample_rate=0) baseplate_observer = TraceBaseplateObserver(client) span_with_sampled_flag = Span("test-id", "test-parent", "test-span-id", True, 0, "test", self.mock_context) self.assertTrue( baseplate_observer.should_sample(span_with_sampled_flag))
def setUp(self): thread_patch = mock.patch("threading.Thread", autospec=True) thread_patch.start() self.addCleanup(thread_patch.stop) configurator = Configurator() configurator.add_route("example", "/example", request_method="GET") configurator.add_view(example_application, route_name="example", renderer="json") configurator.add_route("local_test", "/local_test", request_method="GET") configurator.add_view(local_parent_trace_within_context, route_name="local_test", renderer="json") self.client = make_client("test-service") self.observer = TraceBaseplateObserver(self.client) self.baseplate = Baseplate() self.baseplate.register(self.observer) self.baseplate_configurator = BaseplateConfigurator( self.baseplate, header_trust_handler=StaticTrustHandler(trust_headers=True), ) configurator.include(self.baseplate_configurator.includeme) app = configurator.make_wsgi_app() self.local_span_ids = [] self.local_span_observers = [] self.test_app = webtest.TestApp(app)
def test_should_sample_utilizes_sample_rate(self): client = make_client("test-service", sample_rate=1) baseplate_observer = TraceBaseplateObserver(client) span = Span("test-id", "test-parent", "test-span-id", None, 0, "test", self.mock_context) self.assertTrue(baseplate_observer.should_sample(span)) baseplate_observer.sample_rate = 0 self.assertFalse(baseplate_observer.should_sample(span))
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_configure_tracing_with_defaults_new_style(self): baseplate = Baseplate() self.assertEqual(0, len(baseplate.observers)) client = make_client("test") baseplate.configure_tracing(client) self.assertEqual(1, len(baseplate.observers)) tracing_observer = baseplate.observers[0] self.assertEqual("test", tracing_observer.service_name)
def test_should_sample_utilizes_force_sampling(self): client = make_client("test-service", sample_rate=0) baseplate_observer = TraceBaseplateObserver(client) span_with_forced = Span("test-id", "test-parent", "test-span-id", False, 1, "test", self.mock_context) span_without_forced = Span("test-id", "test-parent", "test-span-id", False, 0, "test", self.mock_context) self.assertTrue(baseplate_observer.should_sample(span_with_forced)) self.assertFalse(baseplate_observer.should_sample(span_without_forced))
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_remote_recorder_setup(self): client = make_client("test-service", tracing_endpoint=Endpoint("test:1111")) baseplate_observer = TraceBaseplateObserver(client) self.assertTrue(isinstance(baseplate_observer.recorder, RemoteRecorder))
def test_sets_hostname(self): client = make_client("test-service") baseplate_observer = TraceBaseplateObserver(client) self.assertIsNotNone(baseplate_observer.hostname)
def test_logging_recorder_setup(self): client = make_client("test-service") baseplate_observer = TraceBaseplateObserver(client) self.assertEqual(type(baseplate_observer.recorder), LoggingRecorder)
def test_null_recorder_setup(self): client = make_client("test-service", log_if_unconfigured=False) baseplate_observer = TraceBaseplateObserver(client) self.assertEqual(type(baseplate_observer.recorder), NullRecorder)