Esempi in Python per TraceConfig.trace_config_ctx

Esempio n. 1

File: test_tracing.py Progetto: 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))

Esempio n. 2

File: test_tracing.py Progetto: 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)
        )

Esempio n. 3

File: test_tracing.py Progetto: 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),
        )

Esempio n. 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

Esempio n. 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)

Esempio n. 6

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

Esempio n. 7

File: test_tracing.py Progetto: 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

Esempio n. 8

File: test_tracing.py Progetto: 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)

Esempio n. 9

File: test_tracing.py Progetto: 3lnc/aiohttp

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

Esempio n. 10

File: test_tracing.py Progetto: 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)