fhjkasdfhksadjh

Author: rjduffner

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

@@ -10,7 +10,6 @@
 from opentelemetry.instrumentation.kafka import KafkaInstrumentor
 from opentelemetry.instrumentation.kafka.package import (
     _instruments_kafka_python,
-    _instruments_kafka_python_ng,
 )
 
 
@@ -54,50 +53,6 @@ def _distribution(name):
         )
         self.assertEqual(package_to_instrument, (_instruments_kafka_python,))
 
-    @patch("opentelemetry.instrumentation.kafka.distribution")
-    def test_instrumentation_dependencies_kafka_python_ng_installed(
-        self, mock_distribution
-    ) -> None:
-        instrumentation = KafkaInstrumentor()
-
-        def _distribution(name):
-            if name == "kafka-python-ng":
-                return None
-            raise PackageNotFoundError
-
-        mock_distribution.side_effect = _distribution
-        package_to_instrument = instrumentation.instrumentation_dependencies()
-
-        self.assertEqual(mock_distribution.call_count, 1)
-        self.assertEqual(
-            mock_distribution.mock_calls, [call("kafka-python-ng")]
-        )
-        self.assertEqual(
-            package_to_instrument, (_instruments_kafka_python_ng,)
-        )
-
-    @patch("opentelemetry.instrumentation.kafka.distribution")
-    def test_instrumentation_dependencies_both_installed(
-        self, mock_distribution
-    ) -> None:
-        instrumentation = KafkaInstrumentor()
-
-        def _distribution(name):
-            # The function returns None here for all names
-            # to simulate both packages being installed
-            return None
-
-        mock_distribution.side_effect = _distribution
-        package_to_instrument = instrumentation.instrumentation_dependencies()
-
-        self.assertEqual(mock_distribution.call_count, 1)
-        self.assertEqual(
-            mock_distribution.mock_calls, [call("kafka-python-ng")]
-        )
-        self.assertEqual(
-            package_to_instrument, (_instruments_kafka_python_ng,)
-        )
-
     @patch("opentelemetry.instrumentation.kafka.distribution")
     def test_instrumentation_dependencies_none_installed(
         self, mock_distribution
@@ -122,7 +77,4 @@ def _distribution(name):
                 call("kafka-python"),
             ],
         )
-        self.assertEqual(
-            package_to_instrument,
-            (_instruments_kafka_python, _instruments_kafka_python_ng),
-        )
+        self.assertEqual(package_to_instrument, (_instruments_kafka_python,))