manually sample before multiprocessing.Process tests

DataDog · Mar 1, 2024 · efcf424 · efcf424
1 parent f2beeea
commit efcf424
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/integration/test_integration_snapshots.py b/tests/integration/test_integration_snapshots.py
@@ -124,7 +124,9 @@ def task(tracer):
             pass
         tracer.shutdown()
 
-    with tracer.trace("parent"):
+    with tracer.trace("parent") as parent:
+        # need to manually sample before Popen
+        tracer._sampler.sample(parent)
         p = multiprocessing.Process(target=task, args=(tracer,))
         p.start()
         p.join()
@@ -152,7 +154,9 @@ def task2(tracer):
             p.join()
         tracer.shutdown()
 
-    with tracer.trace("parent"):
+    with tracer.trace("parent") as parent:
+        # need to manually sample before Popen
+        tracer._sampler.sample(parent)
         p = multiprocessing.Process(target=task, args=(tracer,))
         p.start()
         p.join()