Python TraceConfig.trace_config_ctx Examples

Example #1

File: test_tracing.py Project: ticosax/aiohttp

    async def test_send(self, loop, signal, params, param_obj):
        session = Mock()
        trace_request_ctx = Mock()
        callback = Mock(side_effect=asyncio.coroutine(Mock()))

        trace_config = TraceConfig()
        getattr(trace_config, "on_%s" % signal).append(callback)
        trace_config.freeze()
        trace = Trace(
            session, trace_config,
            trace_config.trace_config_ctx(trace_request_ctx=trace_request_ctx))
        await getattr(trace, "send_%s" % signal)(*params)

        callback.assert_called_once_with(
            session, SimpleNamespace(trace_request_ctx=trace_request_ctx),
            param_obj(*params))

Example #2

File: test_tracing.py Project: 3lnc/aiohttp

    async def test_send(self, loop, signal, params, param_obj):
        session = Mock()
        trace_request_ctx = Mock()
        callback = Mock(side_effect=asyncio.coroutine(Mock()))

        trace_config = TraceConfig()
        getattr(trace_config, "on_%s" % signal).append(callback)
        trace_config.freeze()
        trace = Trace(
            session,
            trace_config,
            trace_config.trace_config_ctx(trace_request_ctx=trace_request_ctx)
        )
        await getattr(trace, "send_%s" % signal)(*params)

        callback.assert_called_once_with(
            session,
            SimpleNamespace(trace_request_ctx=trace_request_ctx),
            param_obj(*params)
        )

Example #3

File: test_tracing.py Project: ikamensh/aiohttp

    async def test_send(self, signal, params, param_obj) -> None:
        session = Mock()
        trace_request_ctx = Mock()
        callback = Mock(side_effect=make_mocked_coro(Mock()))

        trace_config = TraceConfig()
        getattr(trace_config, f"on_{signal}").append(callback)
        trace_config.freeze()
        trace = Trace(
            session,
            trace_config,
            trace_config.trace_config_ctx(trace_request_ctx=trace_request_ctx),
        )
        await getattr(trace, f"send_{signal}")(*params)

        callback.assert_called_once_with(
            session,
            SimpleNamespace(trace_request_ctx=trace_request_ctx),
            param_obj(*params),
        )

Example #4

 def test_trace_config_ctx_request_ctx(self) -> None:
     trace_request_ctx = Mock()
     trace_config = TraceConfig()
     trace_config_ctx = trace_config.trace_config_ctx(
         trace_request_ctx=trace_request_ctx)
     assert trace_config_ctx.trace_request_ctx is trace_request_ctx

Example #5

 def test_trace_config_ctx_factory(self) -> None:
     trace_config = TraceConfig(trace_config_ctx_factory=dict)
     assert isinstance(trace_config.trace_config_ctx(), dict)

Example #6

 def test_trace_config_ctx_default(self) -> None:
     trace_config = TraceConfig()
     assert isinstance(trace_config.trace_config_ctx(), SimpleNamespace)

Example #7

File: test_tracing.py Project: 3lnc/aiohttp

 def test_trace_config_ctx_request_ctx(self):
     trace_request_ctx = Mock()
     trace_config = TraceConfig()
     trace_config_ctx = trace_config.trace_config_ctx(
         trace_request_ctx=trace_request_ctx)
     assert trace_config_ctx.trace_request_ctx is trace_request_ctx

Example #8

File: test_tracing.py Project: 3lnc/aiohttp

 def test_trace_config_ctx_factory(self):
     trace_config = TraceConfig(trace_config_ctx_factory=dict)
     assert isinstance(trace_config.trace_config_ctx(), dict)

Example #9

File: test_tracing.py Project: 3lnc/aiohttp

 def test_trace_config_ctx_default(self):
     trace_config = TraceConfig()
     assert isinstance(trace_config.trace_config_ctx(), SimpleNamespace)

Example #10

File: test_tracing.py Project: vuchau/aiohttp

 def test_trace_config_ctx_class(self):
     trace_config = TraceConfig(trace_config_ctx_class=dict)
     assert isinstance(trace_config.trace_config_ctx(), dict)