def test_null_child(self): mock_observer = mock.Mock(spec=ServerSpanObserver) mock_observer.on_child_span_created.return_value = None server_span = ServerSpan("trace", "parent", "id", None, 0, "name") server_span.register(mock_observer) child_span = server_span.make_child("child_name") self.assertEqual(child_span.observers, [])
def test_null_child(self): mock_observer = mock.Mock(spec=ServerSpanObserver) mock_observer.on_child_span_created.return_value = None server_span = ServerSpan("trace", "parent", "id", "name") server_span.register(mock_observer) child_span = server_span.make_child("child_name") self.assertEqual(child_span.observers, [])
def test_make_local_span_copies_context(self, mock_getrandbits): mock_getrandbits.return_value = 0xCAFE mock_observer = mock.Mock(spec=ServerSpanObserver) mock_context = mock.Mock() mock_cloned_context = mock.Mock() mock_context.clone.return_value = mock_cloned_context server_span = ServerSpan("trace", "parent", "id", None, 0, "name", mock_context) server_span.register(mock_observer) local_span = server_span.make_child("test_op", local=True, component_name="test_component") self.assertNotEqual(local_span.context, mock_context)
def test_make_local_span_copies_context(self, mock_getrandbits): mock_getrandbits.return_value = 0xCAFE mock_observer = mock.Mock(spec=ServerSpanObserver) mock_context = mock.Mock() mock_cloned_context = mock.Mock(spec=WrappedRequestContext) mock_context.clone.return_value = mock_cloned_context server_span = ServerSpan("trace", "parent", "id", None, 0, "name", mock_context) server_span.register(mock_observer) local_span = server_span.make_child("test_op", local=True, component_name="test_component") self.assertNotEqual(local_span.context, mock_context)
def test_make_child(self, mock_getrandbits): mock_getrandbits.return_value = 0xCAFE mock_observer = mock.Mock(spec=ServerSpanObserver) server_span = ServerSpan("trace", "parent", "id", "name") server_span.register(mock_observer) child_span = server_span.make_child("child_name") self.assertEqual(child_span.name, "child_name") self.assertEqual(child_span.id, 0xCAFE) self.assertEqual(child_span.trace_id, "trace") self.assertEqual(child_span.parent_id, "id") self.assertEqual(mock_observer.on_child_span_created.call_count, 1) self.assertEqual(mock_observer.on_child_span_created.call_args, mock.call(child_span))
def test_make_child(self, mock_getrandbits): mock_getrandbits.return_value = 0xCAFE mock_observer = mock.Mock(spec=ServerSpanObserver) server_span = ServerSpan("trace", "parent", "id", None, 0, "name") server_span.register(mock_observer) child_span = server_span.make_child("child_name") self.assertEqual(child_span.name, "child_name") self.assertEqual(child_span.id, 0xCAFE) self.assertEqual(child_span.trace_id, "trace") self.assertEqual(child_span.parent_id, "id") self.assertEqual(mock_observer.on_child_span_created.call_count, 1) self.assertEqual(mock_observer.on_child_span_created.call_args, mock.call(child_span))
def test_make_local_span(self, mock_getrandbits): mock_getrandbits.return_value = 0xCAFE mock_observer = mock.Mock(spec=ServerSpanObserver) mock_context = mock.Mock() mock_cloned_context = mock.Mock() mock_context.clone.return_value = mock_cloned_context server_span = ServerSpan("trace", "parent", "id", None, 0, "name", mock_context) server_span.register(mock_observer) local_span = server_span.make_child("test_op", local=True, component_name="test_component") self.assertEqual(local_span.name, "test_op") self.assertEqual(local_span.id, 0xCAFE) self.assertEqual(local_span.trace_id, "trace") self.assertEqual(local_span.parent_id, "id") self.assertEqual(mock_observer.on_child_span_created.call_count, 1) self.assertEqual(mock_observer.on_child_span_created.call_args, mock.call(local_span))
def test_make_local_span(self, mock_getrandbits): mock_getrandbits.return_value = 0xCAFE mock_observer = mock.Mock(spec=ServerSpanObserver) mock_context = mock.Mock() mock_cloned_context = mock.Mock(spec=WrappedRequestContext) mock_context.clone.return_value = mock_cloned_context server_span = ServerSpan("trace", "parent", "id", None, 0, "name", mock_context) server_span.register(mock_observer) local_span = server_span.make_child("test_op", local=True, component_name="test_component") self.assertEqual(local_span.name, "test_op") self.assertEqual(local_span.id, 0xCAFE) self.assertEqual(local_span.trace_id, "trace") self.assertEqual(local_span.parent_id, "id") self.assertEqual(mock_observer.on_child_span_created.call_count, 1) self.assertEqual(mock_observer.on_child_span_created.call_args, mock.call(local_span))