Python patchの例

コード例 #1

ファイル: test_urllib3.py プロジェクト: twosigmajab/dd-trace-py

    def setUp(self):
        super(BaseUrllib3TestCase, self).setUp()

        patch()
        self.http = urllib3.PoolManager()
        Pin.override(urllib3.connectionpool.HTTPConnectionPool,
                     tracer=self.tracer)

コード例 #2

ファイル: test_urllib3.py プロジェクト: twosigmajab/dd-trace-py

    def test_double_patch(self):
        """Ensure that double patch doesn't duplicate instrumentation"""
        patch()
        connpool = urllib3.connectionpool.HTTPConnectionPool(SOCKET)
        setattr(connpool, "datadog_tracer", self.tracer)

        out = connpool.urlopen("GET", URL_200)
        assert out.status == 200
        spans = self.pop_spans()
        assert len(spans) == 1

コード例 #3

ファイル: test_urllib3.py プロジェクト: goodspark/dd-trace-py

def patch_urllib3():
    patch()
    try:
        yield
    finally:
        unpatch()