update tests

Author: taegyunkim

tests/profiling_v2/collector/test_stack.py

@@ -7,13 +7,13 @@
 import uuid
 
 import pytest
-from tests.profiling_v2.collector import test_collector
 
 from ddtrace import ext
 from ddtrace.internal.datadog.profiling import ddup
 from ddtrace.profiling.collector import stack
 from tests.conftest import get_original_test_name
 from tests.profiling_v2.collector import pprof_utils
+from tests.profiling_v2.collector import test_collector
 
 
 # Python 3.11.9 is not compatible with gevent, https://github.com/gevent/gevent/issues/2040
@@ -25,6 +25,26 @@
 )
 
 
+def func1():
+    return func2()
+
+
+def func2():
+    return func3()
+
+
+def func3():
+    return func4()
+
+
+def func4():
+    return func5()
+
+
+def func5():
+    return time.sleep(1)
+
+
 # Use subprocess as ddup config persists across tests.
 @pytest.mark.subprocess(
     env=dict(
@@ -35,10 +55,9 @@
 def test_collect_truncate():
     import os
 
-    from tests.profiling_v2.collector.test_stack import func1
-
     from ddtrace.profiling import profiler
     from tests.profiling_v2.collector import pprof_utils
+    from tests.profiling_v2.collector.test_stack import func1
 
     pprof_prefix = os.environ["DD_PROFILING_OUTPUT_PPROF"]
     output_filename = pprof_prefix + "." + str(os.getpid())
@@ -830,79 +849,3 @@ def _trace():
 
     for t in threads:
         t.join()
-
-
-@pytest.mark.skipif(not TESTING_GEVENT or sys.version_info < (3, 9), reason="Not testing gevent")
-@pytest.mark.subprocess(ddtrace_run=True)
-def test_collect_gevent_threads():
-    import gevent.monkey
-
-    gevent.monkey.patch_all()
-
-    import os
-    import threading
-    import time
-
-    from ddtrace.internal.datadog.profiling import ddup
-    from ddtrace.profiling.collector import stack
-    from tests.profiling_v2.collector import pprof_utils
-
-    iteration = 100
-    sleep_time = 0.01
-    nb_threads = 15
-
-    # Start some greenthreads: they do nothing we just keep switching between them.
-    def _nothing():
-        for _ in range(iteration):
-            # Do nothing and just switch to another greenlet
-            time.sleep(sleep_time)
-
-    test_name = "test_collect_gevent_threads"
-    pprof_prefix = "/tmp/" + test_name
-    output_filename = pprof_prefix + "." + str(os.getpid())
-
-    assert ddup.is_available
-    ddup.config(env="test", service="test_collect_gevent_threads", version="my_version", output_filename=pprof_prefix)
-    ddup.start()
-
-    with stack.StackCollector():
-        threads = []
-        i_to_tid = {}
-        for i in range(nb_threads):
-            t = threading.Thread(target=_nothing, name="TestThread %d" % i)
-            i_to_tid[i] = t.ident
-            t.start()
-            threads.append(t)
-        for t in threads:
-            t.join()
-
-    ddup.upload()
-
-    profile = pprof_utils.parse_newest_profile(output_filename)
-    samples = pprof_utils.get_samples_with_label_key(profile, "task name")
-    assert len(samples) > 0
-
-    for task_id in range(nb_threads):
-        pprof_utils.assert_profile_has_sample(
-            profile,
-            samples,
-            expected_sample=pprof_utils.StackEvent(
-                task_name="TestThread %d" % task_id,
-                task_id=i_to_tid[task_id],
-                thread_id=i_to_tid[task_id],
-                locations=[
-                    pprof_utils.StackLocation(
-                        filename="test_stack.py",
-                        function_name="_nothing",
-                        line_no=_nothing.__code__.co_firstlineno + 3,
-                    )
-                ],
-            ),
-        )
-    pprof_utils.assert_profile_has_sample(
-        profile,
-        samples,
-        expected_sample=pprof_utils.StackEvent(
-            task_name="MainThread",
-        ),
-    )