Python Span.registerの例

コード例 #1

ファイル: core_tests.py プロジェクト: reddit/baseplate

    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)

コード例 #2

ファイル: core_tests.py プロジェクト: us1415/baseplate

    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)

コード例 #3

ファイル: core_tests.py プロジェクト: dwick/baseplate

    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)

コード例 #4

ファイル: core_tests.py プロジェクト: jackniu1/baseplate

    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)

コード例 #5

ファイル: core_tests.py プロジェクト: dwick/baseplate

    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))

コード例 #6

ファイル: core_tests.py プロジェクト: jackniu1/baseplate

    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))

コード例 #7

ファイル: core_tests.py プロジェクト: us1415/baseplate

    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)

コード例 #8

ファイル: core_tests.py プロジェクト: reddit/baseplate

    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)

コード例 #9

ファイル: core_tests.py プロジェクト: us1415/baseplate

    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)

コード例 #10

ファイル: core_tests.py プロジェクト: reddit/baseplate

    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)