httpx: cleanup None entries in Client mounts before asserting the size

open-telemetry · Aug 25, 2024 · 17f1090 · 17f1090
1 parent bc4d2c5
commit 17f1090
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/instrumentation/opentelemetry-instrumentation-httpx/tests/test_httpx_integration.py b/instrumentation/opentelemetry-instrumentation-httpx/tests/test_httpx_integration.py
@@ -755,8 +755,9 @@ def create_proxy_mounts(self):
             }
 
         def assert_proxy_mounts(self, mounts, num_mounts, transport_type):
-            self.assertEqual(len(mounts), num_mounts)
-            for transport in mounts:
+            valid_mounts = [mount for mount in mounts if mount is not None]
+            self.assertEqual(len(valid_mounts), num_mounts)
+            for transport in valid_mounts:
                 with self.subTest(transport):
                     self.assertIsInstance(
                         transport,