review

Author: dubloom

tests/debugging/test_encoding.py

@@ -271,20 +271,13 @@ def test_process_tags_are_not_included_by_default():
 
 
 def test_process_tags_are_included():
-    from unittest.mock import patch
 
-    from ddtrace.internal.process_tags import _process_tag_reload
-    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
+    from tests.utils import process_tag_reload
+    from ddtrace.internal.settings._config import config
 
     try:
-        with patch("sys.argv", ["/path/to/test_script.py"]), patch("os.getcwd", return_value="/path/to/workdir"):
             config._process_tags_enabled = True
-            _process_tag_reload()
+            process_tag_reload()
             s = Snapshot(
                 probe=create_snapshot_line_probe(probe_id="batch-test", source_file="foo.py", line=42),
                 frame=inspect.currentframe(),
@@ -303,18 +296,9 @@ def test_process_tags_are_included():
             decoded = json.loads(payload.decode())
 
             assert "process_tags" in decoded[0]
-
-            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 decoded[0]["process_tags"] == expected_raw
     finally:
         config._process_tags_enabled = False
-        _process_tag_reload()
+        process_tag_reload()
 
 
 def test_batch_flush_reencode():