Merge pull request #131 from Pennycook/associate-fix

Fix associate() handling of #include from symlink

Author: Pennycook

codebasin/finder.py

@@ -117,7 +117,7 @@ def associator(node: Node) -> Visit:
             association[node].add(platform.name)
             active = node.evaluate_for_platform(
                 platform=platform,
-                filename=filename,
+                filename=self._get_realpath(filename),
                 state=self,
             )
 

tests/include/test_include.py

@@ -2,6 +2,8 @@
 # SPDX-License-Identifier: BSD-3-Clause
 
 import logging
+import os
+import tempfile
 import unittest
 from pathlib import Path
 
@@ -43,6 +45,30 @@ def test_include(self):
             "Mismatch in setmap",
         )
 
+    def test_include_from_symlink(self):
+        """Check included file correctly identifies its parent"""
+        tmp = tempfile.TemporaryDirectory()
+        p = Path(tmp.name)
+        with open(p / "test.cpp", mode="w") as f:
+            f.write('#include "test.h"')
+        open(p / "test.h", mode="w").close()
+        os.symlink(p / "test.cpp", p / "symlink.cpp")
+
+        codebase = CodeBase(p)
+        configuration = {
+            "test": [
+                {
+                    "file": str(p / "symlink.cpp"),
+                    "defines": [],
+                    "include_paths": [],
+                    "include_files": [],
+                },
+            ],
+        }
+        _ = finder.find(self.rootdir, codebase, configuration)
+
+        tmp.cleanup()
+
 
 if __name__ == "__main__":
     unittest.main()