Skip to content

Commit ac0df93

Browse files
committed
Updated tests
1 parent f903674 commit ac0df93

File tree

2 files changed

+11
-4
lines changed

2 files changed

+11
-4
lines changed

instrumentation/opentelemetry-instrumentation-kafka-python/tests/test_instrumentation.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020

2121
from opentelemetry.instrumentation.kafka import KafkaInstrumentor
2222
from opentelemetry.instrumentation.kafka.package import (
23-
_instruments,
2423
_instruments_kafka_python,
2524
_instruments_kafka_python_ng,
2625
)
@@ -134,4 +133,7 @@ def _distribution(name):
134133
call("kafka-python"),
135134
],
136135
)
137-
self.assertEqual(package_to_instrument, _instruments)
136+
self.assertEqual(
137+
package_to_instrument,
138+
(_instruments_kafka_python, _instruments_kafka_python_ng),
139+
)

instrumentation/opentelemetry-instrumentation-psycopg2/tests/test_psycopg2_instrumentation.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020
)
2121
from opentelemetry.instrumentation.psycopg2 import Psycopg2Instrumentor
2222
from opentelemetry.instrumentation.psycopg2.package import (
23-
_instruments,
2423
_instruments_psycopg2,
2524
_instruments_psycopg2_binary,
2625
)
@@ -130,7 +129,13 @@ def _distribution(name):
130129
call("psycopg2-binary"),
131130
],
132131
)
133-
self.assertEqual(package_to_instrument, _instruments)
132+
self.assertEqual(
133+
package_to_instrument,
134+
(
135+
_instruments_psycopg2,
136+
_instruments_psycopg2_binary,
137+
),
138+
)
134139

135140
# This test is to verify that the auto instrumentation path
136141
# will auto instrument psycopg2 or psycopg2-binary is installed.

0 commit comments

Comments
 (0)