1- #include < src/client/query/ impl/query_spans .h>
1+ #include < src/client/impl/observability/operation_span .h>
22#include < tests/common/fake_trace_provider.h>
33
4+ #include < library/cpp/logger/log.h>
5+
46#include < gtest/gtest.h>
57
68using namespace NYdb ;
7- using namespace NYdb ::NQuery;
8- using namespace NYdb ::NMetrics;
9+ using namespace NYdb ::NObservability;
10+
11+ using namespace NYdb ::NTrace;
912using namespace NYdb ::NTests;
1013
1114class QuerySpanTest : public ::testing::Test {
@@ -18,23 +21,23 @@ class QuerySpanTest : public ::testing::Test {
1821};
1922
2023TEST_F (QuerySpanTest, SpanNameFormat) {
21- TQuerySpan span (Tracer, " ExecuteQuery" , " localhost:2135" );
24+ TOperationSpan span (Tracer, " ExecuteQuery" , " localhost:2135" , TLog () );
2225 span.End (EStatus::SUCCESS);
2326
2427 ASSERT_EQ (Tracer->SpanCount (), 1u );
2528 EXPECT_EQ (Tracer->GetLastSpanRecord ().Name , " ydb.ExecuteQuery" );
2629}
2730
2831TEST_F (QuerySpanTest, SpanKindIsClient) {
29- TQuerySpan span (Tracer, " CreateSession" , " localhost:2135" );
32+ TOperationSpan span (Tracer, " CreateSession" , " localhost:2135" , TLog () );
3033 span.End (EStatus::SUCCESS);
3134
3235 ASSERT_EQ (Tracer->SpanCount (), 1u );
3336 EXPECT_EQ (Tracer->GetLastSpanRecord ().Kind , ESpanKind::CLIENT);
3437}
3538
3639TEST_F (QuerySpanTest, DbSystemAttribute) {
37- TQuerySpan span (Tracer, " ExecuteQuery" , " localhost:2135" );
40+ TOperationSpan span (Tracer, " ExecuteQuery" , " localhost:2135" , TLog () );
3841 span.End (EStatus::SUCCESS);
3942
4043 auto fakeSpan = Tracer->GetLastSpan ();
@@ -43,7 +46,7 @@ TEST_F(QuerySpanTest, DbSystemAttribute) {
4346}
4447
4548TEST_F (QuerySpanTest, ServerAddressAndPort) {
46- TQuerySpan span (Tracer, " Commit" , " ydb.server:2135" );
49+ TOperationSpan span (Tracer, " Commit" , " ydb.server:2135" , TLog () );
4750 span.End (EStatus::SUCCESS);
4851
4952 auto fakeSpan = Tracer->GetLastSpan ();
@@ -53,7 +56,7 @@ TEST_F(QuerySpanTest, ServerAddressAndPort) {
5356}
5457
5558TEST_F (QuerySpanTest, ServerAddressCustomPort) {
56- TQuerySpan span (Tracer, " Rollback" , " myhost:9090" );
59+ TOperationSpan span (Tracer, " Rollback" , " myhost:9090" , TLog () );
5760 span.End (EStatus::SUCCESS);
5861
5962 auto fakeSpan = Tracer->GetLastSpan ();
@@ -63,7 +66,7 @@ TEST_F(QuerySpanTest, ServerAddressCustomPort) {
6366}
6467
6568TEST_F (QuerySpanTest, ServerAddressNoPortDefaultsTo2135) {
66- TQuerySpan span (Tracer, " ExecuteQuery" , " myhost" );
69+ TOperationSpan span (Tracer, " ExecuteQuery" , " myhost" , TLog () );
6770 span.End (EStatus::SUCCESS);
6871
6972 auto fakeSpan = Tracer->GetLastSpan ();
@@ -73,7 +76,7 @@ TEST_F(QuerySpanTest, ServerAddressNoPortDefaultsTo2135) {
7376}
7477
7578TEST_F (QuerySpanTest, IPv6EndpointParsing) {
76- TQuerySpan span (Tracer, " ExecuteQuery" , " [::1]:2136" );
79+ TOperationSpan span (Tracer, " ExecuteQuery" , " [::1]:2136" , TLog () );
7780 span.End (EStatus::SUCCESS);
7881
7982 auto fakeSpan = Tracer->GetLastSpan ();
@@ -83,7 +86,7 @@ TEST_F(QuerySpanTest, IPv6EndpointParsing) {
8386}
8487
8588TEST_F (QuerySpanTest, IPv6EndpointNoPort) {
86- TQuerySpan span (Tracer, " ExecuteQuery" , " [fe80::1]" );
89+ TOperationSpan span (Tracer, " ExecuteQuery" , " [fe80::1]" , TLog () );
8790 span.End (EStatus::SUCCESS);
8891
8992 auto fakeSpan = Tracer->GetLastSpan ();
@@ -93,7 +96,7 @@ TEST_F(QuerySpanTest, IPv6EndpointNoPort) {
9396}
9497
9598TEST_F (QuerySpanTest, PeerEndpointAttributes) {
96- TQuerySpan span (Tracer, " ExecuteQuery" , " discovery.ydb:2135" );
99+ TOperationSpan span (Tracer, " ExecuteQuery" , " discovery.ydb:2135" , TLog () );
97100 span.SetPeerEndpoint (" 10.0.0.1:2136" );
98101 span.End (EStatus::SUCCESS);
99102
@@ -104,7 +107,7 @@ TEST_F(QuerySpanTest, PeerEndpointAttributes) {
104107}
105108
106109TEST_F (QuerySpanTest, SuccessStatusRecorded) {
107- TQuerySpan span (Tracer, " Commit" , " localhost:2135" );
110+ TOperationSpan span (Tracer, " Commit" , " localhost:2135" , TLog () );
108111 span.End (EStatus::SUCCESS);
109112
110113 auto fakeSpan = Tracer->GetLastSpan ();
@@ -115,7 +118,7 @@ TEST_F(QuerySpanTest, SuccessStatusRecorded) {
115118}
116119
117120TEST_F (QuerySpanTest, ErrorStatusSetsErrorType) {
118- TQuerySpan span (Tracer, " Rollback" , " localhost:2135" );
121+ TOperationSpan span (Tracer, " Rollback" , " localhost:2135" , TLog () );
119122 span.End (EStatus::UNAVAILABLE);
120123
121124 auto fakeSpan = Tracer->GetLastSpan ();
@@ -126,7 +129,7 @@ TEST_F(QuerySpanTest, ErrorStatusSetsErrorType) {
126129}
127130
128131TEST_F (QuerySpanTest, SpanIsEndedAfterEnd) {
129- TQuerySpan span (Tracer, " ExecuteQuery" , " localhost:2135" );
132+ TOperationSpan span (Tracer, " ExecuteQuery" , " localhost:2135" , TLog () );
130133 auto fakeSpan = Tracer->GetLastSpan ();
131134 ASSERT_NE (fakeSpan, nullptr );
132135
@@ -137,7 +140,7 @@ TEST_F(QuerySpanTest, SpanIsEndedAfterEnd) {
137140
138141TEST_F (QuerySpanTest, NullTracerDoesNotCrash) {
139142 EXPECT_NO_THROW ({
140- TQuerySpan span (nullptr , " ExecuteQuery" , " localhost:2135" );
143+ TOperationSpan span (nullptr , " ExecuteQuery" , " localhost:2135" , TLog () );
141144 span.SetPeerEndpoint (" 10.0.0.1:2136" );
142145 span.AddEvent (" retry" , {{" attempt" , " 1" }});
143146 span.End (EStatus::SUCCESS);
@@ -146,7 +149,7 @@ TEST_F(QuerySpanTest, NullTracerDoesNotCrash) {
146149
147150TEST_F (QuerySpanTest, DestructorEndsSpan) {
148151 auto fakeSpan = [&]() -> std::shared_ptr<TFakeSpan> {
149- TQuerySpan span (Tracer, " CreateSession" , " localhost:2135" );
152+ TOperationSpan span (Tracer, " CreateSession" , " localhost:2135" , TLog () );
150153 return Tracer->GetLastSpan ();
151154 }();
152155
@@ -156,7 +159,7 @@ TEST_F(QuerySpanTest, DestructorEndsSpan) {
156159
157160TEST_F (QuerySpanTest, ExplicitEndThenDestructorDoesNotDoubleEnd) {
158161 auto fakeSpan = [&]() -> std::shared_ptr<TFakeSpan> {
159- TQuerySpan span (Tracer, " Commit" , " localhost:2135" );
162+ TOperationSpan span (Tracer, " Commit" , " localhost:2135" , TLog () );
160163 span.End (EStatus::SUCCESS);
161164 return Tracer->GetLastSpan ();
162165 }();
@@ -166,7 +169,7 @@ TEST_F(QuerySpanTest, ExplicitEndThenDestructorDoesNotDoubleEnd) {
166169}
167170
168171TEST_F (QuerySpanTest, AddEventForwarded) {
169- TQuerySpan span (Tracer, " ExecuteQuery" , " localhost:2135" );
172+ TOperationSpan span (Tracer, " ExecuteQuery" , " localhost:2135" , TLog () );
170173 span.AddEvent (" retry" , {{" ydb.attempt" , " 2" }, {" error.type" , " UNAVAILABLE" }});
171174 span.End (EStatus::SUCCESS);
172175
@@ -180,7 +183,7 @@ TEST_F(QuerySpanTest, AddEventForwarded) {
180183}
181184
182185TEST_F (QuerySpanTest, EmptyPeerEndpointIgnored) {
183- TQuerySpan span (Tracer, " CreateSession" , " localhost:2135" );
186+ TOperationSpan span (Tracer, " CreateSession" , " localhost:2135" , TLog () );
184187 span.SetPeerEndpoint (" " );
185188 span.End (EStatus::SUCCESS);
186189
@@ -194,7 +197,7 @@ TEST_F(QuerySpanTest, AllFourOperationNames) {
194197 const std::vector<std::string> operations = {" CreateSession" , " ExecuteQuery" , " Commit" , " Rollback" };
195198
196199 for (const auto & op : operations) {
197- TQuerySpan span (Tracer, op, " localhost:2135" );
200+ TOperationSpan span (Tracer, op, " localhost:2135" , TLog () );
198201 span.End (EStatus::SUCCESS);
199202 }
200203
@@ -223,7 +226,7 @@ TEST_F(QuerySpanTest, MultipleErrorStatuses) {
223226 };
224227
225228 for (auto status : errorStatuses) {
226- TQuerySpan span (Tracer, " ExecuteQuery" , " localhost:2135" );
229+ TOperationSpan span (Tracer, " ExecuteQuery" , " localhost:2135" , TLog () );
227230 span.End (status);
228231
229232 auto fakeSpan = Tracer->GetLastSpan ();
@@ -236,7 +239,7 @@ TEST_F(QuerySpanTest, MultipleErrorStatuses) {
236239
237240TEST_F (QuerySpanTest, EmptyEndpointDoesNotCrash) {
238241 EXPECT_NO_THROW ({
239- TQuerySpan span (Tracer, " ExecuteQuery" , " " );
242+ TOperationSpan span (Tracer, " ExecuteQuery" , " " , TLog () );
240243 span.End (EStatus::SUCCESS);
241244 });
242245}
0 commit comments