Python Baseplate.make_root_span Examples

Example #1

File: core_tests.py Project: Omosofe/baseplate

    def test_null_root_observer(self):
        mock_context = mock.Mock()
        mock_observer = mock.Mock(spec=BaseplateObserver)
        mock_observer.on_root_span_created.return_value = None

        baseplate = Baseplate()
        baseplate.register(mock_observer)
        root_span = baseplate.make_root_span(mock_context, 1, 2, "name", 3)

        self.assertEqual(root_span.observers, [])

Example #2

File: core_tests.py Project: jackniu1/baseplate

    def test_null_root_observer(self):
        mock_context = mock.Mock()
        mock_observer = mock.Mock(spec=BaseplateObserver)
        mock_observer.on_root_span_created.return_value = None

        baseplate = Baseplate()
        baseplate.register(mock_observer)
        root_span = baseplate.make_root_span(mock_context, 1, 2, "name", 3)

        self.assertEqual(root_span.observers, [])

Example #3

File: core_tests.py Project: dwick/baseplate

    def test_root_observer_made(self):
        mock_context = mock.Mock()
        mock_observer = mock.Mock(spec=BaseplateObserver)

        baseplate = Baseplate()
        baseplate.register(mock_observer)
        root_span = baseplate.make_root_span(mock_context, "name", TraceInfo(1, 2, 3))

        self.assertEqual(baseplate.observers, [mock_observer])
        self.assertEqual(mock_observer.on_root_span_created.call_count, 1)
        self.assertEqual(mock_observer.on_root_span_created.call_args,
            mock.call(mock_context, root_span))

Example #4

File: core_tests.py Project: jackniu1/baseplate

    def test_root_observer_made(self):
        mock_context = mock.Mock()
        mock_observer = mock.Mock(spec=BaseplateObserver)

        baseplate = Baseplate()
        baseplate.register(mock_observer)
        root_span = baseplate.make_root_span(mock_context, 1, 2, "name", 3)

        self.assertEqual(baseplate.observers, [mock_observer])
        self.assertEqual(mock_observer.on_root_span_created.call_count, 1)
        self.assertEqual(mock_observer.on_root_span_created.call_args,
            mock.call(mock_context, root_span))