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))
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)
)
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),
)
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
def test_trace_config_ctx_factory(self) -> None:
trace_config = TraceConfig(trace_config_ctx_factory=dict)
assert isinstance(trace_config.trace_config_ctx(), dict)
def test_trace_config_ctx_default(self) -> None:
trace_config = TraceConfig()
assert isinstance(trace_config.trace_config_ctx(), SimpleNamespace)
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
def test_trace_config_ctx_factory(self):
trace_config = TraceConfig(trace_config_ctx_factory=dict)
assert isinstance(trace_config.trace_config_ctx(), dict)
def test_trace_config_ctx_default(self):
trace_config = TraceConfig()
assert isinstance(trace_config.trace_config_ctx(), SimpleNamespace)
def test_trace_config_ctx_class(self):
trace_config = TraceConfig(trace_config_ctx_class=dict)
assert isinstance(trace_config.trace_config_ctx(), dict)