|
11 | 11 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
12 | 12 | # See the License for the specific language governing permissions and |
13 | 13 | # limitations under the License. |
| 14 | + |
14 | 15 | from unittest import mock |
15 | 16 |
|
16 | 17 | import pymemcache |
|
26 | 27 |
|
27 | 28 | from opentelemetry import trace as trace_api |
28 | 29 | from opentelemetry.instrumentation.pymemcache import PymemcacheInstrumentor |
29 | | -from opentelemetry.semconv.trace import SpanAttributes |
| 30 | +from opentelemetry.semconv._incubating.attributes.db_attributes import ( |
| 31 | + DB_STATEMENT, |
| 32 | + DB_SYSTEM, |
| 33 | +) |
| 34 | +from opentelemetry.semconv._incubating.attributes.net_attributes import ( |
| 35 | + NET_PEER_NAME, |
| 36 | + NET_PEER_PORT, |
| 37 | +) |
30 | 38 | from opentelemetry.test.test_base import TestBase |
31 | 39 | from opentelemetry.trace import get_tracer |
32 | 40 |
|
@@ -82,18 +90,10 @@ def check_spans(self, spans, num_expected, queries_expected): |
82 | 90 | command, *_ = query.split(" ") |
83 | 91 | self.assertEqual(span.name, command) |
84 | 92 | self.assertIs(span.kind, trace_api.SpanKind.CLIENT) |
85 | | - self.assertEqual( |
86 | | - span.attributes[SpanAttributes.NET_PEER_NAME], TEST_HOST |
87 | | - ) |
88 | | - self.assertEqual( |
89 | | - span.attributes[SpanAttributes.NET_PEER_PORT], TEST_PORT |
90 | | - ) |
91 | | - self.assertEqual( |
92 | | - span.attributes[SpanAttributes.DB_SYSTEM], "memcached" |
93 | | - ) |
94 | | - self.assertEqual( |
95 | | - span.attributes[SpanAttributes.DB_STATEMENT], query |
96 | | - ) |
| 93 | + self.assertEqual(span.attributes[NET_PEER_NAME], TEST_HOST) |
| 94 | + self.assertEqual(span.attributes[NET_PEER_PORT], TEST_PORT) |
| 95 | + self.assertEqual(span.attributes[DB_SYSTEM], "memcached") |
| 96 | + self.assertEqual(span.attributes[DB_STATEMENT], query) |
97 | 97 |
|
98 | 98 | def test_set_success(self): |
99 | 99 | client = self.make_client([b"STORED\r\n"]) |
@@ -248,12 +248,8 @@ def test_set_get(self): |
248 | 248 | spans = self.memory_exporter.get_finished_spans() |
249 | 249 |
|
250 | 250 | self.assertEqual(len(spans), 2) |
251 | | - self.assertEqual( |
252 | | - spans[0].attributes[SpanAttributes.NET_PEER_NAME], TEST_HOST |
253 | | - ) |
254 | | - self.assertEqual( |
255 | | - spans[0].attributes[SpanAttributes.NET_PEER_PORT], TEST_PORT |
256 | | - ) |
| 251 | + self.assertEqual(spans[0].attributes[NET_PEER_NAME], TEST_HOST) |
| 252 | + self.assertEqual(spans[0].attributes[NET_PEER_PORT], TEST_PORT) |
257 | 253 |
|
258 | 254 | def test_append_stored(self): |
259 | 255 | client = self.make_client([b"STORED\r\n"]) |
@@ -576,18 +572,10 @@ def check_spans(self, spans, num_expected, queries_expected): |
576 | 572 | command, *_ = query.split(" ") |
577 | 573 | self.assertEqual(span.name, command) |
578 | 574 | self.assertIs(span.kind, trace_api.SpanKind.CLIENT) |
579 | | - self.assertEqual( |
580 | | - span.attributes[SpanAttributes.NET_PEER_NAME], TEST_HOST |
581 | | - ) |
582 | | - self.assertEqual( |
583 | | - span.attributes[SpanAttributes.NET_PEER_PORT], TEST_PORT |
584 | | - ) |
585 | | - self.assertEqual( |
586 | | - span.attributes[SpanAttributes.DB_SYSTEM], "memcached" |
587 | | - ) |
588 | | - self.assertEqual( |
589 | | - span.attributes[SpanAttributes.DB_STATEMENT], query |
590 | | - ) |
| 575 | + self.assertEqual(span.attributes[NET_PEER_NAME], TEST_HOST) |
| 576 | + self.assertEqual(span.attributes[NET_PEER_PORT], TEST_PORT) |
| 577 | + self.assertEqual(span.attributes[DB_SYSTEM], "memcached") |
| 578 | + self.assertEqual(span.attributes[DB_STATEMENT], query) |
591 | 579 |
|
592 | 580 | def test_delete_many_found(self): |
593 | 581 | client = self.make_client([b"STORED\r", b"\n", b"DELETED\r\n"]) |
|
0 commit comments