File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed
Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change 11import contextlib
22import typing
3- from unittest .mock import AsyncMock , MagicMock , patch
3+ from unittest .mock import AsyncMock , MagicMock , Mock , patch
44
55import fastapi
66import litestar
7+ import pytest
78from httpx import AsyncClient
89from litestar .middleware .base import DefineMiddleware
910from litestar .testing import AsyncTestClient
@@ -59,7 +60,7 @@ def test_litestar_opentelemetry_bootstrap(
5960 assert isinstance (opentelemetry_bootstrap_result ["middleware" ][0 ], DefineMiddleware )
6061
6162
62- def test_litestar_opentelemetry_terdown (
63+ def test_litestar_opentelemetry_teardown (
6364 minimal_opentelemetry_config : OpentelemetryConfig ,
6465 magic_mock : MagicMock ,
6566) -> None :
@@ -98,8 +99,10 @@ async def test_handler() -> None:
9899
99100
100101async def test_fastapi_opentelemetry_bootstrap_working (
101- minimal_opentelemetry_config : OpentelemetryConfig ,
102+ minimal_opentelemetry_config : OpentelemetryConfig , monkeypatch : pytest . MonkeyPatch
102103) -> None :
104+ monkeypatch .setattr ("opentelemetry.sdk.trace.TracerProvider.shutdown" , Mock ())
105+
103106 opentelemetry_instrument : typing .Final = FastApiOpentelemetryInstrument (minimal_opentelemetry_config )
104107 opentelemetry_instrument .bootstrap ()
105108 fastapi_application : typing .Final = opentelemetry_instrument .bootstrap_after (fastapi .FastAPI ())
You can’t perform that action at this time.
0 commit comments