fix(pymongo): add pin type check

ddtrace/contrib/internal/pymongo/client.py

@@ -103,7 +103,10 @@ def _trace_topology_select_server(func, args, kwargs):
     # Ensure the pin used on the traced mongo client is passed down to the topology instance
     # This allows us to pass the same pin in traced server objects.
     topology_instance = get_argument_value(args, kwargs, 0, "self")
-    ddtrace.Pin.get_from(topology_instance).onto(server)
+    pin = ddtrace.Pin.get_from(topology_instance)
+
+    if pin is not None:
+        pin.onto(server)
     return server
 
 

releasenotes/notes/pymongo-check-pin-c3cb9d1fbdc82fb3.yaml

@@ -0,0 +1,4 @@
+---
+fixes:
+  - |
+    pymongo: add type checking to solve an issue where ``NoneType`` instead of expected ``Pin`` object would throw an error in ``TracedTopology`` method.

tests/contrib/django/test_django_snapshots.py

@@ -1,4 +1,5 @@
 from contextlib import contextmanager
+import logging
 import os
 from pathlib import Path
 import subprocess
@@ -66,9 +67,10 @@ def daphne_client(django_asgi, additional_env=None):
     # Wait for the server to start up
     try:
         print("Waiting for server to start")
-        client.wait(max_tries=120, delay=0.2, initial_wait=2.0)
+        client.wait(max_tries=130, delay=0.2, initial_wait=5)
         print("Server started")
-    except Exception:
+    except Exception as e:
+        logging.warning(e)
         raise AssertionError(
             "Server failed to start, see stdout and stderr logs"
             "\n=== Captured STDOUT ===\n%s=== End of captured STDOUT ==="