Skip to content

Commit d904f9b

Browse files
Fix lint issues in OpenTelemetry metrics tests.
Apply black formatting and remove an unused import flagged by flake8. Co-authored-by: Cursor <cursoragent@cursor.com>
1 parent cef7462 commit d904f9b

1 file changed

Lines changed: 7 additions & 6 deletions

File tree

tests/opentelemetry/test_metrics.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -824,7 +824,6 @@ def test_query_session_init_metrics_defaults(self):
824824
assert session._metrics_state == "used"
825825

826826
def test_sync_pool_acquire_from_queue_updates_session_count(self, metrics_setup):
827-
from ydb import issues
828827
from ydb.opentelemetry.metrics import QUERY_SESSION_COUNT
829828
from ydb.query.pool import QuerySessionPool
830829

@@ -906,15 +905,15 @@ def test_sync_session_create_increments_session_count(self, metrics_setup, monke
906905
def fake_span_ctx(**kwargs):
907906
yield MagicMock()
908907

909-
monkeypatch.setattr("ydb.query.session.create_ydb_span", lambda *args, **kwargs: MagicMock(attach_context=fake_span_ctx))
908+
monkeypatch.setattr(
909+
"ydb.query.session.create_ydb_span", lambda *args, **kwargs: MagicMock(attach_context=fake_span_ctx)
910+
)
910911
monkeypatch.setattr(qs, "_create_call", MagicMock())
911912
monkeypatch.setattr(qs, "_attach", MagicMock())
912913

913914
qs.create()
914915
assert qs._metrics_counted
915-
assert (
916-
_single_point(metrics_setup, QUERY_SESSION_COUNT).attributes["ydb.query.session.state"] == "used"
917-
)
916+
assert _single_point(metrics_setup, QUERY_SESSION_COUNT).attributes["ydb.query.session.state"] == "used"
918917

919918
def test_sync_session_close_decrements_session_count(self, metrics_setup):
920919
from ydb.opentelemetry.metrics import QUERY_SESSION_COUNT
@@ -1018,7 +1017,9 @@ async def test_async_session_create_increments_session_count(self, metrics_setup
10181017
def fake_span_ctx(**kwargs):
10191018
yield MagicMock()
10201019

1021-
monkeypatch.setattr("ydb.aio.query.session.create_ydb_span", lambda *args, **kwargs: MagicMock(attach_context=fake_span_ctx))
1020+
monkeypatch.setattr(
1021+
"ydb.aio.query.session.create_ydb_span", lambda *args, **kwargs: MagicMock(attach_context=fake_span_ctx)
1022+
)
10221023
monkeypatch.setattr(qs, "_create_call", AsyncMock())
10231024
monkeypatch.setattr(qs, "_attach", AsyncMock())
10241025

0 commit comments

Comments
 (0)