chore(telemetry): add process tags

Author: dubloom

ddtrace/internal/telemetry/data.py

@@ -56,6 +56,9 @@ def _get_application(key):
     This helper packs and unpacks get_application arguments to support caching.
     Cached() annotation only supports functions with one argument
     """
+    # avoid circular dependency
+    from ddtrace.internal.process_tags import process_tags
+
     service, version, env = key
 
     return {
@@ -67,6 +70,7 @@ def _get_application(key):
         "tracer_version": get_version(),
         "runtime_name": platform.python_implementation(),
         "runtime_version": _format_version_info(sys.implementation.version),
+        "process_tags": process_tags,
     }
 
 

tests/telemetry/test_data.py

@@ -6,6 +6,7 @@
 
 import ddtrace
 from ddtrace.internal.constants import DEFAULT_SERVICE_NAME
+from ddtrace.internal.process_tags import _process_tag_reload
 from ddtrace.internal.runtime.container import CGroupInfo
 from ddtrace.internal.telemetry.data import _format_version_info
 from ddtrace.internal.telemetry.data import _get_container_id
@@ -43,6 +44,36 @@ def test_get_application_with_values():
     assert application["env"] == "staging"
 
 
+def test_get_application_with_process_tags():
+    from ddtrace.internal.process_tags.constants import ENTRYPOINT_BASEDIR_TAG
+    from ddtrace.internal.process_tags.constants import ENTRYPOINT_NAME_TAG
+    from ddtrace.internal.process_tags.constants import ENTRYPOINT_TYPE_SCRIPT
+    from ddtrace.internal.process_tags.constants import ENTRYPOINT_TYPE_TAG
+    from ddtrace.internal.process_tags.constants import ENTRYPOINT_WORKDIR_TAG
+    from ddtrace.settings._config import config
+
+    with mock.patch("sys.argv", ["/path/to/test_script.py"]), mock.patch("os.getcwd", return_value="/path/to/workdir"):
+        try:
+            config._process_tags_enabled = True
+            _process_tag_reload()
+
+            application = get_application("", "", "")
+            assert "process_tags" in application
+
+            process_tags = application["process_tags"]
+
+            expected_raw = (
+                f"{ENTRYPOINT_BASEDIR_TAG}:to,"
+                f"{ENTRYPOINT_NAME_TAG}:test_script,"
+                f"{ENTRYPOINT_TYPE_TAG}:{ENTRYPOINT_TYPE_SCRIPT},"
+                f"{ENTRYPOINT_WORKDIR_TAG}:workdir"
+            )
+            assert process_tags == expected_raw
+        finally:
+            config._process_tags_enabled = False
+            _process_tag_reload()
+
+
 def test_application_with_setenv(run_python_code_in_subprocess, monkeypatch):
     """
     validates the return value of get_application when DD_SERVICE, DD_VERSION, and DD_ENV environment variables are set