Skip to content

Commit c13831a

Browse files
committed
Fix hanging tests
1 parent 6dd9137 commit c13831a

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tests/instruments/test_opentelemetry.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
import contextlib
22
import typing
3-
from unittest.mock import AsyncMock, MagicMock, patch
3+
from unittest.mock import AsyncMock, MagicMock, Mock, patch
44

55
import fastapi
66
import litestar
7+
import pytest
78
from httpx import AsyncClient
89
from litestar.middleware.base import DefineMiddleware
910
from 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

100101
async 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())

0 commit comments

Comments
 (0)