mirror of
https://github.com/open-telemetry/opentelemetry-python-contrib.git
synced 2025-07-30 05:32:30 +08:00
Add test for database_index
This commit is contained in:
@ -115,3 +115,32 @@ class TestRedisInstrument(TestBase):
|
|||||||
child_span.attributes.get("service"), self.test_service
|
child_span.attributes.get("service"), self.test_service
|
||||||
)
|
)
|
||||||
self.assertEqual(child_span.name, "GET")
|
self.assertEqual(child_span.name, "GET")
|
||||||
|
|
||||||
|
|
||||||
|
class TestRedisDBIndexInstrument(TestBase):
|
||||||
|
|
||||||
|
def setUp(self):
|
||||||
|
super().setUp()
|
||||||
|
self.redis_client = redis.Redis(port=6379, db=10)
|
||||||
|
self.redis_client.flushall()
|
||||||
|
RedisInstrumentor().instrument(tracer_provider=self.tracer_provider)
|
||||||
|
|
||||||
|
def tearDown(self):
|
||||||
|
super().tearDown()
|
||||||
|
RedisInstrumentor().uninstrument()
|
||||||
|
|
||||||
|
def _check_span(self, span, name):
|
||||||
|
self.assertEqual(span.name, name)
|
||||||
|
self.assertIs(span.status.status_code, trace.status.StatusCode.UNSET)
|
||||||
|
self.assertEqual(span.attributes["net.peer.name"], "localhost")
|
||||||
|
self.assertEqual(span.attributes["net.peer.ip"], 6379)
|
||||||
|
self.assertEqual(span.attributes["db.redis.database_index"], 10)
|
||||||
|
self.assertNotIn("db.name", span.attributes)
|
||||||
|
|
||||||
|
def test_get(self):
|
||||||
|
self.assertIsNone(self.redis_client.get("paneer"))
|
||||||
|
spans = self.memory_exporter.get_finished_spans()
|
||||||
|
self.assertEqual(len(spans), 1)
|
||||||
|
span = spans[0]
|
||||||
|
self._check_span(span, "GET")
|
||||||
|
self.assertEqual(span.attributes.get("db.statement"), "GET paneer")
|
||||||
|
Reference in New Issue
Block a user