Use relative paths for codetf results

integration_tests/base_test.py

@@ -8,12 +8,16 @@
 from codemodder import __VERSION__
 
 
+SAMPLES_DIR = "tests/samples"
+
+
 class CleanRepoMixin:
     @classmethod
     def teardown_class(cls):
         """Ensure any re-written file is undone after integration test class"""
+        # TODO: we should probably use temporary directories instead
         repo = git.Git(os.getcwd())
-        repo.execute(["git", "checkout", os.getcwd() + "/tests/samples"])
+        repo.execute(["git", "checkout", os.path.join(os.getcwd(), SAMPLES_DIR)])
 
         pathlib.Path(cls.output_path).unlink(missing_ok=True)
 
@@ -54,9 +58,9 @@ def _assert_run_fields(self, run, output_path):
         assert run["elapsed"] != ""
         assert (
             run["commandLine"]
-            == f"codemodder tests/samples/ --output {output_path} --codemod-include={self.codemod.name()}"
+            == f"codemodder {SAMPLES_DIR} --output {output_path} --codemod-include={self.codemod.name()}"
         )
-        assert run["directory"] == os.path.abspath("tests/samples/")
+        assert run["directory"] == os.path.abspath(SAMPLES_DIR)
         assert run["sarifs"] == []
 
     def _assert_results_fields(self, results, output_path):
@@ -88,7 +92,10 @@ def _assert_codetf_output(self):
         run = codetf["run"]
         self._assert_run_fields(run, self.output_path)
         results = codetf["results"]
-        self._assert_results_fields(results, self.code_path)
+        # CodeTf2 spec requires relative paths
+        self._assert_results_fields(
+            results, os.path.relpath(self.code_path, SAMPLES_DIR)
+        )
 
     def check_code_before(self):
         with open(self.code_path, "r", encoding="utf-8") as f:
@@ -113,7 +120,7 @@ def test_file_rewritten(self):
             "python",
             "-m",
             "codemodder",
-            "tests/samples/",
+            SAMPLES_DIR,
             "--output",
             self.output_path,
             f"--codemod-include={self.codemod.name()}",

src/codemodder/__main__.py

@@ -74,7 +74,8 @@ def run_codemods_for_file(
 
             codemod_kls.CHANGESET_ALL_FILES.append(
                 ChangeSet(
-                    str(file_context.file_path),
+                    # TODO: we should not be using global state here
+                    str(file_context.file_path.relative_to(global_state.DIRECTORY)),
                     diff,
                     changes=file_context.codemod_changes,
                 ).to_json()

src/codemodder/codemods/base_codemod.py

@@ -26,6 +26,8 @@ class BaseCodemod:
 
     def __init_subclass__(cls, **kwargs):
         super().__init_subclass__(**kwargs)
+        # TODO: this should not belong to codemod classes
+        # Instead it should be owned by a global context that gets passed in
         cls.CHANGESET_ALL_FILES: list = []
 
         if "codemodder.codemods.base_codemod.SemgrepCodemod" in str(cls):

src/codemodder/global_state.py

@@ -3,6 +3,7 @@
 
 # Source code directory codemodder will analyze.
 # Should be overridden when codemodder starts.
+# TODO: we should not be managing global state at all
 DIRECTORY = ""