chore(gunicorn): add gunicorn and gevent test scenarios

riotfile.py

@@ -2589,7 +2589,7 @@ def select_pys(min_version=MIN_PYTHON_VERSION, max_version=MAX_PYTHON_VERSION):
         Venv(
             name="gunicorn",
             command="pytest {cmdargs} tests/contrib/gunicorn",
-            pkgs={"requests": latest},
+            pkgs={"requests": latest, "gevent": latest},
             venvs=[
                 Venv(
                     pys="2.7",

tests/contrib/gunicorn/gevent_wsgi_mw_app.py

@@ -0,0 +1,23 @@
+import ddtrace.bootstrap.sitecustomize  # noqa: F401  # isort: skip
+from ddtrace import tracer  # noqa: I001
+from ddtrace.contrib.wsgi import DDWSGIMiddleware
+from tests.webclient import PingFilter
+
+
+tracer.configure(
+    settings={
+        "FILTERS": [PingFilter()],
+    }
+)
+
+
+def simple_app(environ, start_response):
+    if environ["RAW_URI"] == "/shutdown":
+        tracer.shutdown()
+
+    data = b"Hello, World!\n"
+    start_response("200 OK", [("Content-Type", "text/plain"), ("Content-Length", str(len(data)))])
+    return iter([data])
+
+
+app = DDWSGIMiddleware(simple_app)

tests/contrib/gunicorn/test_gunicorn.py

@@ -3,6 +3,7 @@
 import sys
 from typing import Dict
 from typing import NamedTuple
+from typing import Optional
 
 import pytest
 import tenacity
@@ -41,9 +42,12 @@ def _gunicorn_settings_factory(
     bind="0.0.0.0:8080",  # type: str
     use_ddtracerun=True,  # type: bool
     post_worker_init="",  # type: str
+    gevent_patching=None,  # type: Optional[bool]
 ):
     # type: (...) -> GunicornServerSettings
     """Factory for creating gunicorn settings with simple defaults if settings are not defined."""
+    if gevent_patching is not None:
+        env["DD_GEVENT_PATCH_ALL"] = str(gevent_patching)
     return GunicornServerSettings(
         env=env,
         directory=directory,
@@ -128,3 +132,56 @@ def test_traced_basic(gunicorn_server_settings, gunicorn_server):
     r = gunicorn_server.get("/")
     assert r.status_code == 200
     assert r.content == b"Hello, World!\n"
+
+
+@pytest.mark.parametrize(
+    "gunicorn_server_settings",
+    [
+        # 1. Set `DD_GEVENT_PATCH_ALL` environment variable to true when running with ddtrace-run
+        _gunicorn_settings_factory(
+            app_path="tests.contrib.gunicorn.wsgi_mw_app:app",
+            worker_class="gevent",
+            gevent_patching=True,
+            use_ddtracerun=True,
+        ),
+        # 2. Import ddtrace sitecustomize as first import of application
+        _gunicorn_settings_factory(
+            app_path="tests.contrib.gunicorn.gevent_wsgi_mw_app:app",
+            worker_class="gevent",
+            use_ddtracerun=False,
+        ),
+        # 3. Use post worker init hook to import ddtrace sitecustomize
+        _gunicorn_settings_factory(
+            app_path="tests.contrib.gunicorn.wsgi_mw_app:app",
+            worker_class="gevent",
+            use_ddtracerun=False,
+            post_worker_init=_post_worker_init_ddtrace,
+        ),
+    ],
+)
+def test_correct_gevent_patching_should_succeed(gunicorn_server_settings, gunicorn_server):
+    r = gunicorn_server.get("/")
+    assert r.status_code == 200
+    assert r.content == b"Hello, World!\n"
+
+
+@pytest.mark.parametrize(
+    "gunicorn_server_settings",
+    [
+        _gunicorn_settings_factory(
+            app_path="tests.contrib.gunicorn.wsgi_mw_app:app",
+            worker_class="gevent",
+            gevent_patching=None,
+        ),
+        _gunicorn_settings_factory(
+            app_path="tests.contrib.gunicorn.wsgi_mw_app:app",
+            worker_class="gevent",
+            gevent_patching=False,
+        ),
+    ],
+)
+def test_gevent_patch_falsy_should_fail(gunicorn_server_settings, gunicorn_server):
+    # FIXME: This test should fail as DD_GEVENT_PATCH_ALL is not set to 1.
+    r = gunicorn_server.get("/")
+    assert r.status_code == 200
+    assert r.content == b"Hello, World!\n"

tests/contrib/gunicorn/wsgi_mw_app.py

@@ -1,8 +1,11 @@
 from ddtrace import tracer
 from ddtrace.contrib.wsgi import DDWSGIMiddleware
+from ddtrace.debugging import DynamicInstrumentation
 from tests.webclient import PingFilter
 
 
+DynamicInstrumentation.enable()
+
 tracer.configure(
     settings={
         "FILTERS": [PingFilter()],