async def test_middleware_with_valid_ip(tracer, version, address_in,
address_out):
if address_out is None:
address_out = address_in
app = web.Application()
az.setup(app, tracer)
# Fake transport
transp = Mock()
transp.get_extra_info.return_value = (address_in, "0")
async def handler(request):
return web.Response(body=b"data")
req = make_mocked_request("GET",
"/",
headers={"token": "x"},
transport=transp,
app=app)
middleware = middleware_maker()
with patch("aiojaeger.span.Span.remote_endpoint") as mocked_remote_ep:
await middleware(req, handler)
assert mocked_remote_ep.call_count == 1
args, kwargs = mocked_remote_ep.call_args
assert kwargs[version] == address_out
async def test_middleware_with_default_transport(zipkin_tracer: Tracer):
app = web.Application()
az.setup(app, zipkin_tracer)
async def handler(request):
return web.Response(body=b"data")
req = make_mocked_request("GET", "/aa", headers={"token": "x"}, app=app)
req.match_info.route.resource.canonical = "/{pid}"
middleware = middleware_maker()
await middleware(req, handler)
span = az.request_span(req)
assert span
assert len(zipkin_tracer.transport.records) == 1
rec = zipkin_tracer.transport.records[0]
assert rec.asdict()["tags"][az.HTTP_ROUTE] == "/{pid}"
# noop span does not produce records
headers = {"b3": "1-2-0"}
req_noop = make_mocked_request("GET", "/", headers=headers, app=app)
await middleware(req_noop, handler)
span = az.request_span(req_noop)
assert span
assert len(zipkin_tracer.transport.records) == 1
async def test_middleware_with_handler_404(tracer):
app = web.Application()
az.setup(app, tracer)
async def handler(request):
raise HTTPNotFound
req = make_mocked_request("GET", "/", headers={"token": "x"}, app=app)
middleware = middleware_maker()
with pytest.raises(HTTPException):
await middleware(req, handler)
async def test_middleware_with_not_skip_route(tracer, fake_transport):
async def handler(request):
return web.Response(body=b"data")
app = web.Application()
skip_route = app.router.add_get("/", handler)
az.setup(app, tracer)
match_info = Mock()
match_info.route = skip_route
req = make_mocked_request("GET", "/", headers={"token": "x"}, app=app)
req._match_info = match_info
middleware = middleware_maker(skip_routes=[skip_route])
await middleware(req, handler)
assert len(fake_transport.records) == 0