@@ -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