WIP partially fix distributions test in fastapi

Author: xrmx

instrumentation/opentelemetry-instrumentation-fastapi/tests/test_fastapi_instrumentation.py

@@ -16,7 +16,7 @@
 
 import unittest
 from timeit import default_timer
-from unittest.mock import Mock, patch
+from unittest.mock import patch
 
 import fastapi
 from fastapi.middleware.httpsredirect import HTTPSRedirectMiddleware
@@ -34,9 +34,6 @@
     _server_duration_attrs_old,
 )
 from opentelemetry.instrumentation.asgi import OpenTelemetryMiddleware
-from opentelemetry.instrumentation.auto_instrumentation._load import (
-    _load_instrumentors,
-)
 from opentelemetry.sdk.metrics.export import (
     HistogramDataPoint,
     NumberDataPoint,
@@ -1051,30 +1048,18 @@ class TestAutoInstrumentation(TestBaseAutoFastAPI):
     """
 
     def test_entry_point_exists(self):
-        eps = entry_points("opentelemetry_instrumentor")
-        ep = next(eps)
-        self.assertEqual(ep.dist.key, "opentelemetry-instrumentation-fastapi")
-        self.assertEqual(
-            ep.module_name, "opentelemetry.instrumentation.fastapi"
-        )
-        self.assertEqual(ep.attrs, ("FastAPIInstrumentor",))
+        (ep,) = entry_points(group="opentelemetry_instrumentor")
         self.assertEqual(ep.name, "fastapi")
-        self.assertIsNone(next(eps, None))
 
+    """ FIXME: get_distribution is gone
     @patch("opentelemetry.instrumentation.dependencies.get_distribution")
     def test_instruments_with_fastapi_installed(self, mock_get_distribution):
         mock_get_distribution.side_effect = get_distribution_with_fastapi
         mock_distro = Mock()
         _load_instrumentors(mock_distro)
         mock_get_distribution.assert_called_once_with("fastapi~=0.58")
         self.assertEqual(len(mock_distro.load_instrumentor.call_args_list), 1)
-        args = mock_distro.load_instrumentor.call_args.args
-        ep = args[0]
-        self.assertEqual(ep.dist.key, "opentelemetry-instrumentation-fastapi")
-        self.assertEqual(
-            ep.module_name, "opentelemetry.instrumentation.fastapi"
-        )
-        self.assertEqual(ep.attrs, ("FastAPIInstrumentor",))
+        (ep,) = mock_distro.load_instrumentor.call_args.args
         self.assertEqual(ep.name, "fastapi")
 
     @patch("opentelemetry.instrumentation.dependencies.get_distribution")
@@ -1089,6 +1074,7 @@ def test_instruments_without_fastapi_installed(
             mock_get_distribution("fastapi~=0.58")
         self.assertEqual(len(mock_distro.load_instrumentor.call_args_list), 0)
         mock_distro.load_instrumentor.assert_not_called()
+    """
 
     def _create_app(self):
         # instrumentation is handled by the instrument call